有没有办法从 golang 中的切片创建结构

Question

我需要阅读一个 csv 记录，它有很多字段并转换为结构。 目前我正在做如下

//Proclog is type to hold
type Proclog struct {
    LogType       string `json:"LogType"`
    RootLogID     string `json:"RootLogId"`
    SubLogID      string `json:"SubLogId"`
    TransactionID string `json:"TransactionId"`
    Instance      string `json:"Instance"`
    Operation     string `json:"Operation"`
    Status        string `json:"Status"`
    User          string `json:"User"`
    Hostname      string `json:"Hostname"`
    Protocol      string `json:"Protocol"`
    Target        string `json:"Target"`
    StartTime     string `json:"StartTime"`
    ExecuteTime   string  `json:"ExecuteTime"`
    ResponseCode  string `json:"ResponseCode"`
    FullRequest   string `json:"FullRequest"`
    FullResponse  string `json:"FullResponse"`
}

//ProclogFromCsvRecord convert csv record to LogConsolidation
func ProclogFromCsvRecord(record []string) (*Proclog, error) {
    if len(record) != 17 {
        return nil, csv.ErrFieldCount
    }
    return &Proclog{
        LogType:       record[0],
        RootLogID:     record[1],
        SubLogID:      record[2],
        TransactionID: record[3],
        Instance:      record[4],
        Operation:     record[5],
        Status:        record[6],
        User:          record[7],
        Hostname:      record[8],
        Protocol:      record[9],
        Target:        record[10],
        StartTime:     record[11],
        ExecuteTime:   record[12],
        ResponseCode:  record[13],
        FullRequest:   record[14],
        FullResponse:  record[15],
    }, nil
}

我不想对索引进行硬编码，有没有更好的方法将切片转换为没有硬编码索引的结构？

Answer 1

首先，我认为您的代码在您编写时非常好。 您的代码必须以一种或另一种方式包含字段顺序，而且您不能让它比您做的简单得多，这很好。

如果有的话，您可以使用iota指定字段顺序。

type ProclogField int

const (
        LogType ProclogField = iota
        RootLogID
        SubLogID
        TransactionID
        Instance
        Operation
        Status
        User
        Hostname
        Protocol
        Target
        StartTime
        ExecuteTime
        ResponseCode
        FullRequest
        FullResponse
        ProclogLength
)

现在，您可以在创建 Proclog 结构时用这些常量替换代码中的数字文字。

如果您可以在没有实际结构的情况下生存，则可以直接从切片中定义类型，如下所示：

type Proclog []string

func (p Proclog) Get(f ProclogField) string {
        return p[f]
}

func ProclogFromCsvRecord(record []string) (Proclog, error) {
        if len(record) != int(ProclogLength)+1 {
                return nil, csv.ErrFieldCount
        }
        return Proclog(record), nil
}

然后，如果您需要将此类型转换为 JSON，您可以像这样实现 json.Marshaler 接口：

//go:generate stringer -type=ProclogField
func (p Proclog) MarshalJSON() ([]byte, error) {
        m := make(map[string]string)
        for z := ProclogField(0); z < ProclogLength; z++ {
                m[z.String()] = p.Get(z)
        }
        return json.Marshal(m)
}

但是，这不是您所要求的，我不相信所有这些都比您开始使用的简单代码更好。

Answer 2

package main

import (
    "log"
    "reflect"
    "strconv"
)

type People struct {
    Name   string `mytag:"0"`
    Age    int    `mytag:"1"`
    Nunber int64  `mytag:"2"`
}

func FieldsArrayToStruct(p interface{}, values []interface{}) {
    rv := reflect.ValueOf(p).Elem()
    length := len(values)
    for i := 0; i < rv.NumField(); i++ {
        tagValue, ok := rv.Type().Field(i).Tag.Lookup("mytag")
        typeName := rv.Type().Field(i).Type.Name()
        if !ok {
            continue
        }
        if tagValue == "" {
            continue
        }
        index, err := strconv.Atoi(tagValue)
        if err != nil || index < 0 || index > length-1 {
            continue
        }
        switch values[i].(type) {
        case int:
            if typeName == "int" || typeName == "int32" || typeName == "int64" {
                rv.FieldByIndex(rv.Type().Field(i).Index).SetInt(int64(values[i].(int)))
            }
        case string:
            if typeName == "string" {
                rv.FieldByIndex(rv.Type().Field(i).Index).SetString(values[i].(string))
            }
        }
    }
}

func main() {
    var p = &People{}
    FilesArrayToStruct(p, []interface{}{"Tom", 19, 11110})
    log.Println(p)
}

也许你需要继续完成func FieldsArrayToStruct的switch块。