go给结构体成员变量通过db标签动态赋值

如果要通过结构体标签的值来指定字段名,你可以修改 SetFieldValue 函数,使其接受一个标签名而不是字段名。以下是相应的修改:

package main

import (
	"fmt"
	"reflect"
)

// 定义示例的结构体类型
type Person struct {
	ID   int    `db:"id"`
	Name string `db:"name"`
	Age  int    `db:"age"`
}

// CreateNewPerson 根据结构体类型创建一个新的结构体实例
func CreateNewPerson() interface{} {
	// 获取结构体类型
	personType := reflect.TypeOf(Person{})
	// 使用反射创建结构体实例
	newPerson := reflect.New(personType).Interface()
	return newPerson
}

// SetFieldValueByTag 设置结构体字段的值,通过结构体标签名
func SetFieldValueByTag(structPtr interface{}, tagValue string, value interface{}) error {
	// 获取结构体指针的 reflect.Value
	val := reflect.ValueOf(structPtr)
	// 检查是否为指针类型和结构体类型
	if val.Kind() != reflect.Ptr || val.Elem().Kind() != reflect.Struct {
		return fmt.Errorf("invalid struct pointer")
	}

	// 获取结构体的 reflect.Value
	val = val.Elem()

	// 遍历结构体的字段
	for i := 0; i < val.NumField(); i++ {
		// 获取结构体字段的 reflect.StructField
		field := val.Type().Field(i)

		// 获取结构体字段的标签值
		tag := field.Tag.Get("db")

		// 检查标签值是否等于指定值
		if tag == tagValue {
			// 获取结构体字段的 reflect.Value
			fieldVal := val.Field(i)

			// 设置字段值
			if fieldVal.CanSet() {
				fieldVal.Set(reflect.ValueOf(value))
				return nil
			}

			return fmt.Errorf("cannot set value for field %s", field.Name)
		}
	}

	return fmt.Errorf("field with tag %s not found", tagValue)
}

func main() {
	// 使用反射创建一个新的 Person 对象
	newPerson := CreateNewPerson()

	// 使用反射给每个成员变量赋值,通过结构体标签名
	err := SetFieldValueByTag(newPerson, "name", "John Doe")
	if err != nil {
		fmt.Println(err)
	}

	err = SetFieldValueByTag(newPerson, "age", 25)
	if err != nil {
		fmt.Println(err)
	}

	// 将结果转换为 Person 类型
	person, ok := newPerson.(*Person)
	if ok {
		// 输出新创建的 Person 对象
		fmt.Printf("%+v\n", person)
	} else {
		fmt.Println("Failed to convert to *Person")
	}
}

在这个修改后的示例中,SetFieldValueByTag 函数接受一个结构体指针、标签名和值作为参数。该函数遍历结构体的字段,找到与指定标签值匹配的字段,然后设置字段的值。在 main 函数中,我们通过结构体标签名调用了 SetFieldValueByTag 函数,以达到通过标签值来设置字段值的目的。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值