Golang结构体字段自动赋值方法

package main

import (
    "fmt"
    "reflect"
)

type User struct {
    Name string `default:"匿名用户"`
    Age  int    `default:"18"`
    Email string `default:"no-email@example.com"`
}

func autoAssign(v interface{}) {
    rv := reflect.ValueOf(v)
    if rv.Kind() != reflect.Ptr || rv.Elem().Kind() != reflect.Struct {
        panic("必须传入结构体指针")
    }

    rv = rv.Elem() // 获取指针指向的结构体
    rt := rv.Type()

    for i := 0; i < rv.NumField(); i++ {
        field := rv.Field(i)
        tag := rt.Field(i).Tag.Get("default")

        if tag != "" && field.Interface() == reflect.Zero(field.Type()).Interface() {
            // 字段为空时，使用默认值
            switch field.Kind() {
            case reflect.String:
                field.SetString(tag)
            case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
                if val, err := strconv.Atoi(tag); err == nil {
                    field.SetInt(int64(val))
                }
            }
        }
    }
}

func main() {
    user := User{Name: "", Age: 0}
    autoAssign(&user)
    fmt.Printf("%+v\n", user) // 输出: {Name:匿名用户 Age:18 Email:no-email@example.com}
}

func fillFromMap(dst interface{}, data map[string]interface{}) {
    v := reflect.ValueOf(dst).Elem()
    t := v.Type()

    for i := 0; i < v.NumField(); i++ {
        field := v.Field(i)
        fieldType := t.Field(i)
        tag := fieldType.Tag.Get("json") // 使用json标签作为键名
        if tag == "" {
            tag = fieldType.Name // 没有标签则用字段名
        }

        if val, ok := data[tag]; ok && field.CanSet() {
            switch field.Kind() {
            case reflect.String:
                if s, ok := val.(string); ok {
                    field.SetString(s)
                }
            case reflect.Int:
                if n, ok := val.(int); ok {
                    field.SetInt(int64(n))
                }
            }
        }
    }
}

// 使用示例
user := User{}
data := map[string]interface{}{"Name": "Alice", "Age": 25}
fillFromMap(&user, data)
fmt.Printf("%+v\n", user) // {Name:Alice Age:25 Email:}