go使用mongodb

mongodb的go语言驱动比较流行是mongo-go-driver和mgo,前者是mongodb官方出的go驱动，mgo之前一直是个人在维护的，后来作者由于个原因已经放弃维护，后面出现mgo的分支还在继续维护不过更新频率还是比较低。本文主要介绍mongo-go-driver这个mongodb官方维护的版本，目前mongo-go-driver的版本已经是1.1.2已经可以用在生产环境。

mongodb client

目前大多数golang数据库驱动client 都会用连接池的方式来运行。 mongo-go-driver也不例外。 连接MongoDB代码如下：

client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017"))
	if err != nil {
		log.Fatal(err)
	}
	err = client.Ping(context.TODO(), nil)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Connected to MongoDB!")

运行下main.go程序

# go run main.go
Connected to MongoDB!

这边需要注意下，采用连接池后一般很少自己close client而是继续放在链接池中举行供其他请求使用。 如果你确定已经不需要使用Client，可以使用如下代码收到关闭。

err = client.Disconnect(nil）
if err != nil {
    log.Fatal(err)
}
fmt.Println("Connection to MongoDB closed.")

获取集合handle

collection = client.Database("testing").Collection("students")

为collection创建索引

indexView := collection.Indexes()
	ret,err :=indexView.CreateOne(context.Background(), mongo.IndexModel{
		Keys:  bsonx.Doc{{"name", bsonx.Int32(-1)}},
		Options: options.Index().SetName("testname").SetUnique(true), //这边设置了唯一限定，不设定默认不是唯一的
	})
fmt.Println(ret,err)

CURD操作

添加

func add()  {
	wida := Student{"wida", 32, "8001"}
	amy := Student{"amy", 25, "8002"}
	sunny := Student{"sunny", 35, "8003"}

	ret, err := collection.InsertOne(nil, wida)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("写入一个文档", ret.InsertedID)
	student := []interface{}{amy, sunny}

	ret2, err := collection.InsertMany(nil, student)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("写入多个文档 ", ret2.InsertedIDs)
}

查询

func find()  {
	var result Student
	err := collection.FindOne(nil,  bson.D{{"name", "amy"}, {"age", 25}}).Decode(&result)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Printf("%+v\n", result)
	findOptions := options.Find()
	findOptions.SetLimit(3)

	var results []*Student

	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(nil)

	for _,r := range results {
		fmt.Printf("%+v\n", r)
	}
}

更新

filter := bson.D{{"no", "8001"}}
	update := bson.D{
		{"$set", bson.D{
			{"age", 33},
			{"name", "wida2"},
		}},
	}
	ret, err := collection.UpdateOne(nil, filter, update)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Printf("Matched %v documents and updated %v documents.\n", ret.MatchedCount, ret.ModifiedCount)
}

删除

func del()  {
	deleteResult, err := collection.DeleteMany(context.TODO(), bson.D{{}}) // 这边删除全部文档
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("Deleted %v documents\n", deleteResult.DeletedCount)
}

参考文档

• MongoDB Go Driver
• MongoDB Go Driver Tutorial