go mongo

漆雕疏珂

2023-12-01

映射

MongoDB术语/概念	说明	对比SQL术语/概念
database	数据库	database
collection	集合	table
document	文档	row
field	字段	column
index	index	索引
primary key	主键 MongoDB自动将_id字段设置为主键	primary key

安装

go get go.mongodb.org/mongo-driver

连接mongoDB

func main() {
	// 设置客户端连接配置
	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")
	// mongodb://root:****@dds-bp*******1.mongodb.rds.aliyuncs.com:3717,dds-bp*******2.mongodb.rds.aliyuncs.com:3717/admin?replicaSet=mgset-XXXXX

	// 连接到MongoDB
	// 与mongo.NewClient()等价
	client, err := mongo.Connect(context.TODO(), clientOptions)
	if err != nil {
		log.Fatal(err)
	}

	// 检查连接
	err = client.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Connected to MongoDB!")
}

连接池模式

func ConnectToDB(uri, name string, timeout time.Duration, num uint64) (*mongo.Database, error) {
	ctx, cancel := context.WithTimeout(context.Background(), timeout)
	defer cancel()
	o := options.Client().ApplyURI(uri)
	o.SetMaxPoolSize(num)
	client, err := mongo.Connect(ctx, o)
	if err != nil {
		return nil, err
	}

	return client.Database(name), nil
}

断开与MongoDB的连接

// 断开连接
err = client.Disconnect(context.TODO())
if err != nil {
	log.Fatal(err)
}
fmt.Println("Connection to MongoDB closed.")

CRUD

// 指定获取要操作的数据集
collection := client.Database("q1mi").Collection("student")

BSON

MongoDB中的JSON文档存储在名为BSON(二进制编码的JSON)的二进制表示中。与其他将JSON数据存储为简单字符串和数字的数据库不同，BSON编码扩展了JSON表示，使其包含额外的类型，如int、long、date、浮点数和decimal128。这使得应用程序更容易可靠地处理、排序和比较数据。

连接MongoDB的Go驱动程序中有两大类型表示BSON数据：D和Raw。

类型D家族被用来简洁地构建使用本地Go类型的BSON对象。这对于构造传递给MongoDB的命令特别有用。D家族包括四类:

D：一个BSON文档。这种类型应该在顺序重要的情况下使用，比如- MongoDB命令。
M：一张无序的map。它和D是一样的，只是它不保持顺序。
A：一个BSON数组。
E：D里面的一个元素。

插入文档

s1 := Student{"小红", 12}
s2 := Student{"小兰", 10}
s3 := Student{"小黄", 11}

insertResult, err := collection.InsertOne(context.TODO(), s1)

students := []interface{}{s2, s3}
insertManyResult, err := collection.InsertMany(context.TODO(), students)

更新文档

updateone()方法允许你更新单个文档。它需要一个筛选器文档来匹配数据库中的文档，并需要一个更新文档来描述更新操作。你可以使用bson.D类型来构建筛选文档和更新文档:

filter := bson.D{{"name", "小兰"}}

update := bson.D{
	{"$inc", bson.D{
		{"age", 1},
	}},
}
updateResult, err := collection.UpdateOne(context.TODO(), filter, update)

查找文档

要查找多个文档，请使用collection.Find()。此方法返回一个游标。游标提供了一个文档流，你可以通过它一次迭代和解码一个文档。当游标用完之后，应该关闭游标。下面的示例将使用options包设置一个限制以便只返回两个文档

/ 创建一个Student变量用来接收查询的结果
var result Student
err = collection.FindOne(context.TODO(), filter).Decode(&result)

// 查询多个
// 将选项传递给Find()
findOptions := options.Find()
findOptions.SetLimit(2)

// 定义一个切片用来存储查询结果
var results []*Student

// 把bson.D{{}}作为一个filter来匹配所有文档
cur, err := collection.Find(context.TODO(), bson.D{{}}, findOptions)
if err != nil {
	log.Fatal(err)
}

// 查找多个文档返回一个光标
// 遍历游标允许我们一次解码一个文档
for cur.Next(context.TODO()) {
	// 创建一个值，将单个文档解码为该值
	var elem Student
	err := cur.Decode(&elem)
	if err != nil {
		log.Fatal(err)
	}
	results = append(results, &elem)
}

if err := cur.Err(); err != nil {
	log.Fatal(err)
}

// 完成后关闭游标
cur.Close(context.TODO())
fmt.Printf("Found multiple documents (array of pointers): %#v\n", results)

删除文档

可以使用collection.DeleteOne()或collection.DeleteMany()删除文档。如果你传递bson.D{{}}作为过滤器参数，它将匹配数据集中的所有文档。还可以使用collection. drop()删除整个数据集。

// 删除名字是小黄的那个
deleteResult1, err := collection.DeleteOne(context.TODO(), bson.D{{"name","小黄"}})
if err != nil {
	log.Fatal(err)
}
fmt.Printf("Deleted %v documents in the trainers collection\n", deleteResult1.DeletedCount)
// 删除所有
deleteResult2, err := collection.DeleteMany(context.TODO(), bson.D{{}})
if err != nil {
	log.Fatal(err)
}

解析到字典

G:=client.Database("General")
p:=G.Collection("player")
cur,_:=p.Find(context.Background(),bson.D{},nil)
res := make(map[string]interface{},20)
for cur.Next(context.Background()){
	err:=cur.Decode(res)
	if err!=nil {
		fmt.Println(err)
	}
}
fmt.Println(res)

Q&A

关于go使用mongodb，有两种用法，第一种用法是在接口中每次都直接使用mongoClient.Database().Collection()操作，第二种用法是提前在一个文件中生成collection = mongoClient.Database().Collection()，然后在接口中使用collection操作，哪个比较好一些。

第一种方法是在接口中每次都直接使用mongoClient.Database().Collection()操作。这种方法的优点是它可以确保你在使用每个collection时都是从最新的连接池中获取的。如果你的应用程序需要同时连接多个数据库或集群，则此方法可能更为合适。此外，如果你的应用程序只需要访问少量的collection，那么这种方法也比较容易实现。

然而，这种方法的缺点是它可能会导致冗余的连接或socket创建，这会降低性能。每次请求都要连接一次MongoDB，并创建新的Socket连接。如果你的应用程序需要频繁地访问collection，则此方法可能会对性能产生一定的负面影响。

第二种方法是提前在一个文件中生成collection = mongoClient.Database().Collection()，然后在接口中使用collection操作。这种方法的优点是它可以在你的应用程序启动时就提前创建一个集合，并在需要时重用。这可以减少连接和socket创建的次数，从而提高性能。如果你的应用程序需要频繁访问同一组collection，则此方法可能会更为合适。

但是，这种方法的缺点是如果你的应用程序需要连接多个数据库或集群，则可能需要在每个数据库或集群中创建不同的集合对象。此外，如果你的应用程序需要访问大量的collection，则预先创建所有集合对象可能会对内存产生负面影响。

总之，两种方法都有其优点和缺点，你需要考虑你的应用程序的具体需求和实现，以确定哪种方法最适合你的情况。