MongoDB聚合查询优化：直接返回指定索引的Lobby对象而非嵌套结构

要直接返回lobby映射本身，无需嵌套在firstLobby字段下，可以通过MongoDB的$replaceRoot聚合阶段实现——它能将指定文档替换为聚合结果的根文档。

修改后的代码如下：

func GetLobbyByIndex(id string) error {
    _id, err := ConvertStringToOBjectID(id)
    if err != nil {
        return err
    }
    filter := bson.M{"_id": _id}

    var ctx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()
    cursor, err := AttendanceCollection.Aggregate(ctx, mongo.Pipeline{
        {{"$match", filter}},
        // 使用$replaceRoot将指定数组元素设为根文档
        {{"$replaceRoot", bson.M{
            "newRoot": bson.M{"$arrayElemAt": []interface{}{"$day1.lobbies", 0}},
        }}},
    })
    if err != nil {
        return err
    }

    for cursor.Next(ctx) {
        // 若有定义Lobby结构体，可直接解码到结构体，比如：var result Lobby
        var result = bson.M{}
        cursor.Decode(&result)
        fmt.Println(result)
    }
    return nil
}

注意事项 #

• 如果$day1.lobbies是空数组，$arrayElemAt会返回null，此时$replaceRoot会抛出错误。若需处理这种情况，可结合$ifNull做兜底：

  {"$replaceRoot", bson.M{
      "newRoot": bson.M{"$ifNull": []interface{}{
          bson.M{"$arrayElemAt": []interface{}{"$day1.lobbies", 0}},
          bson.M{}, // 空对象作为默认值，可根据需求调整
      }},
  }}
  

• 若已定义Lobby模型结构体，直接将var result = bson.M{}替换为var result Lobby即可完成类型安全的解码。

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