Golang反射修改變量值的操作代碼

更新時(shí)間：2022年12月06日 08:36:10 作者：Amos01

這篇文章主要介紹了Golang反射修改變量值,也就是Golang反射三大定律中的前兩個(gè)，即從interface{}到反射對(duì)象和從反射對(duì)象到interface{}，需要的朋友可以參考下

1. 前言

前面的隨筆Golang 利用反射對(duì)結(jié)構(gòu)體優(yōu)雅排序的操作方法分享了如何通過(guò)反射獲取變量的類型和值，

也就是Golang反射三大定律中的前兩個(gè)，即從interface{}到反射對(duì)象和從反射對(duì)象到interface{}。

這篇隨筆主要分享通過(guò)反射修改各種類型變量值的方法。

2. 判斷是否可修改

reflect提供func (v Value) CanSet() bool判斷對(duì)象值是否修改。

一般情況下，通過(guò)反射修改變量值，需要滿足以下兩個(gè)條件。

2.1 該值是可尋址的

類似函數(shù)傳參，如果需要在函數(shù)內(nèi)修改入?yún)?shù)的內(nèi)容，那么就需要傳引用，而不是傳值。

函數(shù)內(nèi)修改入?yún)⒅赶虻膬?nèi)容，才能將修改效果“帶出”該函數(shù)的作用域。

同理，反射修改變量的值，應(yīng)當(dāng)是可以尋址的，修改的是反射對(duì)象指向的數(shù)據(jù)內(nèi)容，

因此，通過(guò)反射函數(shù)func ValueOf(i any) Value

入?yún)?code>i是引用時(shí)，i指向的內(nèi)容可尋址，因此返回參數(shù)Value不可修改，Value.Elem可修改。
入?yún)?code>i是地址時(shí)，返回參數(shù)Value可修改。
入?yún)?code>i是引用地址時(shí)，返回參數(shù)Value及Value.Elem均可修改。

上述三種情況如下圖所示，經(jīng)過(guò)尋址的內(nèi)容才有可能是可修改的。

2.2 該值是可導(dǎo)出的

這個(gè)主要是針對(duì)結(jié)構(gòu)體的成員，該成員的字段名的首字母需要是大寫，即是“public”的。

3. 修改slice

slice是引用類型，slice的數(shù)據(jù)結(jié)構(gòu)如下圖所示，通過(guò)反射可以修改slice指向的內(nèi)容。

修改指定下標(biāo)的數(shù)據(jù)內(nèi)容，并且數(shù)據(jù)類型需要和修改前一只，否則會(huì)panic。

func main() {
	s := []int{1, 2, 3}
	valueS := reflect.ValueOf(s)
	// slice 是否可修改 不可整體修改
	fmt.Printf("valueS Kind:%v CanSet:%v Index(0).CanSet:%v\n", valueS.Kind(), valueS.CanSet(), valueS.Index(0).CanSet())
	// 修改指定下標(biāo)的元素值
	valueS.Index(0).Set(reflect.ValueOf(10))
	valueS.Index(1).SetInt(20)
	fmt.Printf("after edit:%v\n", s)
 
	// panic: reflect: call of reflect.Value.SetFloat on int Value
	//valueS.Index(1).SetFloat(100)
}

代碼輸出如下

$ go run main.go
valueS Kind:slice CanSet:false Index(0).CanSet:true
after edit:[10 20 3]

如果需要整體修改修改slice，那么需要傳入slice的地址

func main() {
	s := []int{1, 2, 3}
	// slice的指針
	valuePtrS := reflect.ValueOf(&s)
	fmt.Printf("valuePtrS kind:%v CanSet:%v\n", valuePtrS.Kind(), valuePtrS.CanSet())
	// 獲取指針指向的內(nèi)容
	valueS := valuePtrS.Elem()
	fmt.Printf("valueS kind:%v CanSet:%v\n", valueS.Kind(), valueS.CanSet())
	// 整體修改slice
	valueS.Set(reflect.ValueOf([]int{4, 5, 6, 7}))
	fmt.Printf("replace edit:%v\n", s)
}

代碼輸出如下

$ go run main.go
valuePtrS kind:ptr CanSet:false
valueS kind:slice CanSet:true
replace edit:[4 5 6 7]

4. 修改array

array不是引用類型，因此func ValueOf(i any) Value需要傳入array的地址。

func main() {
	s := [3]int{1, 2, 3}
	// array的指針
	valuePtrS := reflect.ValueOf(&s)
	fmt.Printf("valuePtrS kind:%v CanSet:%v\n", valuePtrS.Kind(), valuePtrS.CanSet())
	// 獲取指針指向的內(nèi)容
	valueS := valuePtrS.Elem()
	fmt.Printf("valueS kind:%v CanSet:%v\n", valueS.Kind(), valueS.CanSet())
	// 修改指定下標(biāo)數(shù)據(jù)
	valueS.Index(0).SetInt(10)
	fmt.Printf("after edit:%v\n", s)
	// 整體修改slice
	valueS.Set(reflect.ValueOf([3]int{4, 5, 6}))
	fmt.Printf("replace edit:%v\n", s)
 
	//panic: reflect.Set: value of type [4]int is not assignable to type [3]int
	//valueS.Set(reflect.ValueOf([4]int{4, 5, 6}))
}

代碼輸出如下

$ go run main.go
valuePtrS kind:ptr CanSet:false
valueS kind:array CanSet:true
after edit:[10 2 3]
replace edit:[4 5 6]

5. 修改結(jié)構(gòu)體

帶修改的結(jié)構(gòu)體的成員的字段名首字母需要大寫。

func main() {
	type myStruct struct {
		Num  int    `json:"num_json" orm:"column:num_orm"`
		Desc string `json:"desc_json" orm:"column:desc_orm"`
	}
	s := myStruct{
		Num:  1,
		Desc: "desc",
	}
	valueS := reflect.ValueOf(&s)
	// 指針本身不可修改 可指向的內(nèi)容
	fmt.Printf("Kind:%v CanSet:%v\n", valueS.Kind(), valueS.CanSet())
	// 獲取指針指向的內(nèi)容
	valueS = valueS.Elem()
	fmt.Printf("Kind:%v CanSet:%v Field(0).CanSet:%v\n", valueS.Kind(), valueS.CanSet(), valueS.Field(0).CanSet())
	// 修改指定成員的值
	valueS.Field(0).SetInt(10)
	fmt.Printf("after edit:%+v\n", s)
	// 替換整體內(nèi)容
	valueS.Set(reflect.ValueOf(myStruct{Num: 100, Desc: "new desc"}))
	fmt.Printf("after replace:%+v\n", s)
}

代碼輸出如下，

$ go run main.go
Kind:ptr CanSet:false
Kind:struct CanSet:true Field(0).CanSet:true
after edit:{Num:10 Desc:desc}
after replace:{Num:100 Desc:new desc}

6. 修改map

反射通過(guò)func (v Value) SetMapIndex(key, elem Value)修改map指定key的value

func main() {
	m := map[int]string{
		1: "1",
		2: "2",
		3: "3",
	}
	valueM := reflect.ValueOf(m)
	// 迭代器訪問(wèn)
	iter := valueM.MapRange()
	for iter.Next() {
		fmt.Printf("key:%v val:%v\n", iter.Key(), iter.Value())
		// 將所有value修改為"a"
		valueM.SetMapIndex(iter.Key(), reflect.ValueOf("a"))
	}
	fmt.Println("--- after edit ---")
	// 通過(guò)key訪問(wèn)
	keys := valueM.MapKeys()
	for i := 0; i < len(keys); i++ {
		fmt.Printf("key:%v val:%v\n", keys[i], valueM.MapIndex(keys[i]))
	}
}

代碼輸出如下