Go语言mgo库插入操作结果验证与错误处理

package main

import (
    "fmt"
    "log"
    "time"

    "gopkg.in/mgo.v2" // 注意：mgo v2 是较旧版本，生产环境推荐使用官方go.mongodb.org/mongo-driver
    "gopkg.in/mgo.v2/bson"
)

// Person 结构体定义
type Person struct {
    ID    bson.ObjectId `bson:"_id,omitempty"` // 自动生成或指定ID
    Name  string        `bson:"name"`
    Phone string        `bson:"phone"`
}

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

    // 设置会话模式：Master表示读写都优先主节点，Strong表示读操作将返回最新数据
    session.SetMode(mgo.Monotonic, true)

    // 2. 启用会话安全模式
    // 这是关键一步！它告诉mgo等待MongoDB服务器的写入确认。
    // 默认的mgo.Safe{}等同于设置W=1，即等待主节点确认写入。
    session.SetSafe(&mgo.Safe{}) // <-- 启用安全模式

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

    // 3. 准备要插入的数据
    person1 := &Person{
        Name:  "Alice",
        Phone: "+1 123 456 7890",
    }

    person2 := &Person{
        Name:  "Bob",
        Phone: "+1 987 654 3210",
    }

    // 4. 执行插入操作并检查错误
    fmt.Println("尝试插入 Person 1...")
    err = c.Insert(person1)
    if err != nil {
        fmt.Printf("插入 Person 1 失败: %v\n", err)
    } else {
        fmt.Printf("插入 Person 1 成功，ID: %s\n", person1.ID.Hex())
    }

    fmt.Println("\n尝试插入 Person 2...")
    err = c.Insert(person2)
    if err != nil {
        fmt.Printf("插入 Person 2 失败: %v\n", err)
    } else {
        fmt.Printf("插入 Person 2 成功，ID: %s\n", person2.ID.Hex())
    }

    // 模拟一个可能失败的插入（例如，如果集合有唯一索引，插入重复数据）
    // 为了演示，这里假设没有唯一索引，如果实际有，第二次插入同名数据会失败
    // person3 := &Person{
    //  Name:  "Alice", // 假设Name是唯一索引
    //  Phone: "+1 111 222 3333",
    // }
    // fmt.Println("\n尝试插入 Person 3 (可能失败)...")
    // err = c.Insert(person3)
    // if err != nil {
    //  fmt.Printf("插入 Person 3 失败: %v\n", err)
    // } else {
    //  fmt.Printf("插入 Person 3 成功，ID: %s\n", person3.ID.Hex())
    // }

    // 验证插入结果（可选，仅为演示，实际场景中通过Insert的错误判断即可）
    count, err := c.Find(bson.M{"name": "Alice"}).Count()
    if err != nil {
        log.Printf("查询失败: %v", err)
    } else {
        fmt.Printf("\n数据库中名为 'Alice' 的记录数: %d\n", count)
    }

    // 清理数据（可选）
    // _, err = c.RemoveAll(bson.M{})
    // if err != nil {
    //  log.Printf("清理数据失败: %v", err)
    // }
}