golang 实现两个结构体复制字段
短信预约 -IT技能 免费直播动态提醒
实际工作中可能会有这样的场景：
两个结构体(可能类型一样), 字段名和类型都一样, 想复制一个结构体的全部或者其中某几个字段的值到另一个(即merge操作),
自然想到可以用反射实现
package main
import "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使用反射将一个结构体的数据直接复制到另一个结构体中（通过相同字段）
看代码吧~
package main
import (
"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)
}
以上为个人经验，希望能给大家一个参考，也希望大家多多支持编程网。如有错误或未考虑完全的地方，望不吝赐教。
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
