Golang?reflect反射的使用實例

更新時間：2023年05月11日 11:36:51 作者：常魚

Golang反射的錯誤大多數(shù)都來自于調(diào)用了一個不適合當前類型的方法，而且，這些錯誤通常是在運行時才會暴露出來，而不是在編譯時，如果我們傳遞的類型在反射代碼中沒有被覆蓋到那么很容易就會panic，本文就介紹一下使用go反射時很大概率會出現(xiàn)的錯誤，需要的可以參考一下

獲取字段名

首先獲取變量的Type變量

t := reflect.TypeOf(u)

需要注意的是，如果傳入的u是個指針，比如&User{"octoboy", 101}

if t.Kind() == reflect.Ptr {
	t = t.Elem()
}

這里通過Kind()函數(shù)獲取變量的類型，判斷如果類型為指針需要使用Elem()獲取指針指向的內(nèi)容。

然后遍歷結(jié)構(gòu)體的字段，獲取其字段名稱

	for i := 0; i < t.NumField(); i++ {
		fmt.Println(t.Field(i).Name)
	}

輸出結(jié)果：

UserName

UserId

獲取字段類型和值

v := reflect.ValueOf(u)
if v.Kind() == reflect.Ptr { //類型為指針 需要取elem
	v = v.Elem()
}

獲取字段的值或者賦值，需要用到ValueOf方法

for i := 0; i < v.NumField(); i++ {
	//v.Field(i).Int() v.Field(i).String() 都可以把值返回出來，相當于斷言 類型不匹配會直接panic
	//直接斷成interface 任意類型
	fmt.Println(v.Field(i).Interface())
}

輸出結(jié)果：

zyg

101

繼續(xù)輸出成員變量的類型

for i := 0; i < v.NumField(); i++ {
	fmt.Println(v.Field(i).Kind())
}

輸出結(jié)果：

string

int

設置字段值

靜態(tài)賦值

	//設置字段值
	va := reflect.ValueOf(&u) //這里必須使用指針 否則后面調(diào)用Set無法使用無地址的值
	if va.Kind() == reflect.Ptr { //類型為指針 需要取elem 意為取它指向的內(nèi)容值
		va = va.Elem()
	for i := 0; i < va.NumField(); i++ {
		//兩種方法取設置字段的值，第二種更為統(tǒng)一
		if va.Field(i).Kind() == reflect.String {
			//重要 如果需要使用set取修改u中的值，需要在ValueOf中傳入u的地址。否則會因為SetString使用了一個不能被尋址的值而造成panic
			va.Field(i).SetString("octoboy")
		}
		if va.Field(i).Kind() == reflect.Int {
			va.Field(i).Set(reflect.ValueOf(123))
		}
	}

interface切片映射成結(jié)構(gòu)體（動態(tài)賦值）

	//練手
	values := []interface{}{"octoboy", 123}
	for i := 0; i < va.NumField(); i++ {
		if reflect.ValueOf(values[i]).Kind() == va.Field(i).Kind() {
			va.Field(i).Set(reflect.ValueOf(values[i]))
		}
	}

打印以上兩種結(jié)構(gòu)題變量

輸出結(jié)果：

&{octoboy 123}

進階—map映射成結(jié)構(gòu)體

有如下代碼

	//練習 把map映射成struct
	set := map[string]interface{}{
		"UserName": "zyg",
		"UserId":   101,
		"Age":      19,
		"Sex":      1,
	}
	user := &User{}
	MapToStruct(set, user)
	fmt.Println(user)

要求將map映射到user結(jié)構(gòu)題中，即如果User的字段名如存在于map的key中，則將對應的value值賦給user結(jié)構(gòu)題的成員變量

有如下實現(xiàn)

//str類型為interface{} 代表可以傳入任意的結(jié)構(gòu)體
func MapToStruct(m map[string]interface{}, str interface{}) {
	val := reflect.ValueOf(str)
	if val.Kind() != reflect.Ptr {//必須是指針 否則無法用Set賦值
		panic(any("must be ptr!"))
	}
	val = val.Elem()
	if val.Kind() != reflect.Struct { //指針指向的必須是結(jié)構(gòu)體
		panic(any("must be struct"))
	}
	for i := 0; i < val.NumField(); i++ {
		name := val.Type().Field(i).Name //value轉(zhuǎn)type后取字段名稱
		if v, ok := m[name]; ok {        //如果根據(jù)tag做映射，就使用val.Type().Field(i).Tag.Get("name")作為key
			if reflect.ValueOf(v).Kind() == val.Field(i).Kind() {
				val.Field(i).Set(reflect.ValueOf(v))
			}
		}
	}
}