mongodb官方的golang驱动基础使用

导入

go get github.com/mongodb/mongo-go-driver/mongo

struct里面获取ObjectID 点击这里

链接mongo服务

	if client, err = mongo.Connect(getContext(), url); err != nil {
		checkErr(err)
	}

判断服务是否可用

	if err = client.Ping(getContext(), readpref.Primary()); err != nil {
		checkErr(err)
	}

选择数据库和集合

	collection = client.Database("testing_base").Collection("howie")

设置集合内数据过期时间

	k := mongo.IndexModel{
		Keys:    bsonx.Doc{{"expiredtime", bsonx.Int32(1)}},
		Options: options.Index().SetExpireAfterSeconds(1 * 60),//60秒后过期，详细请查询完整的代码演示
	}
	_, err = collection.Indexes().CreateOne(getContext(), k)

删除这个集合

	collection.Drop(getContext())

插入一条数据

	if insertOneRes, err = collection.InsertOne(getContext(), howieArray[0]); err != nil {
		checkErr(err)
	}
	fmt.Printf("InsertOne插入的消息ID:%v\n", insertOneRes.InsertedID)

批量插入数据

	if insertManyRes, err = collection.InsertMany(getContext(), howieArray); err != nil {
		checkErr(err)
	}
	fmt.Printf("InsertMany插入的消息ID:%v\n", insertManyRes.InsertedIDs)

查询单条数据

	if err = collection.FindOne(getContext(), bson.D{{"name", "howie_2"}, {"age", 11}}).Decode(&howie); err != nil {
		checkErr(err)
	}
	fmt.Printf("FindOne查询到的数据:%v\n", howie)

查询单条数据后删除该数据

	if err = collection.FindOneAndDelete(getContext(), bson.D{{"name", "howie_3"}}).Decode(&howie); err != nil {
		checkErr(err)
	}
	fmt.Printf("FindOneAndDelete查询到的数据:%v\n", howie)

询单条数据后修改该数据

	if err = collection.FindOneAndUpdate(getContext(), bson.D{{"name", "howie_4"}}, bson.M{"$set": bson.M{"name": "这条数据我需要修改了"}}).Decode(&howie); err != nil {
		checkErr(err)
	}
	fmt.Printf("FindOneAndUpdate查询到的数据:%v\n", howie)

查询单条数据后替换该数据(以前的数据全部清空)

	if err = collection.FindOneAndReplace(getContext(), bson.D{{"name", "howie_5"}}, bson.M{"hero": "这条数据我替换了"}).Decode(&howie); err != nil {
		checkErr(err)
	}
	fmt.Printf("FindOneAndReplace查询到的数据:%v\n", howie)

一次查询多条数据(查询createtime>=3,限制取2条,createtime从大到小排序的数据)

	if cursor, err = collection.Find(getContext(), bson.M{"createtime": bson.M{"$gte": 2}}, options.Find().SetLimit(2), options.Find().SetSort(bson.M{"createtime": -1})); err != nil {
		checkErr(err)
	}
	if err = cursor.Err(); err != nil {
		checkErr(err)
	}
	defer cursor.Close(context.Background())
	for cursor.Next(context.Background()) {
		if err = cursor.Decode(&howie); err != nil {
			checkErr(err)
		}
		howieArrayEmpty = append(howieArrayEmpty, howie)
	}
	fmt.Printf("Find查询到的数据:%v\n", howieArrayEmpty)

查询集合里面有多少数据

	if size, err = collection.Count(getContext(), nil); err != nil {
		checkErr(err)
	}
	fmt.Printf("Count里面有多少条数据:%d\n", size)

查询集合里面有多少数据(查询createtime>=3的数据)

	if size, err = collection.Count(getContext(), bson.M{"createtime": bson.M{"$gte": 3}}); err != nil {
		checkErr(err)
	}
	fmt.Printf("Count里面有多少条数据:%d\n", size)

修改一条数据

	if updateRes, err = collection.UpdateOne(getContext(), bson.M{"name": "howie_2"}, bson.M{"$set": bson.M{"name": "我要改了他的名字"}}); err != nil {
		checkErr(err)
	}
	fmt.Printf("UpdateOne的数据:%d\n", updateRes)

修改多条数据

	if updateRes, err = collection.UpdateMany(getContext(), bson.M{"createtime": bson.M{"$gte": 3}}, bson.M{"$set": bson.M{"name": "我要批量改了他的名字"}}); err != nil {
		checkErr(err)
	}
	fmt.Printf("UpdateMany的数据:%d\n", updateRes)

删除一条数据

	if delRes, err = collection.DeleteOne(getContext(), bson.M{"name": "howie_1"}); err != nil {
		checkErr(err)
	}
	fmt.Printf("DeleteOne删除了多少条数据:%d\n", delRes.DeletedCount)

删除多条数据

	if delRes, err = collection.DeleteMany(getContext(), bson.M{"createtime": bson.M{"$gte": 7}}); err != nil {
		checkErr(err)
	}
	fmt.Printf("DeleteMany删除了多少条数据:%d\n", delRes.DeletedCount)

使用事务

	//不能在单节点使用（副本集可以使用）
func UseSession(client *mongo.Client) {
	client.UseSession(getContext(), func(sctx mongo.SessionContext) error {
		err := sctx.StartTransaction(options.Transaction().
			SetReadConcern(readconcern.Snapshot()).
			SetWriteConcern(writeconcern.New(writeconcern.WMajority())),
		)
		if err != nil {
			return err
		}
		_, err = client.Database("aa").Collection("bb").InsertOne(sctx, bson.D{{"aa", 3}})
		if err != nil {
			_ = sctx.AbortTransaction(sctx)
			return err
		}
		_, err = client.Database("aa").Collection("bb").InsertOne(sctx, bson.D{{"bb", 3}})
		if err != nil {
			_ = sctx.AbortTransaction(sctx)
			return err
		}
		for {
			err = sctx.CommitTransaction(sctx)
			switch e := err.(type) {
			case nil:
				return nil
			case mongo.CommandError:
				if e.HasErrorLabel("UnknownTransactionCommitResult") {
					continue
				}
				return e
			default:
				return e
			}
		}
	})
}

完整演示代码 点击这里

查看mongo BSON详细用法 点击这里

联系 QQ: 3355168235