如何在Golang中根据查询串输入从MongoDB集合获取匹配数据？

我来帮你搞定这两种查询场景的实现，直接结合代码来说明会更清楚：

场景1：单字段匹配查询 #

当用户通过 http://localhost:8080/api/v1/customer?keyword=puneet 这种方式传参时，咱们可以把keyword匹配到集合里的指定字段（比如first_name、last_name，也可以同时匹配多个字段做OR逻辑，看你实际需求）。

实现步骤&代码示例 #

1. 先定义与MongoDB集合对应的Customer结构体：

import (
    "context"
    "encoding/json"
    "net/http"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/primitive"
    "go.mongodb.org/mongo-driver/mongo"
)

type Customer struct {
    ID        primitive.ObjectID `bson:"_id,omitempty"`
    FirstName string             `bson:"first_name"`
    LastName  string             `bson:"last_name"`
    Email     string             `bson:"email"`
    // 其他业务字段...
}

// 假设你已经初始化好MongoDB客户端，全局可用
var client *mongo.Client

2. 编写HTTP Handler处理请求：

func getCustomersHandler(w http.ResponseWriter, r *http.Request) {
    // 提取query参数中的keyword
    keyword := r.URL.Query().Get("keyword")
    if keyword == "" {
        http.Error(w, "keyword参数不能为空", http.StatusBadRequest)
        return
    }

    // 指定要查询的集合
    coll := client.Database("your_db_name").Collection("customers")

    // 构造查询条件：模糊匹配first_name或last_name（忽略大小写）
    filter := bson.M{
        "$or": []bson.M{
            {"first_name": bson.M{"$regex": keyword, "$options": "i"}},
            {"last_name": bson.M{"$regex": keyword, "$options": "i"}},
        },
    }

    // 如果是精确匹配单个字段，直接改成：
    // filter := bson.M{"first_name": keyword}

    // 执行查询
    cursor, err := coll.Find(r.Context(), filter)
    if err != nil {
        http.Error(w, "查询失败："+err.Error(), http.StatusInternalServerError)
        return
    }
    defer cursor.Close(r.Context())

    // 解析查询结果到切片
    var customers []Customer
    if err := cursor.All(r.Context(), &customers); err != nil {
        http.Error(w, "解析结果失败："+err.Error(), http.StatusInternalServerError)
        return
    }

    // 返回JSON格式的结果
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(customers)
}

场景2：多字段组合查询 #

当用户传多个字段参数，比如http://localhost:8080/api/v1/customer?first_name=puneet&last_name=jindal，咱们可以动态解析所有query参数，构造多字段的AND查询条件。

实现步骤&代码示例 #

修改上面的Handler，实现动态生成查询条件：

func getCustomersHandler(w http.ResponseWriter, r *http.Request) {
    // 获取所有query参数
    queryParams := r.URL.Query()
    if len(queryParams) == 0 {
        http.Error(w, "至少需要传入一个查询参数", http.StatusBadRequest)
        return
    }

    coll := client.Database("your_db_name").Collection("customers")

    // 动态构造多字段AND查询条件
    filter := bson.M{}
    for key, values := range queryParams {
        // 取参数的第一个值（query参数可能存在多值情况，根据业务调整）
        value := values[0]
        // 精确匹配字段，如需模糊匹配可替换成$regex逻辑
        filter[key] = value
    }

    // 执行查询、解析结果、返回JSON的步骤和场景1一致，这里省略重复代码
    cursor, err := coll.Find(r.Context(), filter)
    if err != nil {
        http.Error(w, "查询失败："+err.Error(), http.StatusInternalServerError)
        return
    }
    defer cursor.Close(r.Context())

    var customers []Customer
    if err := cursor.All(r.Context(), &customers); err != nil {
        http.Error(w, "解析结果失败："+err.Error(), http.StatusInternalServerError)
        return
    }

    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(customers)
}

额外提示 #

• 注意URL参数和MongoDB字段名的对应关系，如果URL用驼峰（比如firstName），可以做个字段映射转成MongoDB的下划线格式（first_name）。
• 数据量大时，模糊查询建议给对应字段建立文本索引，避免全表扫描影响性能。
• 可以添加参数校验逻辑，比如对邮箱、手机号等格式做校验后再查询。

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