在Go语言开发中，使用mgo驱动操作MongoDB时，重命名集合是一个比较常见的操作需求，通常会在业务迭代需要调整集合结构、统一命名规范等场景下使用。mgo驱动本身没有直接提供重命名集合的方法，但可以通过执行MongoDB的原生命令来实现这个功能。
环境准备
首先需要确保已经安装了mgo驱动，如果还没有安装，可以通过下面的命令完成安装：
go get gopkg.in/mgo.v2
同时需要有一个可连接的MongoDB实例，并且已经通过mgo驱动建立了数据库连接。
核心实现步骤
1. 建立数据库连接
首先需要通过mgo驱动建立到MongoDB的连接，获取数据库会话和操作对象：
package main
import (
"fmt"
"gopkg.in/mgo.v2"
"gopkg.in/mgo.v2/bson"
)
func main() {
// 连接MongoDB，这里假设MongoDB运行在本地默认端口
session, err := mgo.Dial("127.0.0.1:27017")
if err != nil {
panic(err)
}
defer session.Close()
// 设置一致性模式
session.SetMode(mgo.Monotonic, true)
// 获取要操作的数据库，这里以test数据库为例
db := session.DB("test")
}
2. 执行重命名命令
MongoDB提供了renameCollection命令用于重命名集合，我们可以通过mgo驱动的Run方法执行该命令。命令的格式为：
{ renameCollection: "原数据库名.原集合名", to: "目标数据库名.目标集合名", dropTarget: false }
其中dropTarget参数表示如果目标集合已经存在，是否先删除目标集合，默认值为false，如果设置为true且目标集合存在，会直接覆盖。
下面是完整的重命名集合代码示例，假设要把test数据库下的old_collection重命名为new_collection：
package main
import (
"fmt"
"gopkg.in/mgo.v2"
"gopkg.in/mgo.v2/bson"
)
func main() {
// 建立数据库连接
session, err := mgo.Dial("127.0.0.1:27017")
if err != nil {
panic(err)
}
defer session.Close()
session.SetMode(mgo.Monotonic, true)
db := session.DB("test")
// 定义重命名命令参数
renameCmd := bson.M{
"renameCollection": "test.old_collection", // 原集合完整名称：数据库名.集合名
"to": "test.new_collection", // 目标集合完整名称
"dropTarget": false, // 目标集合存在时不覆盖
}
// 执行命令，结果存放在result中
var result bson.M
err = db.Run(renameCmd, &result)
if err != nil {
fmt.Printf("重命名集合失败：%vn", err)
return
}
fmt.Printf("重命名集合成功，返回结果：%vn", result)
}
注意事项
- 执行重命名命令的用户需要有对应数据库的管理权限，否则会返回权限不足的错误。
- 如果原集合不存在，命令会执行失败，返回集合不存在的错误，所以执行前可以先判断集合是否存在。
- 跨数据库重命名集合时，需要保证源数据库和目标数据库在同一个MongoDB实例上，mgo驱动不支持跨实例的集合重命名。
- 重命名操作会阻塞对应集合的读写操作，尽量在业务低峰期执行该操作，避免影响线上业务。
判断集合是否存在的方法
在执行重命名之前，可以先查询当前数据库的集合列表，判断原集合是否存在，避免执行失败：
func collectionExists(db *mgo.Database, collectionName string) bool {
// 获取数据库所有集合名称
collections, err := db.CollectionNames()
if err != nil {
return false
}
for _, name := range collections {
if name == collectionName {
return true
}
}
return false
}
可以在执行重命名命令前调用这个方法校验：
if !collectionExists(db, "old_collection") {
fmt.Println("原集合不存在，无法执行重命名操作")
return
}