mongodb - Golang + MongoDB 嵌入类型(在另一个结构中嵌入一个结构)

Question

假设，我运行一个 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:"-" 省略了该字段，因此不会返回该字段 secret。

现在，我正在打开一个管理员专用路由，我想在其中返回 secret 字段。但是，这意味着要复制 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"`
}

有没有办法将 User 嵌入到 adminUser 中？有点像继承:

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

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

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

实际的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)
}

Answer 1

你应该看看 bson 包的内联标志(记录在 bson.Marshal 下)。它应该允许你做这样的事情:

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

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

在你的情况下，我会从 User 中删除 Secret 字段并且只有 adminUser 中有一个。然后，每当您需要写入 secret 字段时，确保使用 adminUser。