golang 實現(xiàn)兩個結(jié)構(gòu)體復(fù)制字段

更新時間：2021年04月28日 14:27:44 作者：butterfly5211314

這篇文章主要介紹了golang 實現(xiàn)兩個結(jié)構(gòu)體復(fù)制字段，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

實際工作中可能會有這樣的場景：

兩個結(jié)構(gòu)體(可能類型一樣), 字段名和類型都一樣, 想復(fù)制一個結(jié)構(gòu)體的全部或者其中某幾個字段的值到另一個(即merge操作),

自然想到可以用反射實現(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)
	// 簡單判斷下
	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字段類型不一樣, 會忽略.

其實上面的還可以優(yōu)化, 畢竟int32和int還是可以認(rèn)為是"一樣"的類型的,

不過思路就是這樣.

補充：golang使用反射將一個結(jié)構(gòu)體的數(shù)據(jù)直接復(fù)制到另一個結(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)
}

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章: