您的位置:首页技术文章

golang 实现两个结构体复制字段

【字号: 日期:2023-08-25 16:45:28浏览:43作者:馨心

实际工作中可能会有这样的场景:

两个结构体(可能类型一样), 字段名和类型都一样, 想复制一个结构体的全部或者其中某几个字段的值到另一个(即merge操作),

自然想到可以用反射实现

package mainimport 'fmt'import 'reflect'// 用b的所有字段覆盖a的// 如果fields不为空, 表示用b的特定字段覆盖a的// a应该为结构体指针func CopyFields(a interface{}, b interface{}, fields ...string) (err error) {at := reflect.TypeOf(a)av := reflect.ValueOf(a)bt := reflect.TypeOf(b)bv := reflect.ValueOf(b)// 简单判断下if at.Kind() != reflect.Ptr {err = fmt.Errorf('a must be a struct pointer')return}av = reflect.ValueOf(av.Interface())// 要复制哪些字段_fields := make([]string, 0)if len(fields) > 0 {_fields = fields} else {for i := 0; i < bv.NumField(); i++ {_fields = append(_fields, bt.Field(i).Name)}}if len(_fields) == 0 {fmt.Println('no fields to copy')return}// 复制for i := 0; i < len(_fields); i++ {name := _fields[i]f := av.Elem().FieldByName(name)bValue := bv.FieldByName(name)// a中有同名的字段并且类型一致才复制if f.IsValid() && f.Kind() == bValue.Kind() {f.Set(bValue)} else {fmt.Printf('no such field or different kind, fieldName: %sn', name)}}return}type S1 struct { Name string Age int}type S2 struct { Name string Age int32}func main() { s1 := S1{'hello', 22} s2 := S2{'world', 33} fmt.Println(s1, s2) CopyFields(&s1, s2) fmt.Println(s1, s2)}上述例子输出为:

{hello 22} {world 33}

no such field or different kind, fieldName: Age

{world 22} {world 33}

可见s2的Name字段值已经成功被覆盖.

而s2中Age字段和s1中Age字段类型不一样, 会忽略.

其实上面的还可以优化, 毕竟int32和int还是可以认为是'一样'的类型的,

不过思路就是这样.

补充:golang使用反射将一个结构体的数据直接复制到另一个结构体中(通过相同字段)

看代码吧~

package mainimport ('fmt''reflect')type A struct {Name stringGender stringAge int}type B struct {Name stringGender string}//binding type interface 要修改的结构体//value type interace 有数据的结构体func structAssign(binding interface{}, value interface{}) {bVal := reflect.ValueOf(binding).Elem() //获取reflect.Type类型vVal := reflect.ValueOf(value).Elem() //获取reflect.Type类型vTypeOfT := vVal.Type()for i := 0; i < vVal.NumField(); i++ {// 在要修改的结构体中查询有数据结构体中相同属性的字段,有则修改其值name := vTypeOfT.Field(i).Nameif ok := bVal.FieldByName(name).IsValid(); ok {bVal.FieldByName(name).Set(reflect.ValueOf(vVal.Field(i).Interface()))}}}func main() {as := A{}bs := B{Name: 'wfy', Gender: '男'}fmt.Println(as)structAssign(&as, &bs)fmt.Println(as)}

以上为个人经验,希望能给大家一个参考,也希望大家多多支持优爱好网。如有错误或未考虑完全的地方,望不吝赐教。

标签: Golang
相关文章: