如何用Go动态设置MongoDB文档字段，将用户消息存入单文档

要实现把同一用户的所有消息存入单个文档，且以消息时间为字段名、消息文本为值，你需要使用MongoDB的更新操作（而非插入），通过UpdateOne配合$set操作符和upsert选项来完成。

核心思路 #

• 用用户唯一标识（比如userId）作为文档的_id，确保同一用户的消息都存入这个文档。
• 将time.Time转换为标准字符串（如RFC3339格式）作为字段名，避免特殊字符导致MongoDB报错。
• 使用$set追加新的消息字段，同时开启upsert: true，如果用户文档不存在则自动创建。

代码实现 #

import (
    "context"
    "log"
    "time"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo/options"
)

// 保留你定义的消息结构体
type MsgStr struct {
    Id   string    `bson:"_id,omitempty"`
    Time time.Time `bson:"time,omitempty"`
    Data string    `bson:"text,omitempty"`
}

func saveUserMsg(ctx context.Context, userId string, msg MsgStr) {
    // 将时间转换为RFC3339格式的字符串作为字段名
    timeField := msg.Time.Format(time.RFC3339)
    
    // 构造更新指令：用$set添加新的消息字段
    update := bson.M{
        "$set": bson.M{
            timeField: msg.Data,
        },
    }
    
    // 执行更新/插入操作
    updateResult, err := podcastsCollection.UpdateOne(
        ctx,
        bson.M{"_id": userId}, // 查询条件：匹配用户ID
        update,
        options.Update().SetUpsert(true), // 不存在则插入新文档
    )
    if err != nil {
        log.Fatalf("保存消息失败: %v", err)
    }
    
    if updateResult.UpsertedID != nil {
        log.Println("为用户创建了新的消息文档")
    } else {
        log.Println("已将消息追加到用户现有文档")
    }
}

说明 #

• 这种方式会把同一个用户的所有消息都存储在_id为userId的单个文档中，每个消息对应一个以时间字符串为键的字段。
• 彻底解决了原InsertOne方式每条消息生成一个文档的问题。
• RFC3339格式的时间字符串（如2024-05-20T15:04:05Z）是MongoDB字段名的安全选择，无歧义且无特殊字符。

内容的提问来源于stack exchange，提问作者Антон Егоров