113 lines
2.0 KiB
Go
113 lines
2.0 KiB
Go
package registry
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
cda "git.loafle.net/commons/di-go/annotation"
|
|
cur "git.loafle.net/commons/util-go/reflect"
|
|
)
|
|
|
|
type TypeDefinition struct {
|
|
FullName string
|
|
PkgName string
|
|
TypeName string
|
|
Type reflect.Type
|
|
RealType reflect.Type
|
|
|
|
TypeAnnotations map[string]cda.Annotation
|
|
MethodAnnotations map[string]cda.Annotation
|
|
Fields []*FieldDefinition
|
|
}
|
|
|
|
func (td *TypeDefinition) GetAnnotation(name string) cda.Annotation {
|
|
if nil == td.TypeAnnotations {
|
|
return nil
|
|
}
|
|
|
|
return td.TypeAnnotations[name]
|
|
}
|
|
|
|
func (td *TypeDefinition) GetAnnotationByType(at reflect.Type, includeEmbedding bool) cda.Annotation {
|
|
if nil == td.TypeAnnotations {
|
|
return nil
|
|
}
|
|
|
|
for _, v := range td.TypeAnnotations {
|
|
if at == reflect.TypeOf(v) {
|
|
return v
|
|
}
|
|
|
|
if includeEmbedding {
|
|
if checkAnnotation(reflect.TypeOf(v), at) {
|
|
return v
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func checkAnnotation(t reflect.Type, st reflect.Type) bool {
|
|
rt, _, _ := cur.GetTypeInfo(t)
|
|
if reflect.Struct != rt.Kind() {
|
|
return false
|
|
}
|
|
|
|
for i := 0; i < rt.NumField(); i++ {
|
|
f := rt.Field(i)
|
|
|
|
if f.Anonymous {
|
|
if f.Type == st {
|
|
return true
|
|
}
|
|
if checkAnnotation(f.Type, st) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
type FieldDefinition struct {
|
|
FieldName string
|
|
PkgName string
|
|
TypeName string
|
|
Type reflect.Type
|
|
RealType reflect.Type
|
|
|
|
Annotations map[string]cda.Annotation
|
|
}
|
|
|
|
func (fd *FieldDefinition) GetAnnotation(name string) cda.Annotation {
|
|
if nil == fd.Annotations {
|
|
return nil
|
|
}
|
|
return fd.Annotations[name]
|
|
}
|
|
|
|
func (fd *FieldDefinition) GetAnnotationByType(at reflect.Type, includeEmbedding bool) cda.Annotation {
|
|
if nil == fd.Annotations {
|
|
return nil
|
|
}
|
|
|
|
for _, v := range fd.Annotations {
|
|
if at == reflect.TypeOf(v) {
|
|
return v
|
|
}
|
|
|
|
if includeEmbedding {
|
|
if checkAnnotation(reflect.TypeOf(v), at) {
|
|
return v
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func FullName(pkgName, typeName string) string {
|
|
return fmt.Sprintf("%s/%s", pkgName, typeName)
|
|
}
|