实际工作中可能会有这样的场景：
两个结构体(可能类型一样), 字段名和类型都一样, 想复制一个结构体的全部或者其中某几个字段的值到另一个(即merge操作),
自然想到可以用反射实现
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
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: %s\n", 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使用反射将一个结构体的数据直接复制到另一个结构体中（通过相同字段）
看代码吧~
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
package mainimport ( "fmt" "reflect")type A struct { Name string Gender string Age int}type B struct { Name string Gender 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).Name if 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)} |
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://gerrylon.blog.csdn.net/article/details/104262726
