使用Golang Mongo Driver执行简单查询耗时超2秒问题求助

核心问题根源 #

你的代码每次查询都创建新的MongoDB客户端并在查询后断开，这是导致耗时过高的直接原因。MongoDB客户端初始化（含连接池建立、握手等操作）本身属于高开销流程，远超过查询10条以内数据的耗时；而Python客户端默认复用连接池，因此耗时极低。

具体修复步骤 #

1. 复用MongoDB客户端实例 #

将客户端改为全局单例，在服务启动时初始化一次，后续所有查询复用该实例，避免重复创建/断开连接。

修改客户端初始化代码：

package database

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

// 全局复用的客户端与数据库实例
var globalMongoClient *mongo.Client
var globalDatabase *mongo.Database

func InitMongoDB() {
    cfg := config.GetEnvConfig()
    // 添加超时配置，避免默认长等待
    clientOpts := options.Client().ApplyURI(cfg.MongoURI).
        SetConnectTimeout(5 * time.Second).
        SetSocketTimeout(5 * time.Second)

    var err error
    globalMongoClient, err = mongo.Connect(context.TODO(), clientOpts)
    if err != nil {
        log.Fatalf("MongoDB client init failed: %v", err)
    }

    // 验证连接可用性
    if err := globalMongoClient.Ping(context.TODO(), nil); err != nil {
        log.Fatalf("MongoDB ping failed: %v", err)
    }
    globalDatabase = globalMongoClient.Database(cfg.MongoDatabase)
    log.Println("MongoDB client initialized successfully")
}

// 获取数据库实例
func GetDatabase() *mongo.Database {
    return globalDatabase
}

2. 修改查询函数，使用全局实例 #

func getMessageByIdFromDB(id string) (*Message, error) {
    db := database.GetDatabase()
    filter := map[string]string{
        "message_id": id,
    }

    var message Message
    start := time.Now()

    err := db.Collection(collectionName).FindOne(context.TODO(), filter).Decode(&message)

    elapsed := time.Since(start)
    log.Printf("Querying messages took %s", elapsed)

    if err != nil {
        return nil, err
    }
    return &message, nil
}

3. 在服务启动时初始化客户端 #

在main函数中先完成Mongo客户端初始化，服务退出时再统一断开：

func main() {
    // 初始化MongoDB客户端
    database.InitMongoDB()
    defer func() {
        if err := database.GetDatabase().Client().Disconnect(context.TODO()); err != nil {
            log.Fatalf("MongoDB disconnect failed: %v", err)
        }
    }()

    // 启动Web服务等业务逻辑
    // ...
}

额外验证点 #

• 确认message_id字段索引已正确创建（可在初始化时执行一次）：

  _, err := database.GetDatabase().Collection(collectionName).Indexes().CreateOne(
      context.TODO(),
      mongo.IndexModel{Keys: map[string]interface{}{"message_id": 1}},
  )
  

• 测试环境下可在MongoURI中添加directConnection=true，避免客户端不必要的拓扑探测开销。

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