Files
go-mc/nbt/typeinfo.go
2019-05-18 20:10:05 +08:00

49 lines
860 B
Go

package nbt
import (
"reflect"
"sync"
)
type typeInfo struct {
tagName string
nameToIndex map[string]int
}
var tinfoMap sync.Map
func getTypeInfo(typ reflect.Type) *typeInfo {
if ti, ok := tinfoMap.Load(typ); ok {
return ti.(*typeInfo)
}
tinfo := new(typeInfo)
tinfo.nameToIndex = make(map[string]int)
if typ.Kind() == reflect.Struct {
n := typ.NumField()
for i := 0; i < n; i++ {
f := typ.Field(i)
tag := f.Tag.Get("nbt")
if (f.PkgPath != "" && !f.Anonymous) || tag == "-" {
continue // Private field
}
tinfo.nameToIndex[tag] = i
if _, ok := tinfo.nameToIndex[f.Name]; !ok {
tinfo.nameToIndex[f.Name] = i
}
}
}
ti, _ := tinfoMap.LoadOrStore(typ, tinfo)
return ti.(*typeInfo)
}
func (t *typeInfo) findIndexByName(name string) int {
i, ok := t.nameToIndex[name]
if !ok {
return -1
}
return i
}