Golang + MongoDB 嵌入类型（将一个结构嵌入到另一个结构中）

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《Golang + MongoDB 嵌入类型（将一个结构嵌入到另一个结构中）》，介绍一下golang，希望对大家的知识积累有所帮助，助力实战开发！

问题内容

假设，我运行一个 API，当用户对用户资源发出 GET 请求时，我会将相关字段作为 JSON 返回

type User struct {
  Id      bson.ObjectId `json:"id,omitempty" bson:"_id,omitempty"`
  Name    string        `json:"name,omitempty" bson:"name,omitempty"`
  Secret  string        `json:"-,omitempty" bson:"secret,omitempty"`
}

如您所见，User 中的 Secret 字段具有json:"-". 这意味着在大多数操作中我不想返回。在这种情况下，响应将是

{
  "id":1,
  "Name": "John"
}

json:"-"由于省略了字段，因此不会返回字段机密。

现在，我正在打开一条仅限管理员的路线，我想在其中返回秘密字段。但是，这意味着复制 User 结构。

我目前的解决方案如下所示：

type adminUser struct {      
  Id      bson.ObjectId `json:"id,omitempty" bson:"_id,omitempty"`
  Name    string        `json:"name,omitempty" bson:"name,omitempty"`
  Secret  string        `json:"secret,omitempty" bson:"secret,omitempty"`
}

有没有办法将用户嵌入到 adminUser 中？有点像继承：

type adminUser struct {      
  User
  Secret  string        `json:"secret,omitempty" bson:"secret,omitempty"`
}

以上目前不起作用，因为在这种情况下只会返回字段机密。

注意：在实际的代码库中，只有几十个字段。因此，复制代码的成本很高。

实际的mongo查询如下：

func getUser(w http.ResponseWriter, r *http.Request) {
  ....omitted code...

  var user adminUser
  err := common.GetDB(r).C("users").Find(
      bson.M{"_id": userId},
  ).One(&user)
  if err != nil {
      return
  }
  common.ServeJSON(w, &user)
}

正确答案

您应该查看 bson 包的内联标志（记录在bson.Marshal下）。它应该允许您执行以下操作：

type adminUser struct {
    User `bson:",inline"`
    Secret string `json:"secret,omitempty" bson:"secret,omitempty"`
}

但是，现在您会注意到，当您尝试使用这种结构从数据库中读取数据时，会出现重复键错误，因为adminUser和User都包含 key secret。

在您的情况下，我将从中删除该Secret字段，User 并且仅在adminUser. 然后，每当您需要写入该secret字段时，请确保使用adminUser.

文中关于golang的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Golang + MongoDB 嵌入类型（将一个结构嵌入到另一个结构中）》文章吧，也可关注golang学习网公众号了解相关技术文章。