This commit is contained in:
crusader 2017-12-06 15:07:44 +09:00
parent c69ba7c3c0
commit 700f188618
7 changed files with 165 additions and 94 deletions

View File

@ -9,58 +9,137 @@ import (
) )
const ( const (
AnnotationChar = "@"
AnnotationStartChar = "(" AnnotationStartChar = "("
AnnotationEndChar = ")" AnnotationEndChar = ")"
AnnotationSpliter = ";"
AnnotationAttributeSpliter = "," AnnotationAttributeSpliter = ","
AnnotationKeyValueSpliter = "=" AnnotationKeyValueSpliter = "="
) )
var annotationRegistry map[string]reflect.Type
func init() {
annotationRegistry = make(map[string]reflect.Type, 0)
}
func registerAnnotation(name string, t reflect.Type) {
annotationRegistry[name] = t
}
type Annotation interface { type Annotation interface {
parseAttribute(attributes map[string]string) error
} }
// @Inject(name? string) // @Inject(name? string)
// @Resource(name? string) // @Resource(name? string)
func ParseAnnotation(tag reflect.StructTag) (map[string]Annotation, error) { func ParseAnnotation(tag reflect.StructTag) ([]Annotation, error) {
a := strings.Trim(tag.Get(di.AnnotationTag), " ") s := strings.Trim(tag.Get(di.AnnotationTag), " ")
if "" == a { if "" == s {
return nil, nil return nil, nil
} }
rKVs := make(map[string]Annotation) annotations := strings.Split(s, AnnotationSpliter)
if nil == annotations || 0 == len(annotations) {
inject, err := ParseInject(a) return nil, nil
if nil != err {
return nil, err
} }
rKVs[InjectTag] = inject
resource, err := ParseResource(a) rKVs := make([]Annotation, 0)
if nil != err { for _, a := range annotations {
return nil, err a = strings.Trim(a, " ")
if "" == a {
continue
}
aName, attributes, err := ParseAnnotationItem(a)
if nil != err {
return nil, err
}
if "" == aName {
continue
}
annotation, err := NewAnnotation(aName, attributes)
if nil != err {
return nil, err
}
rKVs = append(rKVs, annotation)
} }
rKVs[ResourceTag] = resource
return rKVs, nil return rKVs, nil
} }
func ParseAttribute(a string, startIndex int) (map[string]string, error) { func NewAnnotation(name string, attributes map[string]string) (Annotation, error) {
if startIndex >= len(a) { t, ok := annotationRegistry[name]
if !ok {
return nil, fmt.Errorf("There is no annotation[%s]", name)
}
v := reflect.New(t.Elem())
i := v.Interface().(Annotation)
if nil != attributes {
i.parseAttribute(attributes)
}
return i, nil
}
func ParseAnnotationItem(a string) (name string, attributes map[string]string, err error) {
s := strings.Trim(a, " ")
if "" == s {
return
}
i := strings.Index(s, AnnotationChar)
if -1 == i {
err = fmt.Errorf("Syntax error: annotation must be started %s", AnnotationChar)
return
}
aStart := strings.Index(s, AnnotationStartChar)
if -1 == aStart {
// This is pure annotation ex)@Resource
name = s
return
}
name = s[:aStart]
aEnd := strings.Index(s, AnnotationEndChar)
if -1 == aEnd {
// This is syntax error ex)@Resource(
err = fmt.Errorf("Syntax error: annotation must be ended %s", AnnotationEndChar)
return
}
if 1 >= aEnd-aStart {
return
}
attr := s[aStart+1 : aEnd]
attr = strings.Trim(attr, " ")
if "" == attr {
return
}
attrs, pErr := ParseAttribute(s[aStart+1 : aEnd])
if nil != pErr {
err = pErr
return
}
attributes = attrs
return
}
func ParseAttribute(s string) (map[string]string, error) {
attr := strings.Trim(s, " ")
if "" == attr {
return nil, nil return nil, nil
} }
if AnnotationStartChar != string([]rune(a)[startIndex]) { kvs := strings.Split(attr, AnnotationAttributeSpliter)
return nil, nil if nil == kvs || 0 == len(kvs) {
}
endIndex := strings.Index(a[startIndex+1:], AnnotationEndChar)
if -1 == endIndex {
return nil, fmt.Errorf("DI: Syntax error - annotation(%s) has '%s', but '%s' is not exist", a, AnnotationStartChar, AnnotationEndChar)
}
endIndex = endIndex + startIndex + 1
body := a[startIndex+1 : endIndex]
kvs := strings.Split(body, AnnotationAttributeSpliter)
if 0 == len(kvs) {
return nil, nil return nil, nil
} }

View File

@ -5,7 +5,8 @@ import (
) )
type ComponentAnnotation struct { type ComponentAnnotation struct {
Names []string Annotation
Name string
InitMethod string // func (receiver interface{}, cr ComponentRegistry) error InitMethod string // func (receiver interface{}, cr ComponentRegistry) error
DestroyMethod string // func (receiver interface{}, cr ComponentRegistry) error DestroyMethod string // func (receiver interface{}, cr ComponentRegistry) error
Scope di.ScopeType Scope di.ScopeType

View File

@ -3,37 +3,31 @@ package annotation
// @Inject(name? string) // @Inject(name? string)
import ( import (
"strings" "fmt"
"reflect"
) )
const ( const (
InjectTag = "@Inject" InjectTag = "@Inject"
) )
func init() {
registerAnnotation(InjectTag, reflect.TypeOf((*Inject)(nil)))
}
type Inject struct { type Inject struct {
Annotation Annotation
Name string Name string
} }
func ParseInject(a string) (*Inject, error) { func (a *Inject) parseAttribute(attributes map[string]string) error {
i := strings.Index(a, InjectTag) for k, v := range attributes {
if -1 == i {
return nil, nil
}
inject := &Inject{}
atts, err := ParseAttribute(a, i+len(InjectTag))
if nil != err {
return nil, err
}
for k, v := range atts {
switch k { switch k {
case "name": case "name":
inject.Name = v a.Name = v
default:
return fmt.Errorf("Syntax error: not supported attribute[%s]", k)
} }
} }
return nil
return inject, nil
} }

View File

@ -1,35 +1,31 @@
package annotation package annotation
import "strings" import (
"fmt"
"reflect"
)
const ( const (
ResourceTag = "@Resource" ResourceTag = "@Resource"
) )
func init() {
registerAnnotation(ResourceTag, reflect.TypeOf((*Resource)(nil)))
}
type Resource struct { type Resource struct {
Annotation Annotation
Name string Name string
} }
func ParseResource(a string) (*Resource, error) { func (a *Resource) parseAttribute(attributes map[string]string) error {
i := strings.Index(a, ResourceTag) for k, v := range attributes {
if -1 == i {
return nil, nil
}
inject := &Resource{}
atts, err := ParseAttribute(a, i+len(ResourceTag))
if nil != err {
return nil, err
}
for k, v := range atts {
switch k { switch k {
case "name": case "name":
inject.Name = v a.Name = v
default:
return fmt.Errorf("Syntax error: not supported attribute[%s]", k)
} }
} }
return nil
return inject, nil
} }

View File

@ -23,7 +23,7 @@ type FieldDefinition struct {
Type reflect.Type Type reflect.Type
RealType reflect.Type RealType reflect.Type
Annotations map[string]cda.Annotation Annotations []cda.Annotation
} }
func FullName(pkgName, typeName string) string { func FullName(pkgName, typeName string) string {

View File

@ -4,6 +4,8 @@ import (
"fmt" "fmt"
"log" "log"
"reflect" "reflect"
"runtime"
"strings"
cda "git.loafle.net/commons_go/di/annotation" cda "git.loafle.net/commons_go/di/annotation"
cdur "git.loafle.net/commons_go/di/util/reflect" cdur "git.loafle.net/commons_go/di/util/reflect"
@ -37,6 +39,12 @@ type defaultComponentRegistry struct {
} }
func RegisterType(t reflect.Type, ca *cda.ComponentAnnotation) { func RegisterType(t reflect.Type, ca *cda.ComponentAnnotation) {
pc, _, _, ok := runtime.Caller(1)
details := runtime.FuncForPC(pc)
if ok && details != nil {
log.Printf("called from %s\n", details.Name())
}
registry.RegisterType(t, ca) registry.RegisterType(t, ca)
} }
func (cr *defaultComponentRegistry) RegisterType(t reflect.Type, ca *cda.ComponentAnnotation) { func (cr *defaultComponentRegistry) RegisterType(t reflect.Type, ca *cda.ComponentAnnotation) {
@ -57,24 +65,17 @@ func (cr *defaultComponentRegistry) RegisterType(t reflect.Type, ca *cda.Compone
} }
cr.definitionByType[td.RealType] = td cr.definitionByType[td.RealType] = td
names := make([]string, 0) name := ""
if nil != ca { if nil != ca && "" != strings.Trim(ca.Name, " ") {
if nil != ca.Names && 0 < len(ca.Names) { name = ca.Name
for _, n := range ca.Names { } else {
names = append(names, n) name = td.TypeName
}
}
}
if 0 == len(names) {
names = append(names, td.FullName)
} }
for _, n := range names { if eTD, ok := cr.definitionByName[name]; ok {
if eTD, ok := cr.definitionByName[n]; ok { logging.Logger().Panic(fmt.Sprintf("DI: The name[%s] of Component is exist already type[%s]", name, eTD.FullName))
logging.Logger().Panic(fmt.Sprintf("DI: The name[%s] of Component is exist already type[%s]", n, eTD.FullName))
}
cr.definitionByName[n] = td
} }
cr.definitionByName[name] = td
} }
func GetInstance(t reflect.Type) (interface{}, error) { func GetInstance(t reflect.Type) (interface{}, error) {
@ -115,12 +116,12 @@ func (cr *defaultComponentRegistry) buildDefinition(t reflect.Type) (*TypeDefini
} }
rt, pkgName, tName := cdur.GetTypeInfo(t) rt, pkgName, tName := cdur.GetTypeInfo(t)
cd := &TypeDefinition{} td := &TypeDefinition{}
cd.FullName = FullName(pkgName, tName) td.FullName = FullName(pkgName, tName)
cd.PkgName = pkgName td.PkgName = pkgName
cd.TypeName = tName td.TypeName = tName
cd.Type = t td.Type = t
cd.RealType = rt td.RealType = rt
nf := rt.NumField() nf := rt.NumField()
if 0 < nf { if 0 < nf {
@ -146,9 +147,9 @@ func (cr *defaultComponentRegistry) buildDefinition(t reflect.Type) (*TypeDefini
} }
if 0 < len(fields) { if 0 < len(fields) {
cd.Fields = fields td.Fields = fields
} }
} }
return cd, nil return td, nil
} }

View File

@ -10,25 +10,25 @@ import (
func TestRegisterType(t *testing.T) { func TestRegisterType(t *testing.T) {
RegisterType(reflect.TypeOf((*TestStruct1)(nil)), &cda.ComponentAnnotation{}) RegisterType(reflect.TypeOf((*TestStruct1)(nil)), &cda.ComponentAnnotation{})
RegisterType(reflect.TypeOf((*TestStruct2)(nil)), &cda.ComponentAnnotation{ RegisterType(reflect.TypeOf((*TestStruct2)(nil)), &cda.ComponentAnnotation{
Names: []string{"test1", "test2"}, Name: "test1",
}) })
RegisterType(reflect.TypeOf((*TestStruct3)(nil)), &cda.ComponentAnnotation{ RegisterType(reflect.TypeOf((*TestStruct3)(nil)), &cda.ComponentAnnotation{
Names: []string{"test1", "test2"}, Name: "test2",
}) })
} }
type TestStruct1 struct { type TestStruct1 struct {
Name1 string `annotation:"@Inject"` Name1 string `annotation:"@Inject"`
Name2 string `annotation:"@Inject()"` Name2 string `annotation:"@Inject()"`
Name3 string `annotation:"@Inject(name=test)"` Name3 string `annotation:"@Inject(name=test1)"`
} }
type TestStruct2 struct { type TestStruct2 struct {
Name1 string `annotation:"@Inject"` Name1 string `annotation:"@Inject"`
Name2 string `annotation:"@Inject()"` Name2 string `annotation:"@Inject()"`
Name3 string `annotation:"@Inject(name=test)"` Name3 string `annotation:"@Inject(name=test2)"`
} }
type TestStruct3 struct { type TestStruct3 struct {
Name1 string `annotation:"@Inject"` Name1 string `annotation:"@Inject"`
Name2 string `annotation:"@Inject()"` Name2 string `annotation:"@Inject()"`
Name3 string `annotation:"@Inject(name=test)"` Name3 string `annotation:"@Inject(name=test3)"`
} }