Go语言mgo库：如何可靠地验证MongoDB插入操作的成功与否

session.SetSafe(&mgo.Safe{}) // 启用安全模式

err = c.Insert(&newPerson) // 执行插入操作

if err != nil {
    fmt.Printf("插入失败: %v\n", err)
} else {
    fmt.Print("插入成功!")
}

package main

import (
    "fmt"
    "log"
    "time" // 导入time包，可能在实际应用中用于设置超时或日志

    "gopkg.in/mgo.v2" // 使用mgo v2版本
    "gopkg.in/mgo.v2/bson"
)

// Person 结构体定义，用于映射MongoDB文档
type Person struct {
    ID    bson.ObjectId `bson:"_id,omitempty"` // MongoDB的_id字段，omitempty表示如果为空则不插入
    Name  string        `bson:"name"`
    Phone string        `bson:"phone"`
}

func main() {
    // 1. 连接MongoDB
    // 注意：mgo库已被官方弃用，建议使用 go.mongodb.org/mongo-driver
    // 此处为兼容旧代码或特定需求而使用mgo
    session, err := mgo.Dial("mongodb://localhost:27017")
    if err != nil {
        log.Fatalf("无法连接到MongoDB: %v", err)
    }
    defer session.Close() // 确保会话在函数结束时关闭

    // 2. 设置连接池模式 (可选，但推荐)
    // Monotonic模式确保在单个请求中，所有操作都使用同一个连接
    session.SetMode(mgo.Monotonic, true)

    // !!! 3. 关键步骤：设置会话为安全模式 !!!
    // 这将确保所有写入操作（如Insert）都会等待MongoDB服务器的写入确认。
    // 如果没有此设置，Insert可能在数据实际写入前就返回nil错误。
    session.SetSafe(&mgo.Safe{})

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

    // 清理旧数据，确保每次运行示例时环境干净（可选）
    // _, err = c.RemoveAll(nil)
    // if err != nil && err != mgo.ErrNotFound {
    //  log.Printf("清理数据失败: %v", err)
    // }

    // 5. 准备待插入的数据
    newPerson := Person{
        Name:  "Ale",
        Phone: "+55 53 8116 9639",
    }

    // 6. 执行插入操作并检查结果
    fmt.Printf("尝试插入用户: %s\n", newPerson.Name)
    err = c.Insert(&newPerson) // 执行插入操作

    if err != nil {
        fmt.Printf("插入失败: %v\n", err)
    } else {
        // 插入成功后，如果ID字段是bson.ObjectId类型且为omitempty，mgo会自动填充生成的ID
        fmt.Printf("插入成功！新用户ID: %s (Hex: %s)\n", newPerson.ID.String(), newPerson.ID.Hex())
    }

    // 演示插入失败的情况（例如，如果集合有唯一索引，且尝试插入重复数据）
    // 为此演示，我们假设没有唯一索引，仅展示插入成功后的正常流程。
    // 如果您想测试失败，可以手动创建一个唯一索引在MongoDB中：
    // db.people.createIndex({ "name": 1 }, { unique: true })
    // 然后尝试插入一个同名的Person。

    // 再次尝试插入一个不同的用户
    fmt.Println("\n尝试插入第二个用户...")
    anotherPerson := Person{
        Name:  "Bob",
        Phone: "+1 234 567 8900",
    }
    err = c.Insert(&anotherPerson)
    if err != nil {
        fmt.Printf("第二个用户插入失败: %v\n", err)
    } else {
        fmt.Printf("第二个用户插入成功！新用户ID: %s (Hex: %s)\n", anotherPerson.ID.String(), anotherPerson.ID.Hex())
    }

    // 7. 查询以验证数据（可选，但用于演示）
    fmt.Println("\n验证数据...")
    var people []Person
    err = c.Find(nil).All(&people) // 查询所有文档
    if err != nil {
        log.Fatalf("查询数据失败: %v", err)
    }
    fmt.Println("当前集合中的用户:")
    for _, p := range people {
        fmt.Printf("  ID: %s, Name: %s, Phone: %s\n", p.ID.Hex(), p.Name, p.Phone)
    }
}