Mgo v2 MongoDB聚合查询：如何灵活处理可选查询条件？

在使用Mgo v2进行MongoDB聚合查询时，处理可选查询条件是一个常见的需求。为了灵活处理这些条件，你可以使用Go的bson.M或bson.D来动态构建查询条件。以下是一个示例，展示了如何根据不同的条件动态构建聚合管道。

示例场景

假设你有一个用户集合，你想要根据不同的条件（如年龄、性别、城市等）来查询用户数据。这些条件是可选的，用户可以选择提供或不提供这些条件。

示例代码

package main

import (
    "fmt"
    "log"
    "github.com/globalsign/mgo"
    "github.com/globalsign/mgo/bson"
)

type User struct {
    Name   string `bson:"name"`
    Age    int    `bson:"age"`
    Gender string `bson:"gender"`
    City   string `bson:"city"`
}

func main() {
    session, err := mgo.Dial("localhost")
    if err != nil {
        log.Fatal(err)
    }
    defer session.Close()

    db := session.DB("test")
    collection := db.C("users")

    // 假设这些是用户提供的可选条件
    age := 25
    gender := "male"
    city := "New York"

    // 动态构建匹配条件
    matchConditions := bson.M{}
    if age > 0 {
        matchConditions["age"] = age
    }
    if gender != "" {
        matchConditions["gender"] = gender
    }
    if city != "" {
        matchConditions["city"] = city
    }

    // 构建聚合管道
    pipeline := []bson.M{
        {"$match": matchConditions},
        // 可以继续添加其他聚合阶段
    }

    var results []User
    err = collection.Pipe(pipeline).All(&results)
    if err != nil {
        log.Fatal(err)
    }

    for _, user := range results {
        fmt.Printf("Name: %s, Age: %d, Gender: %s, City: %s\n", user.Name, user.Age, user.Gender, user.City)
    }
}

代码解释

1. 连接MongoDB：首先，我们使用mgo.Dial连接到MongoDB数据库。
2. 定义结构体：定义了一个User结构体来映射MongoDB中的文档。
3. 动态构建查询条件：根据用户提供的条件（如年龄、性别、城市），动态构建matchConditions。如果某个条件没有提供，则不会将其添加到matchConditions中。
4. 构建聚合管道：将matchConditions添加到聚合管道的$match阶段。你可以根据需要继续添加其他聚合阶段。
5. 执行聚合查询：使用collection.Pipe(pipeline).All(&results)执行聚合查询，并将结果存储在results中。
6. 输出结果：遍历results并输出每个用户的信息。

总结

通过动态构建bson.M或bson.D，你可以灵活地处理MongoDB聚合查询中的可选条件。这种方法使得代码更加模块化和可维护，能够适应不同的查询需求。