迁移mgo到qmgo实现MongoDB事务等特性支持

前言

mgo 是一个方便的 golang 操作 mongodb 的库，但是它对 mongodb 的最新特性的支持不足。
偏偏老项目是用 mgo 写的，如果要迁移到官方的 mongo-driver，官方的 mongo-driver 的封装更原始，用法差异较大，迁移成本会比较高。这时候我发现了一个 qmgo 库，是基于官方 mongo-driver 的封装，但是语法上更接近 mgo，适合我进行简单粗暴的迁移。
下面我记录一下迁移过程中使用的方法，是我自己摸索出来的，如果有问题欢迎指正。

批量替换大法

• 批量替换 github.com/globalsign/mgo/bson 为 go.mongodb.org/mongo-driver/bson
• 批量替换 bson.NewObjectId() 为 primitive.NewObjectID()
• 批量替换 bson.ObjectId 为 primitive.ObjectID
• 批量替换 .Find(bson.M 为 .Find(c, bson.M
• 批量替换 .Find(search 为 .Find(c, search
• 批量替换 .Insert( 为 .InsertOne(c,
• 批量替换 .Update( 为 .UpdateOne(c,
• 批量替换 .RemoveAll( 为 .RemoveAll(c,
• 批量替换 .Remove( 为 .Remove(c,
• 批量替换 errors.Is(err, mgo.ErrNotFound) 为 qmgo.IsErrNoDocuments(err)
• 批量替换 .EnsureIndex(mgo.Index{ 为 .CreateOneIndex(c, options.IndexModel{

复制粘贴大法

• 没有 context 的地方，到处复制粘贴 c := context.Background()
• 如果是 gin 的接口方法内，可以直接用 gin 的 c *gin.Context（但是如果多协程操作需要 Copy 一下 gin 的 Context）

函数替代大法

• 替代 bson.IsObjectIdHex()方法：

//判断是不是ObjectID的Hex
func IsObjectIDHex(hex string) bool {
	_, err := primitive.ObjectIDFromHex(hex)
	if err != nil {
		return false
	}
	return true
}

• 替代 bson.ObjectIdHex()方法：

//转换string为primitive.ObjectID
func ObjectIDHex(hex string) primitive.ObjectID {
	id, err := primitive.ObjectIDFromHex(hex)
	if err != nil {
		panic(fmt.Sprintf("Invalid input to ObjectIDHex: %+v", hex))
	}
	return id
}

这些函数可以直接找个辅助方法的包，或者自己的数据库的包，放在里面，然后用批量替换大法，就解决了。

总结

至此，项目在两小时内完成了迁移，并且顺利跑起来了。
感谢 qmgo 的作者，替我做好了轮子。
有机会我也做一些轮子开源出来。