Question

假设我有一个这样的结构：

type Message struct {
    Args   []interface{}
    Kwargs map[string]interface{}
}

message := Message{                                                                                                                                                                                            
    []interface{}{1, 2, 3, 4},                                                                                                                                                                                 
    map[string]interface{}{"a": 2, "b": 3},                                                                                                                                                                    
}

我应该如何编组消息以拥有这样的JSON？

[[1,2,3,4], {"a": 2, "b":3}]

Answer 1

您可以在您的结构中添加编组方法来处理逻辑。

行中的内容

func (m Message) MarshalJSON() ([]byte, error) {
    data := make([]interface{}, 0)
    data = append(data, m.Args)
    data = append(data, m.Kwargs)
    return json.Marshal(data)
}

Try it on the Playground

Answer 2

输出中想要的是一个JSON数组，其中包含Args结构值的Kwargs和message字段，因此可以通过整理以下切片值来获得所需的内容：

[]interface{}{message.Args, message.Kwargs}

例如：

message := Message{
    []interface{}{1, 2, 3, 4},
    map[string]interface{}{"a": 2, "b": 3},
}

err := json.NewEncoder(os.Stdout).
    Encode([]interface{}{message.Args, message.Kwargs})

fmt.Println(err)

以上内容的输出（在Go Playground上尝试）：

[[1,2,3,4],{"a":2,"b":3}]
<nil>

这适用于此特定情况。如果您需要一个将结构值的字段编组为数组元素（例如数组元素）的通用解决方案，则可以创建一个辅助函数，将字段“打包”成切片：

func getFields(i interface{}) (res []interface{}) {
    v := reflect.ValueOf(i)
    if v.Kind() == reflect.Ptr {
        v = v.Elem()
    }
    if v.Kind() != reflect.Struct {
        return nil
    }

    for i := 0; i < v.NumField(); i++ {
        f := v.Field(i)
        if f.CanInterface() {
            res = append(res, f.Interface())
        }
    }
    return res
}

以上getFields()接受结构值和结构指针。使用它的示例：

message := Message{
    []interface{}{1, 2, 3, 4},
    map[string]interface{}{"a": 2, "b": 3},
}

err := json.NewEncoder(os.Stdout).Encode(getFields(message))
fmt.Println(err)

err = json.NewEncoder(os.Stdout).Encode(getFields(&message))
fmt.Println(err)

输出（在Go Playground上尝试）：

[[1,2,3,4],{"a":2,"b":3}]
<nil>
[[1,2,3,4],{"a":2,"b":3}]
<nil>

在go

2 个答案: