Go语言MongoDB数组字段重复值校验与更新报错排查

核心问题分析 #

1. 初始代码逻辑颠倒：当FindOne返回mongo.ErrNoDocuments时，说明目标状态不存在，这正是可以插入的时机，但你却返回了错误；反之如果查到文档，说明状态已存在，应该阻止插入。
2. UpdateOne代码语法错误：$set的值错误嵌套了$set键，导致MongoDB解析报错。

正确实现方案（原子操作，推荐） #

使用MongoDB的原子更新操作，在一个请求中完成「状态存在校验」和「更新操作」，避免并发场景下的重复插入问题：

obId, err := primitive.ObjectIDFromHex(id)
if err != nil {
    return nil, err
}

// 构造要添加的新状态对象
newStatus := bson.M{
    "status":    string(p.Status),
    "message":   p.Message, // 假设p包含message字段，根据实际结构调整
    "createdAt": time.Now().UnixMilli(),
    "updatedAt": nil,
}

// 1. 过滤条件：匹配目标文档，且statusData中不存在该status
filter := bson.M{
    "_id":               obId,
    "statusData.status": bson.M{"$ne": string(p.Status)},
}

// 2. 生成需更新的其他字段文档
doc, err := utils.ToDoc(p)
if err != nil {
    return nil, err
}

// 3. 构造更新操作：$push添加新状态到数组，$set更新其他字段
update := bson.M{
    "$push": bson.M{"statusData": newStatus},
    "$set":  doc, // 直接使用ToDoc生成的bson.D，无需额外嵌套
}

// 执行更新
result, err := r.collection.UpdateOne(ctx, filter, update)
if err != nil {
    return nil, err
}

// 根据结果判断操作状态
if result.MatchedCount == 0 {
    // 状态已存在，无需插入
    return nil, errors.New("状态已存在")
} else if result.ModifiedCount == 1 {
    // 状态添加成功，返回更新后的文档
    var updatedDoc bson.M
    err := r.collection.FindOne(ctx, bson.M{"_id": obId}).Decode(&updatedDoc)
    if err != nil {
        return nil, err
    }
    return updatedDoc, nil
}
return nil, nil

错误代码修正说明 #

1. 初始代码的逻辑修正 #

如果坚持使用「先查询后更新」的方式，需要颠倒错误判断逻辑：

obId, err := primitive.ObjectIDFromHex(id)
if err != nil {
    return nil, err
}

// 查询是否存在该状态
var result bson.M
er := r.collection.FindOne(ctx, bson.M{"_id": obId, "statusData.status": string(p.Status)}).Decode(&result)
if er == nil {
    // 查到文档，说明状态已存在
    return nil, errors.New("状态已存在")
}
if er != mongo.ErrNoDocuments {
    // 其他查询错误
    return nil, er
}

// 状态不存在，执行更新
newStatus := bson.M{
    "status":    string(p.Status),
    "message":   p.Message,
    "createdAt": time.Now().UnixMilli(),
    "updatedAt": nil,
}
doc, err := utils.ToDoc(p)
if err != nil {
    return nil, err
}

// 注意：不能用$set覆盖整个文档，需分别处理数组追加和字段更新
update := bson.M{
    "$push": bson.M{"statusData": newStatus},
    "$set":  doc,
}

res := r.collection.FindOneAndUpdate(ctx, bson.M{"_id": obId}, update, options.FindOneAndUpdate().SetReturnDocument(1))
var updatedDoc bson.M
err = res.Decode(&updatedDoc)
if err != nil {
    return nil, err
}
return updatedDoc, nil

2. UpdateOne错误的语法修正 #

你之前的update中$set部分错误嵌套了$set键：

// 错误写法
"$set": bson.D{{Key: "$set", Value: doc}}

// 正确写法
"$set": doc

因为utils.ToDoc(p)已经生成了包含文档字段的bson.D，直接作为$set的值即可，无需再包裹一层$set键，这就是导致报错的根本原因。

额外注意事项 #

• 原子操作优先：「先查询后更新」存在并发风险，如果两个请求同时查询到状态不存在，会导致重复插入，而原子更新操作可以避免这个问题。
• 状态唯一性校验：如果需要严格保证status的唯一性，可以考虑在statusData.status字段上创建多键索引并设置唯一约束，但要注意数组字段的唯一索引会限制整个集合中所有文档的statusData数组里的status值都不重复，需根据业务需求选择。

内容的提问来源于stack exchange，提问作者King