開發十年，就只剩下這套架構體系了！ >>>   

假如我們有這樣一個包：iface.go

package iface

func GetAddFunc() interface{} {
	return add
}

type i32 int32

func add(a, b i32) i32 {
	return a + b
}

希望可以在包外執行add函式，怎麼辦？此處，因為該函式簽名是不可匯出的，所以，正常思路是使用反射，程式碼可能是這樣：

import (
	"fmt"
	"iface"
	"reflect"
)

func main() {
	addIface := iface.GetAddFunc()
	ret := reflect.ValueOf(addIface).Call([]reflect.Value{
	reflect.ValueOf(1),
	reflect.ValueOf(2),
})[0].Int()
	fmt.Println(ret) // expect: 3
}

但是，執行時發現 panic 了？！錯誤資訊如下：

panic: reflect: Call using int as type iface.i32
 

這可咋辦？使用斷言？型別不可匯出，更加不可能！ 這時，unsafe的高階用法就派上用場了。先上最終程式碼，再來分析實現思路：

import (
	"fmt"
	"iface"
	"unsafe"

	"github.com/henrylee2cn/goutil/tpack"
)

func main() {
	addIface := iface.GetAddFunc()
	ptr := tpack.Unpack(addIface).Pointer()
	add := *(*func(a, b int32) int32)(unsafe.Pointer(&ptr))
	fmt.Println(add(1, 2)) // Output: 3
}
 

該程式碼，使用了本人開源的unsafe進階包tpack。 它可以幫助我們拿到函式的Pointer，之後我們就能通過unsafe強轉成外部定義的等價型別func(a, b int32) int32，進而執行它。

這種方法不但適用範圍廣泛，而且執行時沒有任何效能損耗（等價於直接呼叫iface.add）

那麼，讓我們瞭解一下tpack.Unpack(iface.GetAddFunc()).Pointer()這行程式碼做了什麼事情呢？

首先，addIface是介面型別，而介面型別的底層型別原型如下：

emptyInterface struct {
	typ  *rtype
	word unsafe.Pointer
}

typ欄位儲存型別資訊，word欄位儲存指向資料的位置資訊。我們當前需求只關心word。通過偏移量，我們可以拿到word的值，進而拿到函式在記憶體中的起始位置，即Pointer。

• 計算word偏移量，在64位系統中，該值為8：

ptrOffset := unsafe.Offsetof(new(emptyInterface).word)

• 獲取word值：

word := uintptr(unsafe.Pointer(&addIface)) + ptrOffset

• 拿到函式在記憶體中的起始位置：

ptr := *(*uintptr)(unsafe.Pointer(word))

• 最後，進行強轉型別，由於 ptr 表示函式的起始位置，而unsafe.Pointer要求是變數的指標，因此，需要使用 &ptr 進行指標型別的強轉：

add := *(*func(a, b int32) int32)(unsafe.Pointer(&ptr))

更多有趣的unsafe進階操作，可以瞭解

相關推薦