首先需要先明確下,可設定是針對 reflect.Value 的。普通的變數要轉變成為 reflect.Value 需要先使用 reflect.ValueOf() 來進行轉化。
那麼為什麼要有這麼一個“可設定”的方法呢?比如下面這個例子:
var x float64 = 3.4
v := reflect.ValueOf(x)
fmt.Println(v.CanSet()) // false
golang 裡面的所有函式呼叫都是值複製,所以這裡在呼叫 reflect.ValueOf 的時候,已經複製了一個 x 傳遞進去了,這裡獲取到的 v 是一個 x 複製體的 value。那麼這個時候,我們就希望知道我能不能通過 v 來設定這裡的 x 變數。就需要有個方法來輔助我們做這個事情: CanSet()
但是, 非常明顯,由於我們傳遞的是 x 的一個複製,所以這裡根本無法改變 x 的值。這裡顯示的就是 false。
那麼如果我們把 x 的地址傳遞給裡面呢?下面這個例子:
var x float64 = 3.4
v := reflect.ValueOf(&x)
fmt.Println(v.CanSet()) // false
我們將 x 變數的地址傳遞給 reflect.ValueOf 了。應該是 CanSet 了吧。但是這裡卻要注意一點,這裡的 v 指向的是 x 的指標。所以 CanSet 方法判斷的是 x 的指標是否可以設定。指標是肯定不能設定的,所以這裡還是返回 false。
那麼我們下面需要可以通過這個指標的 value 值來判斷的是,這個指標指向的元素是否可以設定,所幸 reflect 提供了 Elem() 方法來獲取這個“指標指向的元素”。
var x float64 = 3.4
v := reflect.ValueOf(&x)
fmt.Println(v.Elem().CanSet()) // true
終於返回 true 了。但是這個 Elem() 使用的時候有個前提,這裡的 value 必須是指標物件轉換的 reflect.Value。(或者是介面物件轉換的 reflect.Value)。這個前提不難理解吧,如果是一個 int 型別,它怎麼可能有指向的元素呢?所以,使用 Elem 的時候要十分注意這點,因為如果不滿足這個前提,Elem 是直接觸發 panic 的。
在判斷完是否可以設定之後,我們就可以通過 SetXX 系列方法進行對應的設定了。
var x float64 = 3.4
v := reflect.ValueOf(&x)
if v.Elem().CanSet() {
v.Elem().SetFloat(7.1)
}
fmt.Println(x)
對於複雜的 slice, map, struct, pointer 等方法,我寫了一個例子:
package main
import (
"fmt"
"reflect"
)
type Foo interface {
Name() string
}
type FooStruct struct {
A string
}
func (f FooStruct) Name() string {
return f.A
}
type FooPointer struct {
A string
}
func (f *FooPointer) Name() string {
return f.A
}
func main() {
{
// slice
a := []int{1, 2, 3}
val := reflect.ValueOf(&a)
val.Elem().SetLen(2)
val.Elem().Index(0).SetInt(4)
fmt.Println(a) // [4,2]
}
{
// map
a := map[int]string{
1: "foo1",
2: "foo2",
}
val := reflect.ValueOf(&a)
key3 := reflect.ValueOf(3)
val3 := reflect.ValueOf("foo3")
val.Elem().SetMapIndex(key3, val3)
fmt.Println(val) // &map[1:foo1 2:foo2 3:foo3]
}
{
// map
a := map[int]string{
1: "foo1",
2: "foo2",
}
val := reflect.ValueOf(a)
key3 := reflect.ValueOf(3)
val3 := reflect.ValueOf("foo3")
val.SetMapIndex(key3, val3)
fmt.Println(val) // &map[1:foo1 2:foo2 3:foo3]
}
{
// struct
a := FooStruct{}
val := reflect.ValueOf(&a)
val.Elem().FieldByName("A").SetString("foo2")
fmt.Println(a) // {foo2}
}
{
// pointer
a := &FooPointer{}
val := reflect.ValueOf(a)
val.Elem().FieldByName("A").SetString("foo2")
fmt.Println(a) //&{foo2}
}
}
上面的例子如果都能理解,那基本上也就理解了 CanSet 的方法了。
特別可以關注下,map,pointer 在修改的時候並不需要傳遞指標到 reflect.ValueOf 中。因為他們本身就是指標。
所以在呼叫 reflect.ValueOf 的時候,我們必須心裡非常明確,我們要傳遞的變數的底層結構。比如 map, 實際上傳遞的是一個指標,我們沒有必要再將他指標化了。而 slice, 實際上傳遞的是一個 SliceHeader 結構,我們在修改 Slice 的時候,必須要傳遞的是 SliceHeader 的指標。這點往往是需要我們注意的。
在 reflect 包裡面可以看到,除了 CanSet 之外,還有一個 CanAddr 方法。它們兩個有什麼區別呢?
CanAddr 方法和 CanSet 方法不一樣的地方在於:對於一些結構體內的私有欄位,我們可以獲取它的地址,但是不能設定它。
比如下面的例子:
package main
import (
"fmt"
"reflect"
)
type FooStruct struct {
A string
b int
}
func main() {
{
// struct
a := FooStruct{}
val := reflect.ValueOf(&a)
fmt.Println(val.Elem().FieldByName("b").CanSet()) // false
fmt.Println(val.Elem().FieldByName("b").CanAddr()) // true
}
}
所以,CanAddr 是 CanSet 的必要不充分條件。一個 Value 如果 CanAddr, 不一定 CanSet。但是一個變數如果 CanSet,它一定 CanAddr。
假設我們要實現這個 Value 元素 CanSet 或者 CanAddr,我們大概率會相到使用標記位標記。事實也確實是這樣。
我們先看下 Value 的結構:
type Value struct {
typ *rtype
ptr unsafe.Pointer
flag
}
這裡要注意的就是,它是一個巢狀結構,巢狀了一個 flag,而這個 flag 本身就是一個 uintptr。
type flag uintptr
這個 flag 非常重要,它既能表達這個 value 的型別,也能表達一些元資訊(比如是否可定址等)。flag雖然是uint型別,但是它用位來標記表示。
首先它需要表示型別,golang 中的型別有27個:
const (
Invalid Kind = iota
Bool
Int
Int8
Int16
Int32
Int64
Uint
Uint8
Uint16
Uint32
Uint64
Uintptr
Float32
Float64
Complex64
Complex128
Array
Chan
Func
Interface
Map
Ptr
Slice
String
Struct
UnsafePointer
)
所以使用5位(2^5-1=63)就足夠放這麼多型別了。所以 flag 的低5位是結構型別。
第六位 flagStickyRO: 標記是否是結構體內部私有屬性
第七位 flagEmbedR0: 標記是否是巢狀結構體內部私有屬性
第八位 flagIndir: 標記 value 的ptr是否是儲存了一個指標
第九位 flagAddr: 標記這個 value 是否可定址
第十位 flagMethod: 標記 value 是個匿名函式
其中比較不好理解的就是 flagStickyRO,flagEmbedR0
看下面這個例子:
type FooStruct struct {
A string
b int
}
type BarStruct struct {
FooStruct
}
{
b := BarStruct{}
val := reflect.ValueOf(&b)
c := val.Elem().FieldByName("b")
fmt.Println(c.CanAddr())
}
這個例子中的 c 的 flagEmbedR0 標記位就是1了。
所以我們再回去看 CanSet 和 CanAddr 方法
func (v Value) CanAddr() bool {
return v.flag&flagAddr != 0
}
func (v Value) CanSet() bool {
return v.flag&(flagAddr|flagRO) == flagAddr
}
他們的方法就是把 value 的 flag 和 flagAddr 或者 flagRO (flagStickyRO,flagEmbedR0) 做“與”操作。
而他們的區別就是是否判斷 flagRO 的兩個位。所以他們的不同換句話說就是“判斷這個 Value 是否是私有屬性”,私有屬性是隻讀的。不能Set。
在開發 collection (https://github.com/jianfengye/collection)庫的過程中,我就用到這麼一個方法。我希望設計一個方法 func (arr *ObjPointCollection) ToObjs(objs interface{}) error
,這個方法能將 ObjPointCollection 中的 objs reflect.Value 設定為引數 objs 中。
func (arr *ObjPointCollection) ToObjs(objs interface{}) error {
arr.mustNotBeBaseType()
objVal := reflect.ValueOf(objs)
if objVal.Elem().CanSet() {
objVal.Elem().Set(arr.objs)
return nil
}
return errors.New("element should be can set")
}
使用方法:
func TestObjPointCollection_ToObjs(t *testing.T) {
a1 := &Foo{A: "a1", B: 1}
a2 := &Foo{A: "a2", B: 2}
a3 := &Foo{A: "a3", B: 3}
bArr := []*Foo{}
objColl := NewObjPointCollection([]*Foo{a1, a2, a3})
err := objColl.ToObjs(&bArr)
if err != nil {
t.Fatal(err)
}
if len(bArr) != 3 {
t.Fatal("toObjs error len")
}
if bArr[1].A != "a2" {
t.Fatal("toObjs error copy")
}
}
總結
CanAddr 和 CanSet 剛接觸的時候是會有一些懵逼,還是需要稍微理解下 reflect.Value 的 flag 就能完全理解了。
laws-of-reflection
go addressable 詳解
Go語言_反射篇
本作品採用《CC 協議》,轉載必須註明作者和本文連結