MongoDB Go Driver嵌入式文档无法自动填充的相关问题咨询

我正在使用MongoDB Go Driver开发一款教育类Go应用，遇到了嵌入式文档相关问题。我定义了嵌入Author结构体的Blog结构体，但从数据库解码Blog数据时，Author信息无法自动填充。

代码示例 #

type Blog struct {
    ID       primitive.ObjectID `json:"id" bson:"_id"`
    // ... 其他字段
    AuthorID primitive.ObjectID `bson:"author_id"`
    Author   Author             `bson:"-"`
}

type Author struct {
    ID   primitive.ObjectID `bson:"_id"`
    Name string             `bson:"name"`
}

查询代码 #

var blogs []Blog
cursor, err := coll.Find(context.TODO(), bson.D{})
if err != nil {
    log.Println(err)
    return []Blog{}
}

err = cursor.All(context.TODO(), &blogs)

结果问题 #

Author信息始终为空：

"Author": {
    "ID": "000000000000000000000000",
    "Name": ""
}

但AuthorID能正确填充：

{
    "id": "659906e6f3a7ec5556b4fa41",
    "title": "Defense the travel.",
    "content_text": "...",
    "created_at": "2023-03-16T19:06:12.184269",
    "updated_at": "2023-03-16T19:06:12.184272",
    "AuthorID": "659906b7f3a7ec5556b4fa40",
    "Author": {
        "ID": "000000000000000000000000",
        "Name": ""
    }
}

咨询以下问题：

1. 手动填充嵌入式文档是预期行为吗？MongoDB Go Driver不支持自动处理吗？
2. 有没有更高效的方法处理带有bson:"-"标签的嵌入式文档？
3. 移除bson:"-"标签是否可取，会引入潜在问题吗？

解答 #

1. 手动填充是预期行为，Driver不自动关联 #

MongoDB Go Driver只负责将数据库中的BSON数据直接解码到结构体字段，不会自动根据AuthorID去关联查询Author集合并填充Author字段。你给Author字段标记了bson:"-"，意味着Driver解码时会完全忽略这个字段，它只会被初始化为结构体的零值，这就是你看到空ID和空名称的原因。这种跨集合的关联填充需要自己实现，或者借助ORM框架简化，但原生Driver不提供该功能。

2. 高效处理bson:"-"字段的方法 #

如果要保留bson:"-"标签，推荐两种高效填充方式：

• 批量查询+映射填充：先取出所有Blog的AuthorID，一次性查询Author集合获取对应数据，再用map快速匹配填充到每个Blog的Author字段，大幅减少数据库请求次数。示例代码：

  // 获取所有blogs
  err = cursor.All(context.TODO(), &blogs)
  if err != nil { /* 处理错误 */ }
  
  // 收集所有AuthorID
  authorIDs := make([]primitive.ObjectID, 0, len(blogs))
  for _, blog := range blogs {
      authorIDs = append(authorIDs, blog.AuthorID)
  }
  
  // 批量查询作者
  var authors []Author
  authorCursor, err := authorColl.Find(context.TODO(), bson.M{"_id": bson.M{"$in": authorIDs}})
  if err != nil { /* 处理错误 */ }
  err = authorCursor.All(context.TODO(), &authors)
  if err != nil { /* 处理错误 */ }
  
  // 构建作者映射
  authorMap := make(map[primitive.ObjectID]Author)
  for _, a := range authors {
      authorMap[a.ID] = a
  }
  
  // 填充每个Blog的Author字段
  for i := range blogs {
      blogs[i].Author = authorMap[blogs[i].AuthorID]
  }
  

• 自定义解码函数：实现bson.Unmarshaler接口，在解码Blog时手动查询并填充Author字段。但这种方式会在解码单个文档时发起查询，性能不如批量查询，适合小数据量场景。

3. 移除bson:"-"标签的可行性与潜在问题 #

是否移除标签取决于你的数据存储结构：

• 如果数据库中Blog文档本身嵌入了Author完整数据（BSON里有author字段包含ID和Name），移除标签后Driver会自动解码嵌入数据到Author字段，完全没问题，无需手动填充。
• 如果数据库中Blog只有author_id字段，没有嵌入Author数据，移除标签后Driver会尝试找author字段，但数据库里没有，Author还是零值，解决不了问题。

潜在问题：

• 若后续将Blog结构体编码为BSON存入数据库，移除标签后Author字段会被序列化为author字段存入，可能造成数据冗余（已有author_id）；若修改Author字段后保存，会覆盖数据库中的嵌入数据（如果存在）。
• 数据库无author字段时，移除标签不会有数据问题，但Author字段依然是零值，无法解决原问题。

内容的提问来源于stack exchange，提问作者Malinda