如何使用 Go 中接口数组中的值初始化结构体？

偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《如何使用 Go 中接口数组中的值初始化结构体？》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！

我从服务器收到一条消息，例如 [0,"on",[6,1,5,"market",45.7]] 并将其保存到 []interface{} 变量中。我想用这个数组的值初始化结构。 我对 go 完全陌生，尝试这样做：

import "golang.org/x/net/websocket"
...
var msg []interface{}
// Server send response: `[0,"on",[6,1,5,"market",45.7]]`
if err := websocket.Message.Receive(ws, &msg); err != nil {
    logger.Println(err)
} else {
    type Order struct {
        ID int32,
        GID int32,
        CID int32,
        Type string,
        Amount float64
    }
    
    // here msg is [0,"on",[6,1,5,"market",45.7]]
    
    switch msg[1] {
    case "on":
        if rawOrder, ok := msg[2].([]interface{}); ok {
            order := Order{int32(rawOrder[0]), int32(rawOrder[1]), int32(rawOrder[2]), string(rawOrder[3]), float64(rawOrder[4])}
        }
}

但是我收到错误“无法将‘interface{}’类型的表达式转换为‘int32’类型”，下一步是对每个 raworder[i] 类型使用 switch，但它太长了。 我怎样才能轻松做到这一点？

正确答案

如果您知道 websocket 上使用的编解码器始终是 json，您可以正式定义 order 并给它一个 unmarshaljson 函数来为您进行解码。

import "golang.org/x/net/websocket"

type Order struct {
    ID, GID, CID int32
    Type         string
    Amount       float64
}

func (o *Order) UnmarshalJSON(data []byte) error {
    var first []json.RawMessage
    err := json.Unmarshal(data, &first)
    if err != nil {
        return fmt.Errorf("invalid order, must be array: %w", err)
    }
    if len(first) != 3 {
        return fmt.Errorf("invalid order, length must be 3, got %d", len(first))
    }

    var second string
    err = json.Unmarshal(first[1], &second)
    if err != nil {
        return fmt.Errorf("invalid order, second element must be string: %w", err)
    }

    switch second {
    case "on":
        var third []json.RawMessage
        err = json.Unmarshal(first[2], &third)
        if err != nil {
            return fmt.Errorf("invalid order, third element must be array: %w", err)
        }

        if len(third) != 5 {
            return fmt.Errorf("invalid order, element 3 length must be 5, got %d", len(third))
        }

        for i, f := range []interface{}{&o.ID, &o.GID, &o.CID, &o.Type, &o.Amount} {
            err = json.Unmarshal(third[i], f)
            if err != nil {
                return fmt.Errorf("invalid order, wrong type for element 3[%d]: %w", i, err)
            }
        }
        return nil
    }

    return fmt.Errorf("invalid order, unknown type %q", second)
}

...
var msg *Order
// Server send response: `[0,"on",[6,1,5,"market",45.7]]`
if err := websocket.JSON.Receive(ws, &msg); err != nil {
    logger.Println(err)
}

// msg is now an &Order{ID:6, GID:1, CID:5, Type:"market", Amount:45.7}

}

unmarshaljson 函数之所以庞大，是因为你的 api 很糟糕。如果您控制服务器；那么你应该避免在同一个数组中使用混合类型，并且应该避免将数组用于关系数据。

今天关于《如何使用 Go 中接口数组中的值初始化结构体？》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！