Go语言mgo库：原子性检查插入操作结果的指南

package main

import (
    "fmt"
    "log"

    "gopkg.in/mgo.v2" // mgo v2
    "gopkg.in/mgo.v2/bson" // bson for mgo v2
)

// Person 结构体用于映射MongoDB中的文档
type Person struct {
    ID    bson.ObjectId `bson:"_id,omitempty"` // MongoDB的_id字段
    Name  string        `bson:"name"`
    Phone string        `bson:"phone"`
}

func main() {
    // 连接到MongoDB
    session, err := mgo.Dial("mongodb://localhost:27017/testdb")
    if err != nil {
        log.Fatalf("无法连接到MongoDB: %v", err)
    }
    defer session.Close() // 确保会话在程序结束时关闭

    // 1. 设置会话的安全模式
    // 这将确保mgo等待MongoDB服务器对写操作的确认。
    // 默认情况下，mgo.Safe{} 等价于 W: 1 (等待主节点确认)。
    session.SetSafe(&mgo.Safe{})
    fmt.Println("mgo会话已设置为安全模式。")

    // 获取数据库和集合
    c := session.DB("testdb").C("people")

    // ... 后续插入操作 ...
}

func main() {
    // ... (连接和设置安全模式的代码同上) ...

    c := session.DB("testdb").C("people")

    // 准备要插入的数据
    newPerson := &Person{
        ID:    bson.NewObjectId(), // 生成一个新的ObjectId
        Name:  "张三",
        Phone: "+86 13800138000",
    }

    // 2. 执行插入操作并检查错误
    err = c.Insert(newPerson)
    if err != nil {
        // 如果err不为nil，表示插入失败
        fmt.Printf("插入操作失败: %v\n", err)
        // 可以在这里根据错误类型进行更细致的处理
        if mgo.Is
        if mgo.Is
    } else {
        // 如果err为nil，表示插入成功
        fmt.Printf("成功插入新用户: %s (ID: %s)\n", newPerson.Name, newPerson.ID.Hex())
    }

    // 尝试插入一个可能导致错误的文档（例如，如果Name字段有唯一索引）
    // 假设我们有一个唯一索引在 Name 字段上
    duplicatePerson := &Person{
        ID:    bson.NewObjectId(),
        Name:  "张三", // 再次插入同名用户
        Phone: "+86 13912345678",
    }
    fmt.Println("\n尝试插入一个可能重复的文档...")
    err = c.Insert(duplicatePerson)
    if err != nil {
        fmt.Printf("插入重复文档失败: %v\n", err)
        // 检查是否是重复键错误
        if mgo.Is
        if mgo.Is
        if mgo.IsDup(err) {
            fmt.Println("这是一个重复键错误。")
        }
    } else {
        fmt.Printf("成功插入重复用户: %s (ID: %s)\n", duplicatePerson.Name, duplicatePerson.ID.Hex())
    }
}

package main

import (
    "fmt"
    "log"

    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson"
)

// Person 结构体用于映射MongoDB中的文档
type Person struct {
    ID    bson.ObjectId `bson:"_id,omitempty"` // MongoDB的_id字段
    Name  string        `bson:"name"`
    Phone string        `bson:"phone"`
}

func main() {
    // 连接到MongoDB
    // 请确保MongoDB服务在本地运行于27017端口
    session, err := mgo.Dial("mongodb://localhost:27017/testdb")
    if err != nil {
        log.Fatalf("无法连接到MongoDB: %v", err)
    }
    defer session.Close() // 确保会话在程序结束时关闭

    // 设置会话的安全模式
    // 这将确保mgo等待MongoDB服务器对写操作的确认。
    // 默认情况下，mgo.Safe{} 等价于 W: 1 (等待主节点确认)。
    session.SetSafe(&mgo.Safe{})
    fmt.Println("mgo会话已设置为安全模式。")

    // 获取数据库和集合
    dbName := "testdb"
    collectionName := "people"
    c := session.DB(dbName).C(collectionName)

    // 清理之前的测试数据（可选）
    _, err = c.RemoveAll(bson.M{"name": "张三"})
    if err != nil && err != mgo.ErrNotFound {
        fmt.Printf("清理数据失败: %v\n", err)
    } else {
        fmt.Println("已清理测试数据。")
    }

    // 准备要插入的数据
    newPerson := &Person{
        ID:    bson.NewObjectId(), // 生成一个新的ObjectId
        Name:  "张三",
        Phone: "+86 13800138000",
    }

    // 执行第一次插入操作并检查错误
    fmt.Println("\n尝试第一次插入...")
    err = c.Insert(newPerson)
    if err != nil {
        fmt.Printf("第一次插入操作失败: %v\n", err)
    } else {
        fmt.Printf("成功插入新用户: %s (ID: %s)\n", newPerson.Name, newPerson.ID.Hex())
    }

    // 尝试插入一个可能导致错误的文档
    // 假设我们在 'people' 集合的 'name' 字段上创建了一个唯一索引。
    // 可以通过 mongo shell 执行: db.people.createIndex({name: 1}, {unique: true})
    // 如果没有唯一索引，这个插入会成功。
    duplicatePerson := &Person{
        ID:    bson.NewObjectId(),
        Name:  "张三", // 再次插入同名用户
        Phone: "+86 13912345678",
    }
    fmt.Println("\n尝试第二次插入（同名用户，可能导致唯一索引冲突）...")
    err = c.Insert(duplicatePerson)
    if err != nil {
        fmt.Printf("第二次插入操作失败: %v\n", err)
        // 检查是否是重复键错误
        if mgo.IsDup(err) {
            fmt.Println("这是一个重复键错误（E11000 duplicate key error）。")
        }
    } else {
        fmt.Printf("成功插入第二个用户: %s (ID: %s)\n", duplicatePerson.Name, duplicatePerson.ID.Hex())
    }

    // 验证数据是否真的存在（可选，但用于演示）
    var result Person
    err = c.Find(bson.M{"name": "张三"}).One(&result)
    if err != nil {
        fmt.Printf("\n查询 '张三' 失败: %v\n", err)
    } else {
        fmt.Printf("\n通过查询验证，数据库中存在用户: %s (ID: %s)\n", result.Name, result.ID.Hex())
    }
}