如何使用ObjectID正确bson.MarshalJSON(myStruct)?

时间:2017-11-30 07:26:16

标签: mongodb go mgo

当我从我的数据库中抓取一个帖子并尝试将其渲染为JSON时,我遇到了一些问题:

type PostBSON struct {
  Id      bson.ObjectId `bson:"_id,omitempty"`
  Title   string        `bson:"title"`
}

// ...

postBSON := PostBSON{}
id := bson.ObjectIdHex(postJSON.Id)
err = c.Find(bson.M{"_id": id}).One(&postBSON)

// ...

response, err := bson.MarshalJSON(postBSON)

MarshalJSON不会为我处理hexing Id(ObjectId)。我得到了:

{"Id":{"$oid":"5a1f65646d4864a967028cce"}, "Title": "blah"}

清理输出的正确方法是什么?

{"Id":"5a1f65646d4864a967028cce", "Title": "blah"}

修改:我编写了自己的stringify as described here。 这是一个高性能的解决方案吗? 它是愚蠢的吗?

func (p PostBSON) String() string {
  return fmt.Sprintf(`
        {
          "_id": "%s",
          "title": "%s",
          "content": "%s",
          "created": "%s"
        }`,
    p.Id.Hex(),
    p.Title,
    p.Content,
    p.Id.Time(),
  )

1 个答案:

答案 0 :(得分:2)

您可以实现MarshalJSON以满足json.Marshaler接口,例如:

func (a PostBSON) MarshalJSON() ([]byte, error) {
    m := map[string]interface{}{
        "id": a.Id.Hex(),
        "title": a.Title,
    }
    return json.Marshal(m)
}