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

Posted 2023-03-29

【中文标题】Golang + MongoDB 嵌入类型（将一个结构嵌入到另一个结构中）

【英文标题】：Golang + MongoDB embedded type (embedding a struct in another struct)

【发布时间】：2013-10-17 06:00:33

【问题描述】：

假设，我运行一个 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)

【参考方案1】：

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

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

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

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

【讨论】：

在这种情况下，我收到运行时错误：内部服务器错误 struct main.adminUser 中的密钥“秘密”重复！

更新了我的答案。我没有意识到涉及到重复的键。

作为记录的旁注，",inline" 标签的使用甚至适用于普通字段（非嵌入/非匿名）。跨度>

拯救了我的一天。谢谢。

【参考方案2】：

另一种选择是声明一个接口。

type SecureModel interface 
    SecureMe()

确保您的模型实现它：

type User struct 
    Id       bson.ObjectId `json:"id,omitempty" bson:"_id,omitempty"`
    Username string        `json:"username" bson:"username"`
    Secret   string        `json:"secret,omitempty" bson:"secret"`


func (u *User) SecureMe() 
    u.Secret = ""

并且只根据调用的路由来调用它。

// I am being sent to a non-admin, secure me.
if _, ok := user.(SecureModel); ok 
    user.(SecureModel).SecureMe()

// Marshall to JSON, etc.
...

编辑：这里使用接口的原因是您可能使用通用方法通过网络发送任意模型。