mgo - Golang MongoDB 驱动使用指南

mgo 是 Go 语言的一个 MongoDB 驱动，提供了简单而强大的 MongoDB 操作接口。虽然目前官方推荐使用新的官方驱动 go.mongodb.org/mongo-driver，但 mgo 仍然被广泛使用。

安装 mgo

go get gopkg.in/mgo.v2

基本用法

1. 连接 MongoDB

package main

import (
	"fmt"
	"gopkg.in/mgo.v2"
	"log"
)

func main() {
	// 创建连接
	session, err := mgo.Dial("mongodb://localhost:27017")
	if err != nil {
		log.Fatal(err)
	}
	defer session.Close()

	// 检查连接是否正常
	err = session.Ping()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("Connected to MongoDB!")
}

2. 选择数据库和集合

// 获取数据库
db := session.DB("testdb")

// 获取集合
collection := db.C("users")

// 或者直接通过session获取集合
collection := session.DB("testdb").C("users")

3. 插入文档

type User struct {
	Name  string
	Age   int
	Email string
}

user := User{
	Name:  "张三",
	Age:   30,
	Email: "zhangsan@example.com",
}

err = collection.Insert(&user)
if err != nil {
	log.Fatal(err)
}
fmt.Println("Document inserted!")

4. 查询文档

// 查询单个文档
var result User
err = collection.Find(bson.M{"name": "张三"}).One(&result)
if err != nil {
	log.Fatal(err)
}
fmt.Printf("Found document: %+v\n", result)

// 查询多个文档
var users []User
err = collection.Find(bson.M{"age": bson.M{"$gt": 25}}).All(&users)
if err != nil {
	log.Fatal(err)
}
fmt.Printf("Found %d documents\n", len(users))
for _, u := range users {
	fmt.Printf("%+v\n", u)
}

5. 更新文档

// 更新单个文档
err = collection.Update(
	bson.M{"name": "张三"},
	bson.M{"$set": bson.M{"age": 31}},
)
if err != nil {
	log.Fatal(err)
}

// 更新多个文档
_, err = collection.UpdateAll(
	bson.M{"age": bson.M{"$gt": 25}},
	bson.M{"$inc": bson.M{"age": 1}},
)
if err != nil {
	log.Fatal(err)
}

6. 删除文档

// 删除单个文档
err = collection.Remove(bson.M{"name": "张三"})
if err != nil {
	log.Fatal(err)
}

// 删除多个文档
_, err = collection.RemoveAll(bson.M{"age": bson.M{"$gt": 30}})
if err != nil {
	log.Fatal(err)
}

7. 索引操作

// 创建索引
index := mgo.Index{
	Key:        []string{"name", "email"},
	Unique:     true,
	DropDups:   true,
	Background: true,
	Sparse:     true,
}

err = collection.EnsureIndex(index)
if err != nil {
	log.Fatal(err)
}

// 删除索引
err = collection.DropIndex("name", "email")
if err != nil {
	log.Fatal(err)
}

高级特性

1. 批量操作

bulk := collection.Bulk()
bulk.Insert(User{Name: "李四", Age: 28, Email: "lisi@example.com"})
bulk.Insert(User{Name: "王五", Age: 35, Email: "wangwu@example.com"})
bulk.Update(bson.M{"name": "张三"}, bson.M{"$set": bson.M{"age": 32}})
_, err = bulk.Run()
if err != nil {
	log.Fatal(err)
}

2. 聚合操作

pipe := collection.Pipe([]bson.M{
	{"$match": bson.M{"age": bson.M{"$gt": 25}}},
	{"$group": bson.M{
		"_id":   "$age",
		"count": bson.M{"$sum": 1},
	}},
})

var results []bson.M
err = pipe.All(&results)
if err != nil {
	log.Fatal(err)
}
for _, res := range results {
	fmt.Printf("Age %v has %v users\n", res["_id"], res["count"])
}

3. 事务支持

// 注意：mgo本身不支持多文档事务，这是MongoDB 4.0+的特性
// 可以通过以下方式模拟简单事务

// 开始"事务"
session.SetSafe(&mgo.Safe{})

// 执行操作
err = collection.Insert(&User{Name: "赵六", Age: 40})
if err != nil {
	session.ResetSafe() // 回滚
	log.Fatal(err)
}

// 提交"事务"
session.SetSafe(nil)

性能优化

1. 连接池管理:

   session.SetPoolLimit(100) // 设置连接池大小
   session.SetMode(mgo.Monotonic, true) // 设置一致性模式
   

2. 批量查询优化:

   query := collection.Find(bson.M{"age": bson.M{"$gt": 25}})
   query.Batch(100) // 设置批量大小
   query.Prefetch(0.25) // 设置预取比例
   

3. 选择字段:

   err = collection.Find(bson.M{"name": "张三"}).Select(bson.M{"name": 1, "email": 1}).One(&result)
   

注意事项

1. mgo 已经不再积极维护，新项目建议使用官方驱动 go.mongodb.org/mongo-driver
2. 确保正确处理错误和关闭会话
3. 对于生产环境，考虑添加连接超时和重试逻辑
4. 注意BSON和Go类型的映射关系

mgo 提供了简单直观的API，适合快速开发MongoDB应用。虽然它不再是官方推荐的选择，但在许多现有项目中仍然广泛使用。