如何在Go语言中高效地将MongoDB文档转换为JSON API响应

package main

import (
    "encoding/json"
    "fmt"
    "log"

    "gopkg.in/mgo.v1"
    "gopkg.in/mgo.v1/bson"
)

// 假设这是你的MongoDB会话和集合
var myCollection *mgo.Collection

func init() {
    // 实际应用中，你需要建立MongoDB连接
    // 这是一个模拟的初始化，实际需要替换为你的MongoDB连接逻辑
    session, err := mgo.Dial("mongodb://localhost:27017") // 替换为你的MongoDB连接字符串
    if err != nil {
        log.Fatalf("Failed to connect to MongoDB: %v", err)
    }
    session.SetMode(mgo.Monotonic, true)
    myCollection = session.DB("mydatabase").C("mycollection")

    // 插入一些测试数据（如果集合为空）
    count, _ := myCollection.Count()
    if count == 0 {
        myCollection.Insert(
            bson.M{"name": "Alice", "age": 30, "city": "New York"},
            bson.M{"name": "Bob", "age": 25, "city": "London"},
            bson.M{"name": "Alice", "age": 32, "city": "Paris"},
        )
        fmt.Println("Inserted test data.")
    }
}

// GetDocumentsAsJSON retrieves documents from Mongo and returns them as a JSON byte slice
func GetDocumentsAsJSON(name string) ([]byte, error) {
    var results []bson.M // 声明一个bson.M切片来存储查询结果

    // 执行查询，并将结果直接反序列化到 []bson.M
    err := myCollection.Find(
        bson.M{"name": name},
    ).All(&results)

    if err != nil {
        return nil, fmt.Errorf("failed to query MongoDB: %w", err)
    }

    // 使用 encoding/json 包将 []bson.M 序列化为 JSON 字节切片
    jsonData, err := json.Marshal(results)
    if err != nil {
        return nil, fmt.Errorf("failed to marshal JSON: %w", err)
    }

    return jsonData, nil
}

func main() {
    // 示例用法
    nameToFind := "Alice"
    jsonResponse, err := GetDocumentsAsJSON(nameToFind)
    if err != nil {
        log.Fatalf("Error getting documents: %v", err)
    }

    fmt.Printf("JSON API Response for name '%s':\n%s\n", nameToFind, string(jsonResponse))

    nameToFind = "Bob"
    jsonResponse, err = GetDocumentsAsJSON(nameToFind)
    if err != nil {
        log.Fatalf("Error getting documents: %v", err)
    }
    fmt.Printf("JSON API Response for name '%s':\n%s\n", nameToFind, string(jsonResponse))

    // 清理（可选）
    // defer func() {
    //  if myCollection != nil {
    //      myCollection.Database.Session.Close()
    //  }
    // }()
}