golang 實現兩個結構體復制字段
實際工作中可能會有這樣的場景:
兩個結構體(可能類型一樣), 字段名和類型都一樣, 想復制一個結構體的全部或者其中某幾個字段的值到另一個(即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) }
以上為個人經驗,希望能給大傢一個參考,也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方,望不吝賜教。
推薦閱讀:
- Go語言基礎反射示例詳解
- Golang反射獲取變量類型和值的方法詳解
- Go利用反射reflect實現獲取接口變量信息
- go語言通過反射創建結構體、賦值、並調用對應的操作
- Golang 中反射的應用實例詳解