如何在Golang中使用MongoDB的$lookup获取计数？聚合联表咨询

嘿，看你已经在Golang里实现了MongoDB的聚合联表操作，现在想通过$lookup拿到关联文档的计数值对吧？这其实很容易实现，核心思路是利用$lookup返回关联文档数组后，用$size操作符获取数组的长度（也就是你要的计数值）。我给你具体的代码示例和步骤说明：

实现思路 #

1. 先用$lookup完成联表操作，将关联集合的文档以数组形式返回
2. 通过$addFields或$project操作符，用$size计算该数组的长度，得到关联文档的计数
3. 可选：如果不需要返回完整的关联数组，可以用$project过滤掉，减少数据传输

Golang代码示例 #

假设你的主集合是posts，关联集合是comments（对应你输出里的resultField评论数据），我们来实现统计每篇文章的评论数：

1. 定义对应的结构体 #

首先定义存储聚合结果的结构体（根据你的实际字段调整）：

package main

import (
    "context"
    "log"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

// PostResult 存储聚合后的文章结果，包含评论计数字段
type PostResult struct {
    ID           int      `bson:"_id"`
    AuthorName   string   `bson:"author_name"`
    Category     int      `bson:"category"`
    Content      string   `bson:"content"`
    DateTime     int      `bson:"date_time"`
    Excerpt      string   `bson:"excerpt"`
    Image        string   `bson:"image"`
    CommentCount int      `bson:"comment_count"`
    // 如果需要保留完整评论数组，可以加上下面的字段
    // ResultField []Comment `bson:"resultField"`
}

// Comment 评论结构体（如果需要保留完整数组的话）
type Comment struct {
    ID        int    `bson:"_id"`
    Comment   string `bson:"comment"`
    CommentOn int    `bson:"comment_on"`
    Email     string `bson:"email"`
    Name      string `bson:"name"`
}

2. 构建聚合管道 #

func main() {
    // 连接MongoDB
    client, err := mongo.Connect(context.TODO(), options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
    defer client.Disconnect(context.TODO())

    // 获取主集合（posts）
    collection := client.Database("your_db_name").Collection("posts")

    // 构建聚合管道
    pipeline := []bson.M{
        // 第一步：执行$lookup联表
        {
            "$lookup": bson.M{
                "from":         "comments",   // 关联的评论集合名
                "localField":   "_id",        // 主集合中用于关联的字段（文章ID）
                "foreignField": "post_id",    // 评论集合中对应的外键字段（关联的文章ID）
                "as":           "resultField",// 存储关联结果的数组字段名
            },
        },
        // 第二步：添加评论计数字段
        {
            "$addFields": bson.M{
                "comment_count": bson.M{"$size": "$resultField"}, // 用$size计算数组长度，即评论数
            },
        },
        // 第三步：可选，过滤不需要返回的字段
        {
            "$project": bson.M{
                "_id":           1,
                "author_name":   1,
                "category":      1,
                "content":       1,
                "date_time":     1,
                "excerpt":       1,
                "image":         1,
                "comment_count": 1, // 保留评论计数字段
                // 如果不需要完整评论数组，就去掉下面这行
                // "resultField":   1,
            },
        },
    }

    // 执行聚合操作
    cursor, err := collection.Aggregate(context.TODO(), pipeline)
    if err != nil {
        log.Fatal(err)
    }
    defer cursor.Close(context.TODO())

    // 解析聚合结果
    var results []PostResult
    if err := cursor.All(context.TODO(), &results); err != nil {
        log.Fatal(err)
    }

    // 打印结果
    for _, post := range results {
        log.Printf("文章ID: %d, 作者: %s, 评论数: %d", post.ID, post.AuthorName, post.CommentCount)
    }
}

关键说明 #

• $size操作符直接作用于$lookup返回的数组字段，就能得到关联文档的数量，不需要额外的分组或计数操作
• 如果你只需要计数值，不需要完整的关联数组，一定要在$project里去掉resultField，这样能减少数据库返回的数据量，提升查询性能
• 确保你的localField和foreignField字段类型一致（比如都是int类型），否则$lookup可能匹配不到数据

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