Golang + MongoDB嵌入式(在另一个struct中嵌入一个struct)

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:"-"省略了字段.

现在,我正在打开一个仅限管理员的路线,我想返回秘密字段.但是,这意味着重复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)
}

Answer 1

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

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

不过,现在你会发现,你得到重复键错误,当您试图从这种结构的数据库中读取,因为两者adminUser和User包含的关键secret.

在你的情况下,我将删除该Secret字段,User 并只有一个adminUser.然后,无论何时需要写入secret字段,请确保使用adminUser.