zcy
/
background
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add

master
DESKTOP-4RNDQIC\29019 2021-02-07 00:45:59 +08:00
parent 3507633164
commit 62fb8c540c
2 changed files with 150 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
test/portData_test.go
View File

@ -58,17 +58,16 @@ func TestPortDocToElastic(t *testing.T) {
func TestReflect(t *testing.T){ func TestReflect(t *testing.T){
type XX struct{ type XX struct{
a int16 `json:"bb"` A int16 `json:"bb"`
b string `json: "cc"` B string `json: "cc" xml:"test"`
} }
x := utils.ReflectMakeNew(reflect.TypeOf(1212)) unmar := map[string] interface{}{
y := utils.ReflectMakeNew(reflect.TypeOf(XX{})) "bb" : 2,
"cc": "hello",
log.Print(x) }
log.Print(y) xx := XX{}
remap := utils.ReflectTagMap(reflect.TypeOf(XX{})) utils.UnmarshalJson(&xx,unmar)
log.Print(remap) log.Print(xx)
t.Log(x)
} }
func TestQueryDoc(t *testing.T){ func TestQueryDoc(t *testing.T){

146
utils/reflect.go
View File

@ -1,6 +1,11 @@
package utils package utils
import "reflect" import (
"log"
"reflect"
"strings"
)
func ReflectMakeNew(t reflect.Type) interface{} { func ReflectMakeNew(t reflect.Type) interface{} {
retptr := reflect.New(t) retptr := reflect.New(t)
@ -11,11 +16,142 @@ func ReflectMakeNew(t reflect.Type) interface{} {
type TagMap map[string] string type TagMap map[string] string
func ReflectTagMap(t reflect.Type) map[string] TagMap{ func ReflectTagMap(t reflect.Type) map[string] TagMap{
ret := map[string] TagMap{} log.Print(t.Name())
ret := make(map[string] TagMap)
num := t.NumField() num := t.NumField()
for i := 0;i < num;i++{ for i := 0;i < num;i++{
s := t.Field(i).Tag sub := strings.Split(string(t.Field(i).Tag),"\"")
ret[string(s)] = t.Field(i).Name for k,v := range(sub){
if len(v) > 0{
if k%2 == 0{
v = strings.Trim(v," ")
}
if v[len(v) - 1] == ':'{
if _,ok := ret[v[:len(v) - 1]];ok{
if ((k + 1) < len(sub) - 1) && ((sub[k + 1][len(sub[k + 1]) - 1] ) == ':'){
continue;
}else{
ret[v[:len(v) - 1]][sub[k + 1]] = string(t.Field(i).Name)
}
}else{
ret[v[:len(v) - 1]] = make(TagMap)
log.Print()
if ((k + 1) < len(sub) - 1) && ((sub[k + 1][len(sub[k + 1]) - 1] ) == ':'){
continue;
}else{
ret[v[:len(v) - 1]][sub[k + 1]] = string(t.Field(i).Name)
}
}
}else{
continue
}
}
}
} }
return ret return ret
} }
func SameKind(typ1 reflect.Kind,typ2 reflect.Kind) bool{
switch typ1{
case reflect.Int:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Int8:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Int16:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Int32:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Uint:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Uint8:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Uint16:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Uint32:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Uint64:
if(typ2 == reflect.Int ||typ2 == reflect.Int16 ||typ2 == reflect.Int32 ||typ2 == reflect.Int64||typ2 == reflect.Int8){
return true
}
if(typ2 == reflect.Uint ||typ2 == reflect.Uint16 ||typ2 == reflect.Uint32 ||typ2 == reflect.Uint64||typ2 == reflect.Uint8){
return true
}
case reflect.Float32:
if(typ2 == reflect.Float32 ||typ2 == reflect.Float64){
return true
}
case reflect.Float64:
if(typ2 == reflect.Float32 ||typ2 == reflect.Float64){
return true
}
default:
return (typ1 == typ2)
}
return false
}
func UnmarshalJson(value interface{}, maps map[string]interface{}){
valueof := reflect.ValueOf(value)
if !valueof.Elem().CanAddr(){
log.Print("should be addr")
return
}
remap := ReflectTagMap(valueof.Elem().Type())
_,ok := remap["json"]
if !ok{
return
}
for k,v := range(maps){
log.Print(k)
if filedName,ok := remap["json"][k];ok{
log.Print(valueof.Elem().FieldByName(filedName).Kind(),reflect.TypeOf(v).Kind().String())
if SameKind(valueof.Elem().FieldByName(filedName).Kind(),reflect.TypeOf(v).Kind()){
if(valueof.Elem().FieldByName(filedName).CanSet()){
valueof.Elem().FieldByName(filedName).Set(reflect.ValueOf(v).Convert(valueof.Elem().FieldByName(filedName).Type()))
}
}
}
}
}