一尘不染

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

go

假设,我运行一个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"`
}

有没有一种方法可以将User嵌入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)
}

阅读 339

收藏
2020-07-02

共1个答案

一尘不染

您应该看一下bson包的内联标志(记录在bson.Marshal下)。它应该允许您执行以下操作:

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

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

在您的情况下,我将从中删除该Secret字段,User
而只保留其中的一个adminUser。然后,每当需要写入secret字段时,请确保使用adminUser

2020-07-02