Go语言mgo实践：优雅地忽略结构体字段不写入MongoDB

package main

import (
    "crypto/sha1"
    "encoding/base64"
    "fmt"
    "log"
    "time"

    "gopkg.in/mgo.v2" // 注意：mgo是较旧的驱动，推荐使用go.mongodb.org/mongo-driver
    "gopkg.in/mgo.v2/bson"
)

// Person 结构体定义
type Person struct {
    ID        bson.ObjectId `bson:"_id,omitempty"` // MongoDB文档ID
    Name      string        `bson:"name"`          // 存储姓名
    SSN       string        `bson:"-"`             // 使用bson:"-"标签，此字段将不被持久化
    HashedSSN string        `bson:"hashedSSN"`     // 存储SSN的哈希值
    CreatedAt time.Time     `bson:"createdAt"`     // 记录创建时间
}

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

    // 设置会话模式，确保读写一致性
    session.SetMode(mgo.Monotonic, true)

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

    // 准备一个Person实例
    bob := Person{
        ID:        bson.NewObjectId(),
        Name:      "Bob",
        SSN:       "fake_ssn_12345", // 这是一个不应被存储的敏感字段
        CreatedAt: time.Now(),
    }

    // 计算SSN的哈希值并赋值给HashedSSN
    hasher := sha1.New()
    hasher.Write([]byte(bob.SSN))
    bob.HashedSSN = base64.URLEncoding.EncodeToString(hasher.Sum(nil))

    fmt.Printf("准备插入的Person对象（Go内存中）： %+v\n", bob)

    // 插入文档
    err = c.Insert(bob)
    if err != nil {
        log.Fatalf("插入文档失败: %v", err)
    }
    fmt.Println("文档插入成功。")

    // 从数据库中查询该文档，验证SSN是否被忽略
    var result Person
    err = c.FindId(bob.ID).One(&result)
    if err != nil {
        log.Fatalf("查询文档失败: %v", err)
    }

    fmt.Printf("从MongoDB查询到的Person对象： %+v\n", result)

    // 验证SSN字段是否为空（因为它没有被持久化）
    if result.SSN == "" {
        fmt.Println("验证成功：SSN字段未被持久化到数据库中。")
    } else {
        fmt.Printf("验证失败：SSN字段意外地被持久化，值为: %s\n", result.SSN)
    }

    // 进一步验证，可以查询原始BSON数据
    var rawBSON map[string]interface{}
    err = c.FindId(bob.ID).One(&rawBSON)
    if err != nil {
        log.Fatalf("查询原始BSON失败: %v", err)
    }
    fmt.Printf("从MongoDB查询到的原始BSON数据： %+v\n", rawBSON)
    if _, ok := rawBSON["ssn"]; !ok {
        fmt.Println("验证成功：原始BSON数据中不包含'ssn'字段。")
    } else {
        fmt.Println("验证失败：原始BSON数据中包含'ssn'字段。")
    }
}