有没有办法从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

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块。

Answer 2

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

如果有的话，您可以使用 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)
}

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

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

Is there a way to create a struct from a slice in golang

csv

struct

record

go

slice