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"`
}

如您所见，用户中的秘密字段为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。

Answer 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.
...

编辑：此处使用界面的原因是您可以使用常用方法通过线路发送任意模型。

Golang + MongoDB嵌入式（在另一个struct中嵌入一个struct）

2 个答案: