解组可能会也可能不会返回数组的JSON？

Question

我正在从第三方网站(家庭电力使用)中检索JSON,并且根据我从网站请求的内容,返回的JSON可能是也可能不是数组.例如,如果我请求我的智能电表列表,我会得到这个(由于大尺寸,结果被截断):

{"gwrcmds":{"gwrcmd":{"gcmd":"SPA_UserGetSmartMeterList","gdata":{"gip":{"version":"1"...

其中gwrcmd是单个元素.

但如果我要求过去半小时用电,我会得到这个:

{"gwrcmds":{"gwrcmd":[{"gcmd":"DeviceGetChart","gdata":{"gip":{"version":"1" ...

看看gwrcmd现在是一个数组？

在我的Go应用程序中,我有一个看起来像这样的结构(再次,截断,因为它持续了一段时间."Version"下面有更多的子结构和属性:

type Response struct {
    Gwrcmds struct {
        Gwrcmd struct {
            Gcmd  string
            Gdata struct {
                Gip struct {
                    Version string

如果gwrcmd是数组,则Gwrcmd需要是a []struct { },但如果不是,则只是常规的struct { }

问题是json.Unmarshal如果JSON有一个数组且结构没有切片(反之亦然),则只返回错误.

我是否需要创建一个复制第一个结构的第二个结构(除了[]struct { }替代结构),或者有更好的方法吗？我想到了接口的东西,但我还没有碰到那些,所以我不是百分之百确定它们.

Answer 1

通常,只要您拥有未知类型的JSON值,您就可以使用json.RawMessage它来查看它,并将其正确地解组为相应的类型.一个简化的例子:

// The A here can be either an object, or a JSON array of objects.
type Response struct {
    RawAWrapper struct {
        RawA json.RawMessage `json:"a"`
    }
    A  A   `json:"-"`
    As []A `json:"-"`
}

type A struct {
    B string
}

func (r *Response) UnmarshalJSON(b []byte) error {
    if err := json.Unmarshal(b, &r.RawAWrapper); err != nil {
        return err
    }
    if r.RawAWrapper.RawA[0] == '[' {
        return json.Unmarshal(r.RawAWrapper.RawA, &r.As)
    }
    return json.Unmarshal(r.RawAWrapper.RawA, &r.A)
}

游乐场:http://play.golang.org/p/2d_OrGltDu.

尽管根据第一个字节猜测内容对我来说似乎并不太健壮.通常你会在你的JSON中找到某种线索(比如与动态相同的一个length或type一个级别的字段),告诉你是否有一个对象或一个数组.

也可以看看:

• 如何使用golang websockets解码多种消息类型？
• 部分JSON解组到Go中的地图中

Answer 2

您可以尝试制作自定义 json unmarshal 方法，例如

func (a *GwrcmCustom) UnmarshalJSON(b []byte) (err error) {
    g, ga := Gwrcmd{}, []Gwrcmd{}
    if err = json.Unmarshal(b, &g); err == nil {
        *a = make([]Gwrcmd, 1)
        []Gwrcmd(*a)[0] = Gwrcmd(g)
        return
    }
    if err = json.Unmarshal(b, &ga); err == nil {
        *a = GwrcmCustom(ga)
        return
    }
    return
}

GwrcmCustom是一个自定义类型，切片Gwrcm

type GwrcmCustom []Gwrcmd

所以我们总是会得到切片

试试这个Go playground

我希望这个能帮上忙