This commit is contained in:
crusader 2018-03-21 18:28:21 +09:00
parent ae21044261
commit 52e9415760
3 changed files with 79 additions and 53 deletions

View File

@ -4,8 +4,6 @@ import (
"fmt" "fmt"
"reflect" "reflect"
"git.loafle.net/commons_go/logging"
cur "git.loafle.net/commons_go/util/reflect" cur "git.loafle.net/commons_go/util/reflect"
) )
@ -15,9 +13,9 @@ func init() {
annotationRegistry = make(map[string]*AnnotationDefinition, 0) annotationRegistry = make(map[string]*AnnotationDefinition, 0)
} }
func RegisterAnnotation(name string, t reflect.Type) { func RegisterAnnotation(name string, t reflect.Type) error {
if _, ok := annotationRegistry[name]; ok { if _, ok := annotationRegistry[name]; ok {
logging.Logger().Panic(fmt.Sprintf("DI: name[%s] of annotation exist already.", name)) return fmt.Errorf("DI: name[%s] of annotation exist already", name)
} }
meta := getMetaFields(t) meta := getMetaFields(t)
@ -30,6 +28,8 @@ func RegisterAnnotation(name string, t reflect.Type) {
} }
annotationRegistry[name] = def annotationRegistry[name] = def
return nil
} }
type AnnotationDefinition struct { type AnnotationDefinition struct {

View File

@ -8,7 +8,6 @@ import (
cda "git.loafle.net/commons_go/di/annotation" cda "git.loafle.net/commons_go/di/annotation"
cdia "git.loafle.net/commons_go/di/injection/annotation" cdia "git.loafle.net/commons_go/di/injection/annotation"
"git.loafle.net/commons_go/logging"
cur "git.loafle.net/commons_go/util/reflect" cur "git.loafle.net/commons_go/util/reflect"
) )
@ -19,13 +18,13 @@ func init() {
var registry ComponentRegistry var registry ComponentRegistry
type ComponentRegistry interface { type ComponentRegistry interface {
RegisterType(t reflect.Type) RegisterType(t reflect.Type) error
RegisterResource(name string, resource interface{}) error RegisterResource(name string, resource interface{}) error
GetInstance(t reflect.Type) interface{} GetInstance(t reflect.Type) (interface{}, error)
GetInstances(ts []reflect.Type) []interface{} GetInstances(ts []reflect.Type) ([]interface{}, error)
GetInstanceByName(name string) interface{} GetInstanceByName(name string) (interface{}, error)
GetInstancesByAnnotationName(n string) []interface{} GetInstancesByAnnotationName(n string) ([]interface{}, error)
} }
func newRegistry() ComponentRegistry { func newRegistry() ComponentRegistry {
@ -45,30 +44,30 @@ type defaultComponentRegistry struct {
resourceByName map[string]interface{} resourceByName map[string]interface{}
} }
func RegisterType(t reflect.Type) { func RegisterType(t reflect.Type) error {
pc, _, _, ok := runtime.Caller(1) pc, _, _, ok := runtime.Caller(1)
details := runtime.FuncForPC(pc) details := runtime.FuncForPC(pc)
if ok && details != nil { if ok && details != nil {
logging.Logger().Debug(fmt.Sprintf("called from %s", details.Name())) return fmt.Errorf("called from %s", details.Name())
} }
registry.RegisterType(t) return registry.RegisterType(t)
} }
func (cr *defaultComponentRegistry) RegisterType(t reflect.Type) { func (cr *defaultComponentRegistry) RegisterType(t reflect.Type) error {
if nil == t { if nil == t {
logging.Logger().Panic("DI: t[reflect.Type] is nil") return fmt.Errorf("DI: t[reflect.Type] is nil")
} }
if !cur.IsTypeKind(t, reflect.Struct, true) { if !cur.IsTypeKind(t, reflect.Struct, true) {
logging.Logger().Panic(fmt.Sprintf("DI: t[reflect.Type] must be specified but is %v", t)) return fmt.Errorf("DI: t[reflect.Type] must be specified but is %v", t)
} }
td, err := cr.buildDefinition(t) td, err := cr.buildDefinition(t)
if nil != err { if nil != err {
logging.Logger().Panic(fmt.Sprintf("DI: %v", err)) return fmt.Errorf("DI: %v", err)
} }
if _, ok := cr.definitionByType[td.RealType]; ok { if _, ok := cr.definitionByType[td.RealType]; ok {
logging.Logger().Panic(fmt.Sprintf("DI: The type[%s] of Component is exist already", td.FullName)) return fmt.Errorf("DI: The type[%s] of Component is exist already", td.FullName)
} }
cr.definitionByType[td.RealType] = td cr.definitionByType[td.RealType] = td
@ -82,9 +81,11 @@ func (cr *defaultComponentRegistry) RegisterType(t reflect.Type) {
} }
if eTD, ok := cr.definitionByName[name]; ok { if eTD, ok := cr.definitionByName[name]; ok {
logging.Logger().Panic(fmt.Sprintf("DI: The name[%s] of Component is exist already type[%s]", name, eTD.FullName)) return fmt.Errorf("DI: The name[%s] of Component is exist already type[%s]", name, eTD.FullName)
} }
cr.definitionByName[name] = td cr.definitionByName[name] = td
return nil
} }
func RegisterResource(name string, resource interface{}) error { func RegisterResource(name string, resource interface{}) error {
@ -101,20 +102,20 @@ func (cr *defaultComponentRegistry) RegisterResource(name string, resource inter
// GetInstance returns instance of type // GetInstance returns instance of type
// t must be pointer of struct // t must be pointer of struct
func GetInstance(t reflect.Type) interface{} { func GetInstance(t reflect.Type) (interface{}, error) {
return registry.GetInstance(t) return registry.GetInstance(t)
} }
func (cr *defaultComponentRegistry) GetInstance(t reflect.Type) interface{} { func (cr *defaultComponentRegistry) GetInstance(t reflect.Type) (interface{}, error) {
if nil == t { if nil == t {
logging.Logger().Panic(fmt.Sprintf("DI: t[reflect.Type] is nil")) return nil, fmt.Errorf("DI: t[reflect.Type] is nil")
} }
if reflect.Ptr != t.Kind() { if reflect.Ptr != t.Kind() {
logging.Logger().Panic(fmt.Sprintf("DI: t[reflect.Type] must be pointer of struct")) return nil, fmt.Errorf("DI: t[reflect.Type] must be pointer of struct")
} }
if reflect.Struct != t.Elem().Kind() { if reflect.Struct != t.Elem().Kind() {
logging.Logger().Panic(fmt.Sprintf("DI: t[reflect.Type] must be pointer of struct")) return nil, fmt.Errorf("DI: t[reflect.Type] must be pointer of struct")
} }
var err error var err error
@ -124,13 +125,13 @@ func (cr *defaultComponentRegistry) GetInstance(t reflect.Type) interface{} {
if !ok { if !ok {
td, err = cr.buildDefinition(t) td, err = cr.buildDefinition(t)
if nil != err { if nil != err {
logging.Logger().Panic(fmt.Sprintf("DI: %v", err)) return nil, fmt.Errorf("DI: %v", err)
} }
} }
comV, ok := cr.instanceByType[td.RealType] comV, ok := cr.instanceByType[td.RealType]
if ok { if ok {
return comV return comV, nil
} }
v := reflect.New(rt) v := reflect.New(rt)
@ -143,14 +144,16 @@ func (cr *defaultComponentRegistry) GetInstance(t reflect.Type) interface{} {
f := rv.FieldByName(fd.FieldName) f := rv.FieldByName(fd.FieldName)
if !f.IsValid() { if !f.IsValid() {
logging.Logger().Panic(fmt.Sprintf("DI: Field[%s] is not valid", fd.FieldName)) return nil, fmt.Errorf("DI: Field[%s] is not valid", fd.FieldName)
} }
if !f.CanSet() { if !f.CanSet() {
logging.Logger().Panic(fmt.Sprintf("DI: Field[%s] can not set", fd.FieldName)) return nil, fmt.Errorf("DI: Field[%s] can not set", fd.FieldName)
} }
if annotation = fd.GetAnnotationByType(reflect.TypeOf((*cdia.Inject)(nil)), false); nil != annotation { if annotation = fd.GetAnnotationByType(reflect.TypeOf((*cdia.Inject)(nil)), false); nil != annotation {
fV = cr.GetInstance(fd.Type) if fV, err = cr.GetInstance(fd.Type); nil != err {
return nil, err
}
} }
if annotation = fd.GetAnnotationByType(reflect.TypeOf((*cdia.Resource)(nil)), false); nil != annotation { if annotation = fd.GetAnnotationByType(reflect.TypeOf((*cdia.Resource)(nil)), false); nil != annotation {
@ -158,64 +161,76 @@ func (cr *defaultComponentRegistry) GetInstance(t reflect.Type) interface{} {
if "" == n { if "" == n {
n = fd.FieldName n = fd.FieldName
} }
fV = cr.GetInstanceByName(n) if fV, err = cr.GetInstanceByName(n); nil != err {
return nil, err
}
} }
if nil != err { if nil != err {
logging.Logger().Panic(fmt.Sprintf("DI: %v", err)) return nil, fmt.Errorf("DI: %v", err)
} }
f.Set(reflect.ValueOf(fV)) f.Set(reflect.ValueOf(fV))
} }
cr.instanceByType[td.RealType] = v.Interface() cr.instanceByType[td.RealType] = v.Interface()
return v.Interface() return v.Interface(), nil
} }
func GetInstanceByName(name string) interface{} { func GetInstanceByName(name string) (interface{}, error) {
return registry.GetInstanceByName(name) return registry.GetInstanceByName(name)
} }
func (cr *defaultComponentRegistry) GetInstanceByName(name string) interface{} { func (cr *defaultComponentRegistry) GetInstanceByName(name string) (interface{}, error) {
v, ok := cr.resourceByName[name] v, ok := cr.resourceByName[name]
if ok { if ok {
return v return v, nil
} }
logging.Logger().Panic(fmt.Sprintf("DI: Resource[%s] is not exist", name)) return nil, fmt.Errorf("DI: Resource[%s] is not exist", name)
return nil
} }
// GetInstances returns instance of annotated // GetInstances returns instance of annotated
// n must be name of registered annotation // n must be name of registered annotation
func GetInstances(ts []reflect.Type) []interface{} { func GetInstances(ts []reflect.Type) ([]interface{}, error) {
return registry.GetInstances(ts) return registry.GetInstances(ts)
} }
func (cr *defaultComponentRegistry) GetInstances(ts []reflect.Type) []interface{} { func (cr *defaultComponentRegistry) GetInstances(ts []reflect.Type) ([]interface{}, error) {
var (
i interface{}
err error
)
instances := make([]interface{}, 0) instances := make([]interface{}, 0)
for _, t := range ts { for _, t := range ts {
if i := cr.GetInstance(t); nil != i { if i, err = cr.GetInstance(t); nil != err {
return nil, err
}
instances = append(instances, i) instances = append(instances, i)
} }
}
return instances return instances, nil
} }
// GetInstancesByAnnotationName returns instance of annotated // GetInstancesByAnnotationName returns instance of annotated
// n must be name of registered annotation // n must be name of registered annotation
func GetInstancesByAnnotationName(n string) []interface{} { func GetInstancesByAnnotationName(n string) ([]interface{}, error) {
return registry.GetInstancesByAnnotationName(n) return registry.GetInstancesByAnnotationName(n)
} }
func (cr *defaultComponentRegistry) GetInstancesByAnnotationName(n string) []interface{} { func (cr *defaultComponentRegistry) GetInstancesByAnnotationName(n string) ([]interface{}, error) {
var (
i interface{}
err error
)
instances := make([]interface{}, 0) instances := make([]interface{}, 0)
for _, td := range cr.definitionByType { for _, td := range cr.definitionByType {
if nil != td.GetAnnotation(n) { if nil != td.GetAnnotation(n) {
instances = append(instances, cr.GetInstance(td.Type)) if i, err = cr.GetInstance(td.Type); nil != err {
return nil, err
}
instances = append(instances, i)
} }
} }
return instances return instances, nil
} }
func (cr *defaultComponentRegistry) buildDefinition(t reflect.Type) (*TypeDefinition, error) { func (cr *defaultComponentRegistry) buildDefinition(t reflect.Type) (*TypeDefinition, error) {

View File

@ -10,6 +10,11 @@ import (
) )
func TestRegisterType(t *testing.T) { func TestRegisterType(t *testing.T) {
var (
err error
css []interface{}
cs interface{}
)
RegisterType(reflect.TypeOf((*AService)(nil))) RegisterType(reflect.TypeOf((*AService)(nil)))
RegisterType(reflect.TypeOf((*BService)(nil))) RegisterType(reflect.TypeOf((*BService)(nil)))
// RegisterType(reflect.TypeOf((*TestStruct2)(nil)), &cda.ComponentAnnotation{ // RegisterType(reflect.TypeOf((*TestStruct2)(nil)), &cda.ComponentAnnotation{
@ -24,17 +29,23 @@ func TestRegisterType(t *testing.T) {
RegisterResource("List", []string{"dfdkf", "skgkfg"}) RegisterResource("List", []string{"dfdkf", "skgkfg"})
css := GetInstancesByAnnotationName(cdia.ComponentTag) if css, err = GetInstancesByAnnotationName(cdia.ComponentTag); nil != err {
log.Printf("%v \n", err)
}
log.Printf("%v", css) log.Printf("%v", css)
css2 := GetInstances([]reflect.Type{ if css, err = GetInstances([]reflect.Type{
reflect.TypeOf((*AService)(nil)), reflect.TypeOf((*AService)(nil)),
reflect.TypeOf((*BService)(nil)), reflect.TypeOf((*BService)(nil)),
}) }); nil != err {
log.Printf("%v", css2) log.Printf("%v \n", err)
}
log.Printf("%v", css)
cs := GetInstance(reflect.TypeOf((*CService)(nil))).(*CService) if cs, err = GetInstance(reflect.TypeOf((*CService)(nil))); nil != err {
log.Printf("%v", cs) log.Printf("%v \n", err)
}
log.Printf("%v", cs.(*CService))
} }