自定义解组为一种类型而不是输入不同的类型

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《自定义解组为一种类型而不是输入不同的类型》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

我正在学习 go，我有一个问题：

我有一个 info 类型，其定义如下：

type info struct {
    id          id     `json:"id,omitempty"`
    displayname string `json:"display_name,omitempty"`
}

我制作了一个自定义 unmarshalljson 函数来解组此结构，因为作为输入我有：

• 一个 [] 接口{}，其位置 [0] 为一个整数，[1] 为一个字符串
• boolean 始终等于 false，表示该字段为 null

我希望当输入是 false 时，info 为零。

这是 unmarshalljson 函数

func (i *Info) UnmarshalJSON(data []byte) error {
var v []interface{}
if err := json.Unmarshal(data, &v); err != nil {
    var v bool
    if err = json.Unmarshal(data, &v); err != nil {
        return err
    }
    return nil
}
i.ID = ID(v[0].(float64))
i.DisplayName = v[1].(string)
return nil
}

这很丑陋，我想知道是否有更好的选择。 非常感谢。

正确答案

首先，您应该对意外的类型和长度更加警惕，以避免恐慌。然后，您可以解组到 []json.rawmessage 以推迟元素的解组，直到准备好为止。最后，您应该警惕无效的 true。

这是我尽力而为的，请其他人随意编辑（here是一个演示）：

func (i *Info) UnmarshalJSON(data []byte) error {
    var raw interface{}
    if err := json.Unmarshal(data, &raw); err != nil {
        return err
    }

    switch r := raw.(type) {
    case []interface{}:
    case bool:
        if r {
            return errors.New("unexpected true, must be array or false")
        }
        return nil
    default:
        return fmt.Errorf("unexpected type %T, must be array or false", r)
    }

    var v []json.RawMessage
    if err := json.Unmarshal(data, &v); err != nil {
        return err
    }
    if len(v) != 2 {
        return fmt.Errorf("unexpected length %d, must be 2", len(v))
    }

    if err := json.Unmarshal(v[0], &i.ID); err != nil {
        return err
    }
    if err := json.Unmarshal(v[1], &i.DisplayName); err != nil {
        return err
    }
    return nil
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《自定义解组为一种类型而不是输入不同的类型》文章吧，也可关注golang学习网公众号了解相关技术文章。