使用Golang连接MongoDB方法详解

MongoDB Driver For Golang

Golang连接MongoDB的库有很多，本文主要讲解使用MongoDB官方驱动 go.mongodb.org/mongo-driver/mongo 连接MongoDB的方法。

支持使用URI类型的字符串连接MongoDB，字符串格式支持两种类型：

• 标准连接字符串格式
• DNS种子列表连接格式

接下来以标准连接字符串格式来做讲解。

使用Golang连接MongoDB

首先讲下标准连接字符串格式URI，格式为：

mongodb://[username:password@]host1[:port1][,...hostN[:portN]][/[defaultauthdb][?options]]

通过上篇文章《一文了解MongoDB的各种部署模式》我们知道Golang有几种不同的部署方式，不同的部署方式对应的URI也有些不同。

单节点（standalone）模式的格式为：

mongodb://username:password@mongodb0.example.com:27017

副本集模式的格式为：

mongodb://username:password@mongodb0.example.com:27017,mongodb1.example.com:27017,mongodb2.example.com:27017/?replicaSet=myRepl

对于副本集模式，需要指定副本集配置中列出的mongod实例的主机名，还需要指定replicaSet选项。

分片集群模式的格式为：

mongodb://username:password@mongos0.example.com:27017,mongos1.example.com:27017,mongos2.example.com:27017

不管那种模式，如果用户名或密码包含如下字符：

: / ? # [ ] @

必须将这些字符转换为百分比编码。

获取 mongo-driver/mongo 包：

go get go.mongodb.org/mongo-driver/mongo

示例代码如下：

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

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

func main() {
  // 这里uri使用副本集模式，如果你的MongoDB是其他模式，改为上面其他模式的uri即可
	uri := os.Getenv("mongodb://username:password@mongodb0.example.com:27017,mongodb1.example.com:27017,mongodb2.example.com:27017/?replicaSet=myRepl")
	opts :=  options.Client().ApplyURI(uri)
	client, err := mongo.Connect(context.TODO(), opts)
	if err != nil {
		panic(err)
	}
	coll := client.Database("sample_mflix").Collection("movies")
	title := "Back to the Future"

	var result bson.M
	err = coll.FindOne(context.TODO(), bson.D{{"title", title}}).Decode(&result)
	if err == mongo.ErrNoDocuments {
		fmt.Printf("No document was found with the title %s\n", title)
		return
	}
	if err != nil {
		panic(err)
	}

	jsonData, err := json.MarshalIndent(result, "", "    ")
	if err != nil {
		panic(err)
	}
	fmt.Printf("%s\n", jsonData)
}

也可以设置连接池的最大和最小连接数，示例代码如下：

opts :=  options.Client().ApplyURI(uri)
opts.MaxPoolSize = 20
opts.MinPoolSize = 10