golang mongodb连接池实现-Go服务器开发

在Go语言中，可以使用MongoDB官方提供的go-mongo-driver库来进行数据库操作。下面介绍如何通过连接池实现MongoDB的连接复用。

连接池初始化

首先需要初始化一个连接池，可以使用sync.Pool类型来实现简单的连接池。下面是一个示例：

var clientPool *sync.Pool

func InitClientPool(uri string) {

clientPool = &sync.Pool{
    New: func() interface{} {
        ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
        defer cancel()
        client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
        if err != nil {
            log.Fatal(err)
        }
        return client
    },
}

}

其中，New函数返回一个新创建的mongo.Client对象，如果发生错误则会退出程序。uri参数是MongoDB服务器地址。

从连接池获取连接

接下来，在需要访问MongoDB数据库时，可以通过Get方法从连接池中获取一个已有的mongo.Client对象：

client := clientPool.Get().(*mongo.Client)

注意要将返回值断言为*mongo.Client类型。

将连接归还给连接池

完成对MongoDB数据库操作后，需要将已经使用的mongo.Client对象放回到连接池中以供复用：

clientPool.Put(client)

完整示例代码

以下是完整示例代码：

package main

import (
    "context"
    "log"
    "sync"
    "time"

    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

var clientPool *sync.Pool

func main() {
    uri := "mongodb://localhost:27017"
    InitClientPool(uri)

    client := clientPool.Get().(*mongo.Client)
    defer clientPool.Put(client)

    collection := client.Database("testdb").Collection("mycollection")
    filter := bson.M{"name": "Alice"}
    var result bson.M
    err = collection.FindOne(context.Background(), filter).Decode(&result)
}

func InitClientPool(uri string) {
    clientPool = &sync.Pool{
        New: func() interface{} {
            ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
            defer cancel()
            client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
            if err != nil {
                log.Fatal(err)
            }
            return client
        },
    }
}

注意：这里只是一个示例，实际使用中还需要考虑连接池的最大容量、空闲连接的超时时间等因素。