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。