如何使用连接池将 mgo 会话转换为 mongo-go-driver 客户端？

Question

很久很久以前，当我们使用 mgo.v2 时，我们创建了一些包装函数来复制会话、设置读取首选项并返回该函数以供其他库使用，例如func NewMonotonicConnection() (conn *Connection, success bool) {    conn := &Connection{        session: baseSession.Copy(),    }    conn.session.SetMode(mongo.Monotonic, true)    return conn, true}现在，我们只需在 init 函数中传递默认客户端（使用 mongo.Connect 初始化并传递到连接单例中），然后像这样使用：func NewMonotonicConnection() (conn *Connection, success bool) {    conn = defaultConnection    return conn, true}我的理解是，要利用连接池，您需要使用相同的客户端（包含在 defaultConn 中），并且会话现在在 / .All()cursor 拆卸内部隐式处理。如果我在这里错了，请纠正我。如果我们仍然可以在这些连接上设置 readpref（例如，在返回之前在此连接上设置 NearestMode），那就太好了，但是社区/标准的做法是什么？我知道我可以一遍又一遍地调用 mongo.Connect，但是那贵吗？我可以创建不同的客户端 - 每个客户端都有不同的 readpref - 但我在想，如果在该连接上发生写入，它永远不会返回从从属设备读取。看起来我*可以显式创建会话，但我不确定我应该或者在新驱动程序中显式管理这些会话是否有任何影响。

Answer 1

在这个任务中，我通过 mongo-go-driver 代码库学到了一些东西，我认为在结束这个问题之前我应该与世界分享这些东西。如果我错了 - 请纠正我。

Connect()如果您想利用连接池，则不应一遍又一遍地调用。看起来每次调用 Connect() 时都会创建一个新套接字。defer Close()这意味着随着时间的推移，除非您每次都手动进行操作，否则存在套接字耗尽的风险。

All()在 mongo-go-driver 中，当您调用执行查询时（例如），会话会在幕后自动处理。您可以*显式地创建和拆除会话，但您不能使用我上面提出的单例方法来使用它，而不必更改所有调用方函数。这是因为您无法再在会话上调用查询操作，而是必须在数据库操作本身上使用 WithSession 函数来使用它

我意识到writeconcern,readpref和readconcern都可以设置为：

• 客户端级别（如果不覆盖，这些是所有内容都将使用的默认值）

• 会话级别

• 数据库级

• 查询级别

所以我所做的是创建数据库选项并重载 *mongo.Database 例如：

// Database is a meta-helper that allows us to wrap and overload

// the standard *mongo.Database type

type Database struct {

    *mongo.Database

}

// NewEventualConnection returns a new instantiated Connection

// to the DB using the 'Nearest' read preference.

// Per https://github.com/go-mgo/mgo/blob/v2/session.go#L61

// Eventual is the same as Nearest, but may change servers between reads.

// Nearest: The driver reads from a member whose network latency falls within

// the acceptable latency window. Reads in the nearest mode do not consider

// whether a member is a primary or secondary when routing read operations;

// primaries and secondaries are treated equivalently.

func NewEventualConnection() (conn *Connection, success bool) {

    conn = &Connection{

        client: baseConnection.client,

        dbOptions: options.Database().

            SetReadConcern(readconcern.Local()).

            SetReadPreference(readpref.Nearest()).

            SetWriteConcern(writeconcern.New(

                writeconcern.W(1))),

    }

    return conn, true

}

// GetDB returns an overloaded Database object

func (conn Connection) GetDB(dbname string) *Database {

    dbByName := &Database{conn.client.Database(dbname, conn.dbOptions)}

}

这使我能够利用连接池并保持与我们的代码库的向后兼容性。希望这对其他人有帮助。