使用GO-FIBER与MongoDB Go Driver更新文档被覆盖的问题

当前使用Go-Fiber框架和MongoDB Go Driver实现用户更新接口时，希望仅修改请求体中传入的字段，但现有代码会将未传入的字段设为零值（如空字符串、0、nil）并覆盖原有文档数据。

问题原因 #

现有代码中，$set操作符里硬编码了所有User结构体字段，无论请求体是否传入对应值，未传入的字段会被赋值为结构体的零值，最终导致这些字段覆盖数据库中的原有数据。

解决方案 #

方法1：手动筛选非零值字段构建$set #

逐个检查请求体绑定后的结构体字段，仅将非零值的字段加入$set的更新文档中：

func UpdateOneUser(c *fiber.Ctx) error {
    params := c.Params("id")
    body := new(models.User)
    
    id, err := primitive.ObjectIDFromHex(params)
    if err != nil {
        return c.Status(500).SendString("invalid objectid")
    }

    if err := c.BodyParser(&body); err != nil {
        return c.Status(400).SendString("invalid body")
    }
    
    filter := bson.M{"_id": id}
    updateSet := bson.M{}

    // 仅添加非零值字段到更新集合
    if body.Name != "" {
        updateSet["name"] = body.Name
    }
    if body.Username != "" {
        updateSet["username"] = body.Username
    }
    if body.FirstName != "" {
        updateSet["first_name"] = body.FirstName
    }
    if body.LastName != "" {
        updateSet["last_name"] = body.LastName
    }
    if body.Email != "" {
        updateSet["email"] = body.Email
    }
    if body.PhoneNumber != "" {
        updateSet["phone_number"] = body.PhoneNumber
    }
    if body.Contry != "" {
        updateSet["contry"] = body.Contry
    }
    if body.Age != 0 { // 注意：如果年龄允许为0，需要调整判断逻辑
        updateSet["age"] = body.Age
    }
    if body.ChildAccounts != nil {
        updateSet["child_accounts"] = body.ChildAccounts
    }
    if body.Groups != nil {
        updateSet["groups"] = body.Groups
    }

    // 如果没有要更新的字段，直接返回
    if len(updateSet) == 0 {
        return c.Status(400).SendString("no fields to update")
    }

    update := bson.M{"$set": updateSet}
    
    result, err := db.User.UpdateOne(context.Background(), filter, update)
    if err != nil {
        return c.Status(500).SendString("user not found or update failed")
    }
    fmt.Println(result)

    return c.JSON(updateSet)
}

方法2：用反射自动收集非零值字段 #

当结构体字段较多时，手动筛选效率低，可通过反射遍历结构体字段，自动收集非零值字段：

import (
    "reflect"
    "go.mongodb.org/mongo-driver/bson/primitive"
)

func getNonZeroFields(obj interface{}) bson.M {
    result := bson.M{}
    val := reflect.ValueOf(obj).Elem()
    typ := val.Type()

    for i := 0; i < val.NumField(); i++ {
        field := val.Field(i)
        fieldName := typ.Field(i).Tag.Get("bson") // 假设结构体字段有bson标签
        if fieldName == "" {
            fieldName = typ.Field(i).Name // 如果没有bson标签，用结构体字段名
        }
        // 跳过空的bson标签（比如忽略字段）
        if fieldName == "-" {
            continue
        }

        // 判断字段是否为零值
        isZero := false
        switch field.Kind() {
        case reflect.String:
            isZero = field.String() == ""
        case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
            isZero = field.Int() == 0
        case reflect.Slice, reflect.Map:
            isZero = field.Len() == 0
        case reflect.Ptr:
            isZero = field.IsNil()
        // 可根据结构体字段类型扩展其他判断
        }

        if !isZero {
            result[fieldName] = field.Interface()
        }
    }
    return result
}

// 更新接口修改如下
func UpdateOneUser(c *fiber.Ctx) error {
    params := c.Params("id")
    body := new(models.User)
    
    id, err := primitive.ObjectIDFromHex(params)
    if err != nil {
        return c.Status(500).SendString("invalid objectid")
    }

    if err := c.BodyParser(&body); err != nil {
        return c.Status(400).SendString("invalid body")
    }
    
    filter := bson.M{"_id": id}
    updateSet := getNonZeroFields(body)

    if len(updateSet) == 0 {
        return c.Status(400).SendString("no fields to update")
    }

    update := bson.M{"$set": updateSet}
    
    result, err := db.User.UpdateOne(context.Background(), filter, update)
    if err != nil {
        return c.Status(500).SendString("user not found or update failed")
    }
    fmt.Println(result)

    return c.JSON(updateSet)
}

方法3：直接用bson.M接收请求体 #

跳过结构体绑定，直接用bson.M接收请求体，这样$set里就只有请求体传入的字段，不会包含零值字段：

func UpdateOneUser(c *fiber.Ctx) error {
    params := c.Params("id")
    var body bson.M
    
    id, err := primitive.ObjectIDFromHex(params)
    if err != nil {
        return c.Status(500).SendString("invalid objectid")
    }

    if err := c.BodyParser(&body); err != nil {
        return c.Status(400).SendString("invalid body")
    }
    
    // 可选：校验请求体中的字段是否为允许更新的字段，防止恶意字段更新
    allowedFields := map[string]bool{
        "name": true, "username": true, "first_name": true, 
        "last_name": true, "email": true, "phone_number": true,
        "contry": true, "age": true, "child_accounts": true, "groups": true,
    }
    for key := range body {
        if !allowedFields[key] {
            return c.Status(400).SendString("field " + key + " is not allowed to update")
        }
    }

    filter := bson.M{"_id": id}
    update := bson.M{"$set": body}
    
    result, err := db.User.UpdateOne(context.Background(), filter, update)
    if err != nil {
        return c.Status(500).SendString("user not found or update failed")
    }
    fmt.Println(result)

    return c.JSON(body)
}

注意事项 #

• 方法1需要注意字段的零值判断逻辑，比如如果age字段允许设置为0，就不能用body.Age != 0的判断，需要结合业务调整。
• 方法2的反射逻辑需要根据结构体的实际字段类型扩展判断，比如处理布尔值、浮点型等。
• 方法3必须添加字段校验，防止客户端传入不允许更新的字段（如_id）导致意外修改。

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