Go中如何为结构体批量条件赋值？GraphQL resolver场景优化

针对你遇到的重复nil判断赋值冗余问题，这里提供几种实用的优化方案：

方案1：反射批量处理字段 #

利用Go的反射机制遍历输入结构体的字段，自动匹配并更新用户模型的对应字段，适合字段命名一致且数量较多的场景：

import "reflect"

func mergeUserUpdates(oldUser *model.User, input *model.ModifyUserInput) {
    oldVal := reflect.ValueOf(oldUser).Elem()
    inputVal := reflect.ValueOf(input).Elem()

    for i := 0; i < inputVal.NumField(); i++ {
        inputField := inputVal.Field(i)
        fieldName := inputVal.Type().Field(i).Name
        
        // 跳过ID字段和空指针字段
        if fieldName == "ID" || inputField.IsNil() {
            continue
        }

        // 定位到旧用户模型的对应字段并赋值
        oldField := oldVal.FieldByName(fieldName)
        if oldField.IsValid() && oldField.CanSet() {
            if inputField.Kind() == reflect.Ptr {
                oldField.Set(inputField.Elem())
            } else {
                oldField.Set(inputField)
            }
        }
    }
}

在Resolver中调用该函数即可简化代码：

func (r *mutationResolver) ModifyUser(ctx context.Context, input *model.ModifyUserInput) (*model.User, error) {
    id := input.ID
    us, ok := r.Resolver.UserStore[id]
    if !ok {
        return nil, fmt.Errorf("not found")
    }

    mergeUserUpdates(&us, input)

    r.Resolver.UserStore[id] = us
    return &us, nil
}

注意：反射会带来轻微性能开销，高并发场景需权衡；需保证ModifyUserInput与User的字段名完全匹配，否则需额外添加字段映射逻辑。

方案2：手动封装类型专属辅助函数（无反射，性能最优） #

如果不想引入反射，可以按字段类型封装小函数，减少重复代码：

func updateStringField(target *string, source *string) {
    if source != nil {
        *target = *source
    }
}

func updateBoolField(target *bool, source *bool) {
    if source != nil {
        *target = *source
    }
}

// 可根据字段类型扩展其他辅助函数，比如int、自定义指针类型等

Resolver中调用示例：

func (r *mutationResolver) ModifyUser(ctx context.Context, input *model.ModifyUserInput) (*model.User, error) {
    id := input.ID
    us, ok := r.Resolver.UserStore[id]
    if !ok {
        return nil, fmt.Errorf("not found")
    }

    updateStringField(&us.FirstName, input.FirstName)
    updateStringField(&us.LastName, input.LastName)
    updateStringField(&us.Password, input.Password)
    updateStringField(&us.Email, input.Email)
    
    if input.ProfileImage != nil {
        us.ProfileImage = input.ProfileImage
    }
    if input.InTomorrow != nil {
        us.InTomorrow = input.InTomorrow
    }
    if input.DefaultDaysIn != nil {
        us.DefaultDaysIn = input.DefaultDaysIn
    }

    r.Resolver.UserStore[id] = us
    return &us, nil
}

这种方式无性能损耗，代码可读性强，适合字段数量中等的场景。

方案3：借助第三方库实现结构体合并（进阶） #

可以用mapstructure库将输入结构体转为map，过滤空值后合并到用户模型，灵活性更高：

import "github.com/mitchellh/mapstructure"

func mergeUserUpdates(oldUser *model.User, input *model.ModifyUserInput) error {
    inputMap := make(map[string]interface{})
    if err := mapstructure.Decode(input, &inputMap); err != nil {
        return err
    }
    
    // 删除不需要更新的ID字段，过滤空值
    delete(inputMap, "ID")
    for k, v := range inputMap {
        if v == nil {
            delete(inputMap, k)
        }
    }
    
    // 将处理后的map合并到旧用户模型
    return mapstructure.Decode(inputMap, oldUser)
}

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