为空自定义类型省略 JSON

Question

我正在尝试为可能的 nil 数据库类型编写自定义封送器。它的结构与sql.NullFloat64 类型完全相同：

type NullFloat64 sql.NullFloat64

func (ni *NullFloat64) MarshalJSON() ([]byte, error) {
    if !ni.Valid {
        return []byte("null"), nil
    }
    return json.Marshal(ni.Float64)
}

这是要编组的较大struct 的一部分：

type Data struct {
    X time.Time `json:"x"`
    Y float32 `json:"y"`
    Stderr NullFloat64 `json:"stderr"`
}

如果我尝试json.Marshal() 这个struct，它可以正常工作，创建：

{"x":"2017-01-12T23:36:12-05:00","y":4,"stderr":null}

如果值为null，我想完全省略 JSON 键。我将json:"stderr,omitempty" 添加到Data。

根据here 的建议，我尝试从MarshalJSON 返回一个nil 值，但得到了：

json: error calling MarshalJSON for type common.NullFloat64: unexpected end of JSON input

我还尝试将Data 更新为：

type Data struct {
    X time.Time `json:"x"`
    Y float32 `json:"y"`
    Stderr *NullFloat64 `json:"stderr,omitempty"`
}

和编组：

Data {
    X: datetime,
    Y: value,
    Stderr: &stderr,
}

但是在从MarshalJSON 返回nil 时出现与以前一样的错误。

那么，如何为自定义类型实现MarshalJSON 并在编组时省略键？ 谢谢您的帮助！

Answer 1

如果你创建，你可以这样输入：

Data {
    X: datetime,
    Y: value,
    Stderr: nil,
}

omitempty 将介入并“做正确的事”。很遗憾，我很确定这对您没有帮助。

如果你真的想省略基于字段的内部状态，你需要在你的结构上实现json.Marshaller，而不是它的子结构。最简单的方法如下：

func (d Data) MarshalJSON() ([]byte, error) {
    if !d.Stderr.Valid {
        return json.Marshal(Data{d.X, d.Y, nil})
    }
    return json.Marshal(d)
}