實(shí)際工作中可能會(huì)有這樣的場(chǎng)景:
兩個(gè)結(jié)構(gòu)體(可能類型一樣), 字段名和類型都一樣, 想復(fù)制一個(gè)結(jié)構(gòu)體的全部或者其中某幾個(gè)字段的值到另一個(gè)(即merge操作),
自然想到可以用反射實(shí)現(xiàn)
package main
import "fmt"
import "reflect"
// 用b的所有字段覆蓋a的
// 如果fields不為空, 表示用b的特定字段覆蓋a的
// a應(yīng)該為結(jié)構(gòu)體指針
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)
// 簡(jiǎn)單判斷下
if at.Kind() != reflect.Ptr {
err = fmt.Errorf("a must be a struct pointer")
return
}
av = reflect.ValueOf(av.Interface())
// 要復(fù)制哪些字段
_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
}
// 復(fù)制
for i := 0; i len(_fields); i++ {
name := _fields[i]
f := av.Elem().FieldByName(name)
bValue := bv.FieldByName(name)
// a中有同名的字段并且類型一致才復(fù)制
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字段值已經(jīng)成功被覆蓋.
而s2中Age字段和s1中Age字段類型不一樣, 會(huì)忽略.
其實(shí)上面的還可以優(yōu)化, 畢竟int32和int還是可以認(rèn)為是"一樣"的類型的,
不過思路就是這樣.
補(bǔ)充:golang使用反射將一個(gè)結(jié)構(gòu)體的數(shù)據(jù)直接復(fù)制到另一個(gè)結(jié)構(gòu)體中(通過相同字段)
看代碼吧~
package main
import (
"fmt"
"reflect"
)
type A struct {
Name string
Gender string
Age int
}
type B struct {
Name string
Gender string
}
//binding type interface 要修改的結(jié)構(gòu)體
//value type interace 有數(shù)據(jù)的結(jié)構(gòu)體
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++ {
// 在要修改的結(jié)構(gòu)體中查詢有數(shù)據(jù)結(jié)構(gòu)體中相同屬性的字段,有則修改其值
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)
}
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
您可能感興趣的文章:- golang修改結(jié)構(gòu)體中的切片值方法
- golang 結(jié)構(gòu)體初始化時(shí)賦值格式介紹
- golang結(jié)構(gòu)體與json格式串實(shí)例代碼
- Golang 利用反射對(duì)結(jié)構(gòu)體優(yōu)雅排序的操作方法