project init
This commit is contained in:
commit
c40b3be3cc
69
.gitignore
vendored
Normal file
69
.gitignore
vendored
Normal file
|
@ -0,0 +1,69 @@
|
|||
# Created by .ignore support plugin (hsz.mobi)
|
||||
### JetBrains template
|
||||
# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and Webstorm
|
||||
# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
|
||||
|
||||
# User-specific stuff:
|
||||
.idea/**/workspace.xml
|
||||
.idea/**/tasks.xml
|
||||
.idea/dictionaries
|
||||
|
||||
# Sensitive or high-churn files:
|
||||
.idea/**/dataSources/
|
||||
.idea/**/dataSources.ids
|
||||
.idea/**/dataSources.xml
|
||||
.idea/**/dataSources.local.xml
|
||||
.idea/**/sqlDataSources.xml
|
||||
.idea/**/dynamic.xml
|
||||
.idea/**/uiDesigner.xml
|
||||
|
||||
# Gradle:
|
||||
.idea/**/gradle.xml
|
||||
.idea/**/libraries
|
||||
|
||||
# Mongo Explorer plugin:
|
||||
.idea/**/mongoSettings.xml
|
||||
|
||||
## File-based project format:
|
||||
*.iws
|
||||
|
||||
## Plugin-specific files:
|
||||
|
||||
# IntelliJ
|
||||
/out/
|
||||
|
||||
# mpeltonen/sbt-idea plugin
|
||||
.idea_modules/
|
||||
|
||||
# JIRA plugin
|
||||
atlassian-ide-plugin.xml
|
||||
|
||||
# Crashlytics plugin (for Android Studio and IntelliJ)
|
||||
com_crashlytics_export_strings.xml
|
||||
crashlytics.properties
|
||||
crashlytics-build.properties
|
||||
fabric.properties
|
||||
### Go template
|
||||
# Binaries for programs and plugins
|
||||
*.exe
|
||||
*.dll
|
||||
*.so
|
||||
*.dylib
|
||||
|
||||
# Test binary, build with `go test -c`
|
||||
*.test
|
||||
|
||||
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||
*.out
|
||||
|
||||
# Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
|
||||
.glide/
|
||||
.idea/
|
||||
*.iml
|
||||
|
||||
vendor/
|
||||
glide.lock
|
||||
.DS_Store
|
||||
dist/
|
||||
debug
|
||||
debug.test
|
32
.vscode/launch.json
vendored
Normal file
32
.vscode/launch.json
vendored
Normal file
|
@ -0,0 +1,32 @@
|
|||
{
|
||||
"version": "0.2.0",
|
||||
"configurations": [
|
||||
{
|
||||
"name": "Debug",
|
||||
"type": "go",
|
||||
"request": "launch",
|
||||
"mode": "debug",
|
||||
"remotePath": "",
|
||||
"port": 2345,
|
||||
"host": "127.0.0.1",
|
||||
"program": "${workspaceRoot}/main.go",
|
||||
"env": {},
|
||||
"args": [],
|
||||
"showLog": true
|
||||
},
|
||||
{
|
||||
"name": "File Debug",
|
||||
"type": "go",
|
||||
"request": "launch",
|
||||
"mode": "debug",
|
||||
"remotePath": "",
|
||||
"port": 2345,
|
||||
"host": "127.0.0.1",
|
||||
"program": "${fileDirname}",
|
||||
"env": {},
|
||||
"args": [],
|
||||
"showLog": true
|
||||
}
|
||||
|
||||
]
|
||||
}
|
3
.vscode/settings.json
vendored
Normal file
3
.vscode/settings.json
vendored
Normal file
|
@ -0,0 +1,3 @@
|
|||
// Place your settings in this file to overwrite default and user settings.
|
||||
{
|
||||
}
|
148
annotation/annotation.go
Normal file
148
annotation/annotation.go
Normal file
|
@ -0,0 +1,148 @@
|
|||
package annotation
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
AnnotationTag = "annotation"
|
||||
AnnotationMetaTag = "annotation-meta"
|
||||
AnnotationChar = "@"
|
||||
AnnotationStartChar = "("
|
||||
AnnotationEndChar = ")"
|
||||
AnnotationSpliter = ";"
|
||||
AnnotationAttributeSpliter = ","
|
||||
AnnotationKeyValueSpliter = "="
|
||||
)
|
||||
|
||||
type Annotation interface {
|
||||
}
|
||||
|
||||
// @Inject(name? string)
|
||||
// @Resource(name? string)
|
||||
func ParseAnnotation(tag reflect.StructTag) (map[string]Annotation, error) {
|
||||
s := strings.Trim(tag.Get(AnnotationTag), " ")
|
||||
if "" == s {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
annotations := strings.Split(s, AnnotationSpliter)
|
||||
if nil == annotations || 0 == len(annotations) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
rKVs := make(map[string]Annotation, 0)
|
||||
for _, a := range annotations {
|
||||
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[aName] = annotation
|
||||
}
|
||||
|
||||
return rKVs, nil
|
||||
}
|
||||
|
||||
func newAnnotation(name string, attributes map[string]string) (Annotation, error) {
|
||||
def, ok := annotationRegistry[name]
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("There is no annotation[%s]", name)
|
||||
}
|
||||
|
||||
v := reflect.New(def.rt)
|
||||
i := v.Interface().(Annotation)
|
||||
|
||||
if nil != attributes {
|
||||
setMetaAttributes(def, v.Elem(), 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 := parseAttributeString(s[aStart+1 : aEnd])
|
||||
if nil != pErr {
|
||||
err = pErr
|
||||
return
|
||||
}
|
||||
|
||||
attributes = attrs
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func parseAttributeString(s string) (map[string]string, error) {
|
||||
attr := strings.Trim(s, " ")
|
||||
if "" == attr {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
kvs := strings.Split(attr, AnnotationAttributeSpliter)
|
||||
if nil == kvs || 0 == len(kvs) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
rKVs := make(map[string]string)
|
||||
for i := 0; i < len(kvs); i++ {
|
||||
s := strings.Trim(kvs[i], " ")
|
||||
if "" == s {
|
||||
continue
|
||||
}
|
||||
kv := strings.Split(s, AnnotationKeyValueSpliter)
|
||||
k := strings.Trim(kv[0], " ")
|
||||
v := strings.Trim(kv[1], " ")
|
||||
rKVs[k] = v
|
||||
}
|
||||
|
||||
return rKVs, nil
|
||||
}
|
40
annotation/definition.go
Normal file
40
annotation/definition.go
Normal file
|
@ -0,0 +1,40 @@
|
|||
package annotation
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
|
||||
cur "git.loafle.net/commons/util-go/reflect"
|
||||
)
|
||||
|
||||
var annotationRegistry map[string]*AnnotationDefinition
|
||||
|
||||
func init() {
|
||||
annotationRegistry = make(map[string]*AnnotationDefinition, 0)
|
||||
}
|
||||
|
||||
func RegisterAnnotation(name string, t reflect.Type) error {
|
||||
if _, ok := annotationRegistry[name]; ok {
|
||||
return fmt.Errorf("DI: name[%s] of annotation exist already", name)
|
||||
}
|
||||
|
||||
meta := getMetaFields(t)
|
||||
fRT, _, _ := cur.GetTypeInfo(t)
|
||||
|
||||
def := &AnnotationDefinition{
|
||||
t: t,
|
||||
rt: fRT,
|
||||
fields: meta,
|
||||
}
|
||||
|
||||
annotationRegistry[name] = def
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type AnnotationDefinition struct {
|
||||
t reflect.Type
|
||||
rt reflect.Type
|
||||
|
||||
fields map[string]*AnnotationFieldMeta
|
||||
}
|
87
annotation/meta.go
Normal file
87
annotation/meta.go
Normal file
|
@ -0,0 +1,87 @@
|
|||
package annotation
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"strings"
|
||||
|
||||
cur "git.loafle.net/commons/util-go/reflect"
|
||||
)
|
||||
|
||||
type AnnotationFieldMeta struct {
|
||||
fieldName string
|
||||
options anotationMetaOptions
|
||||
}
|
||||
|
||||
type anotationMetaOptions string
|
||||
|
||||
func (o anotationMetaOptions) Contains(optionName string) bool {
|
||||
if len(o) == 0 {
|
||||
return false
|
||||
}
|
||||
s := string(o)
|
||||
for s != "" {
|
||||
var next string
|
||||
i := strings.Index(s, AnnotationAttributeSpliter)
|
||||
if i >= 0 {
|
||||
s, next = s[:i], s[i+1:]
|
||||
}
|
||||
if s == optionName {
|
||||
return true
|
||||
}
|
||||
s = next
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func getMetaFields(t reflect.Type) map[string]*AnnotationFieldMeta {
|
||||
fields := make(map[string]*AnnotationFieldMeta, 0)
|
||||
rt, _, _ := cur.GetTypeInfo(t)
|
||||
if reflect.Struct != rt.Kind() {
|
||||
return fields
|
||||
}
|
||||
|
||||
for i := 0; i < rt.NumField(); i++ {
|
||||
f := rt.Field(i)
|
||||
|
||||
if f.Anonymous {
|
||||
pFields := getMetaFields(f.Type)
|
||||
for k, v := range pFields {
|
||||
fields[k] = v
|
||||
}
|
||||
} else {
|
||||
name, metaOptions := parseAnnotationMeta(f.Tag)
|
||||
if "" == name {
|
||||
continue
|
||||
}
|
||||
fields[name] = &AnnotationFieldMeta{
|
||||
fieldName: f.Name,
|
||||
options: metaOptions,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return fields
|
||||
}
|
||||
|
||||
func parseAnnotationMeta(tag reflect.StructTag) (string, anotationMetaOptions) {
|
||||
s := strings.Trim(tag.Get(AnnotationMetaTag), " ")
|
||||
if "" == s {
|
||||
return "", ""
|
||||
}
|
||||
|
||||
if idx := strings.Index(s, AnnotationAttributeSpliter); idx != -1 {
|
||||
return s[:idx], anotationMetaOptions(s[idx+1:])
|
||||
}
|
||||
|
||||
return s, anotationMetaOptions("")
|
||||
}
|
||||
|
||||
func setMetaAttributes(def *AnnotationDefinition, rv reflect.Value, attributes map[string]string) {
|
||||
for k, v := range attributes {
|
||||
meta := def.fields[k]
|
||||
|
||||
f := rv.FieldByName(meta.fieldName)
|
||||
|
||||
f.Set(reflect.ValueOf(v))
|
||||
}
|
||||
}
|
5
annotation/type.go
Normal file
5
annotation/type.go
Normal file
|
@ -0,0 +1,5 @@
|
|||
package annotation
|
||||
|
||||
type TypeAnnotation interface {
|
||||
Annotation
|
||||
}
|
25
constants.go
Normal file
25
constants.go
Normal file
|
@ -0,0 +1,25 @@
|
|||
package di
|
||||
|
||||
// `annotation:"@Inject(name? string)"`
|
||||
// field 에 적용
|
||||
// 1. 타입으로 매칭 없으면 에러 2. 여러개의 타입이 검색되면 그 중에 name으로 매칭
|
||||
|
||||
// `annotation:"@Resource(name? string)"`
|
||||
// field 에 적용
|
||||
// 1. 이름으로 매칭 2. 타입으로 매칭
|
||||
// 이름이 지정되지 않으면 field 이름으로 먼저 찾고 없으면 타입으로 매칭
|
||||
|
||||
// @Component
|
||||
// Component 등록 시에 파라미터로 제공
|
||||
// names []string
|
||||
// initMethod string
|
||||
// destroyMethod string
|
||||
// scope enum singleton, transiant
|
||||
|
||||
type ScopeType string
|
||||
|
||||
const (
|
||||
ScopeTypeDefault ScopeType = "Default" // == ScopeTypeSingleton
|
||||
ScopeTypeSingleton ScopeType = "Singleton"
|
||||
ScopeTypeTransiant ScopeType = "Transiant"
|
||||
)
|
2
glide.yaml
Normal file
2
glide.yaml
Normal file
|
@ -0,0 +1,2 @@
|
|||
package: git.loafle.net/commons/di-go
|
||||
import: []
|
26
injection/annotation/component.go
Normal file
26
injection/annotation/component.go
Normal file
|
@ -0,0 +1,26 @@
|
|||
package annotation
|
||||
|
||||
// @Inject(name? string)
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
|
||||
"git.loafle.net/commons/di-go"
|
||||
cda "git.loafle.net/commons/di-go/annotation"
|
||||
)
|
||||
|
||||
const (
|
||||
ComponentTag = "@Component"
|
||||
)
|
||||
|
||||
func init() {
|
||||
cda.RegisterAnnotation(ComponentTag, reflect.TypeOf((*Component)(nil)))
|
||||
}
|
||||
|
||||
type Component struct {
|
||||
cda.Annotation
|
||||
Name string `annotation-meta:"name"`
|
||||
InitMethod string `annotation-meta:"initMethod"` // func (receiver interface{}, cr ComponentRegistry) error
|
||||
DestroyMethod string `annotation-meta:"destroyMethod"` // func (receiver interface{}, cr ComponentRegistry) error
|
||||
Scope di.ScopeType `annotation-meta:"scope"`
|
||||
}
|
22
injection/annotation/inject.go
Normal file
22
injection/annotation/inject.go
Normal file
|
@ -0,0 +1,22 @@
|
|||
package annotation
|
||||
|
||||
// @Inject(name? string)
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
|
||||
cda "git.loafle.net/commons/di-go/annotation"
|
||||
)
|
||||
|
||||
const (
|
||||
InjectTag = "@Inject"
|
||||
)
|
||||
|
||||
func init() {
|
||||
cda.RegisterAnnotation(InjectTag, reflect.TypeOf((*Inject)(nil)))
|
||||
}
|
||||
|
||||
type Inject struct {
|
||||
cda.Annotation
|
||||
Name string `annotation-meta:"name"`
|
||||
}
|
29
injection/annotation/inject_test.go
Normal file
29
injection/annotation/inject_test.go
Normal file
|
@ -0,0 +1,29 @@
|
|||
package annotation
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestParseInject(t *testing.T) {
|
||||
// a1 := "@Inject"
|
||||
// a, err := ParseInject(a1)
|
||||
// if nil != err {
|
||||
// log.Printf("%v", err)
|
||||
// }
|
||||
// log.Printf("%v", a)
|
||||
|
||||
// a2 := "@Inject()"
|
||||
// a, err = ParseInject(a2)
|
||||
// if nil != err {
|
||||
// log.Printf("%v", err)
|
||||
// }
|
||||
// log.Printf("%v", a)
|
||||
|
||||
// a3 := "@Inject(name=string)"
|
||||
// a, err = ParseInject(a3)
|
||||
// if nil != err {
|
||||
// log.Printf("%v", err)
|
||||
// }
|
||||
// log.Printf("%v", a)
|
||||
|
||||
}
|
20
injection/annotation/resource.go
Normal file
20
injection/annotation/resource.go
Normal file
|
@ -0,0 +1,20 @@
|
|||
package annotation
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
|
||||
cda "git.loafle.net/commons/di-go/annotation"
|
||||
)
|
||||
|
||||
const (
|
||||
ResourceTag = "@Resource"
|
||||
)
|
||||
|
||||
func init() {
|
||||
cda.RegisterAnnotation(ResourceTag, reflect.TypeOf((*Resource)(nil)))
|
||||
}
|
||||
|
||||
type Resource struct {
|
||||
cda.Annotation
|
||||
Name string `annotation-meta:"name"`
|
||||
}
|
29
injection/annotation/resource_test.go
Normal file
29
injection/annotation/resource_test.go
Normal file
|
@ -0,0 +1,29 @@
|
|||
package annotation
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestParseResource(t *testing.T) {
|
||||
// a1 := "@Resource"
|
||||
// a, err := ParseResource(a1)
|
||||
// if nil != err {
|
||||
// log.Printf("%v", err)
|
||||
// }
|
||||
// log.Printf("%v", a)
|
||||
|
||||
// a2 := "@Resource( )"
|
||||
// a, err = ParseResource(a2)
|
||||
// if nil != err {
|
||||
// log.Printf("%v", err)
|
||||
// }
|
||||
// log.Printf("%v", a)
|
||||
|
||||
// a3 := "@Resource(name=string1232)"
|
||||
// a, err = ParseResource(a3)
|
||||
// if nil != err {
|
||||
// log.Printf("%v", err)
|
||||
// }
|
||||
// log.Printf("%v", a)
|
||||
|
||||
}
|
111
registry/definition.go
Normal file
111
registry/definition.go
Normal file
|
@ -0,0 +1,111 @@
|
|||
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
|
||||
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)
|
||||
}
|
313
registry/registry.go
Normal file
313
registry/registry.go
Normal file
|
@ -0,0 +1,313 @@
|
|||
package registry
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"reflect"
|
||||
"runtime"
|
||||
"strings"
|
||||
|
||||
cda "git.loafle.net/commons/di-go/annotation"
|
||||
cdia "git.loafle.net/commons/di-go/injection/annotation"
|
||||
cur "git.loafle.net/commons/util-go/reflect"
|
||||
)
|
||||
|
||||
func init() {
|
||||
registry = newRegistry()
|
||||
}
|
||||
|
||||
var registry ComponentRegistry
|
||||
|
||||
type ComponentRegistry interface {
|
||||
RegisterType(t reflect.Type)
|
||||
RegisterResource(name string, resource interface{}) error
|
||||
|
||||
GetInstance(t reflect.Type) (interface{}, error)
|
||||
GetInstances(ts []reflect.Type) ([]interface{}, error)
|
||||
GetInstanceByName(name string) (interface{}, error)
|
||||
GetInstancesByAnnotationName(n string) ([]interface{}, error)
|
||||
}
|
||||
|
||||
func newRegistry() ComponentRegistry {
|
||||
r := &defaultComponentRegistry{}
|
||||
r.definitionByType = make(map[reflect.Type]*TypeDefinition, 0)
|
||||
r.definitionByName = make(map[string]*TypeDefinition, 0)
|
||||
r.instanceByType = make(map[reflect.Type]interface{}, 0)
|
||||
r.resourceByName = make(map[string]interface{}, 0)
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
type defaultComponentRegistry struct {
|
||||
definitionByType map[reflect.Type]*TypeDefinition
|
||||
definitionByName map[string]*TypeDefinition
|
||||
instanceByType map[reflect.Type]interface{}
|
||||
resourceByName map[string]interface{}
|
||||
}
|
||||
|
||||
func RegisterType(t reflect.Type) {
|
||||
pc, _, _, ok := runtime.Caller(1)
|
||||
details := runtime.FuncForPC(pc)
|
||||
if ok && details != nil {
|
||||
log.Printf("called from %s", details.Name())
|
||||
}
|
||||
|
||||
registry.RegisterType(t)
|
||||
}
|
||||
func (cr *defaultComponentRegistry) RegisterType(t reflect.Type) {
|
||||
if nil == t {
|
||||
log.Panicf("DI: t[reflect.Type] is nil")
|
||||
}
|
||||
if !cur.IsTypeKind(t, reflect.Struct, true) {
|
||||
log.Panicf("DI: t[reflect.Type] must be specified but is %v", t)
|
||||
}
|
||||
|
||||
td, err := cr.buildDefinition(t)
|
||||
if nil != err {
|
||||
log.Panicf("DI: %v", err)
|
||||
}
|
||||
|
||||
if _, ok := cr.definitionByType[td.RealType]; ok {
|
||||
log.Panicf("DI: The type[%s] of Component is exist already", td.FullName)
|
||||
}
|
||||
cr.definitionByType[td.RealType] = td
|
||||
|
||||
name := td.TypeName
|
||||
|
||||
if a := td.GetAnnotationByType(reflect.TypeOf((*cdia.Component)(nil)), true); nil != a {
|
||||
ca := a.(*cdia.Component)
|
||||
if "" != strings.Trim(ca.Name, " ") {
|
||||
name = ca.Name
|
||||
}
|
||||
}
|
||||
|
||||
if eTD, ok := cr.definitionByName[name]; ok {
|
||||
log.Panicf("DI: The name[%s] of Component is exist already type[%s]", name, eTD.FullName)
|
||||
}
|
||||
cr.definitionByName[name] = td
|
||||
}
|
||||
|
||||
func RegisterResource(name string, resource interface{}) error {
|
||||
return registry.RegisterResource(name, resource)
|
||||
}
|
||||
func (cr *defaultComponentRegistry) RegisterResource(name string, resource interface{}) error {
|
||||
if _, ok := cr.resourceByName[name]; ok {
|
||||
return fmt.Errorf("DI: Resource[%s] is already exist", name)
|
||||
}
|
||||
cr.resourceByName[name] = resource
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetInstance returns instance of type
|
||||
// t must be pointer of struct
|
||||
func GetInstance(t reflect.Type) (interface{}, error) {
|
||||
return registry.GetInstance(t)
|
||||
}
|
||||
func (cr *defaultComponentRegistry) GetInstance(t reflect.Type) (interface{}, error) {
|
||||
if nil == t {
|
||||
return nil, fmt.Errorf("DI: t[reflect.Type] is nil")
|
||||
}
|
||||
|
||||
if reflect.Ptr != t.Kind() {
|
||||
return nil, fmt.Errorf("DI: t[reflect.Type] must be pointer of struct")
|
||||
}
|
||||
|
||||
if reflect.Struct != t.Elem().Kind() {
|
||||
return nil, fmt.Errorf("DI: t[reflect.Type] must be pointer of struct")
|
||||
}
|
||||
|
||||
var (
|
||||
td *TypeDefinition
|
||||
comV interface{}
|
||||
ok bool
|
||||
err error
|
||||
)
|
||||
|
||||
rt, _, _ := cur.GetTypeInfo(t)
|
||||
if td, ok = cr.definitionByType[rt]; !ok {
|
||||
if td, err = cr.buildDefinition(t); nil != err {
|
||||
return nil, fmt.Errorf("DI: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
if comV, ok = cr.instanceByType[td.RealType]; ok {
|
||||
return comV, nil
|
||||
}
|
||||
|
||||
v := reflect.New(rt)
|
||||
rv := v.Elem()
|
||||
|
||||
var (
|
||||
annotation cda.Annotation
|
||||
fV interface{}
|
||||
)
|
||||
|
||||
for _, fd := range td.Fields {
|
||||
f := rv.FieldByName(fd.FieldName)
|
||||
|
||||
if !f.IsValid() {
|
||||
return nil, fmt.Errorf("DI: Field[%s] is not valid", fd.FieldName)
|
||||
}
|
||||
if !f.CanSet() {
|
||||
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 fV, err = cr.GetInstance(fd.Type); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if annotation = fd.GetAnnotationByType(reflect.TypeOf((*cdia.Resource)(nil)), false); nil != annotation {
|
||||
n := annotation.(*cdia.Resource).Name
|
||||
if "" == n {
|
||||
n = fd.FieldName
|
||||
}
|
||||
if fV, err = cr.GetInstanceByName(n); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if nil != err {
|
||||
return nil, fmt.Errorf("DI: %v", err)
|
||||
}
|
||||
f.Set(reflect.ValueOf(fV))
|
||||
}
|
||||
|
||||
cr.instanceByType[td.RealType] = v.Interface()
|
||||
|
||||
return v.Interface(), nil
|
||||
}
|
||||
|
||||
func GetInstanceByName(name string) (interface{}, error) {
|
||||
return registry.GetInstanceByName(name)
|
||||
}
|
||||
func (cr *defaultComponentRegistry) GetInstanceByName(name string) (interface{}, error) {
|
||||
v, ok := cr.resourceByName[name]
|
||||
if ok {
|
||||
return v, nil
|
||||
}
|
||||
return nil, fmt.Errorf("DI: Resource[%s] is not exist", name)
|
||||
}
|
||||
|
||||
// GetInstances returns instance of annotated
|
||||
// n must be name of registered annotation
|
||||
func GetInstances(ts []reflect.Type) ([]interface{}, error) {
|
||||
return registry.GetInstances(ts)
|
||||
}
|
||||
func (cr *defaultComponentRegistry) GetInstances(ts []reflect.Type) ([]interface{}, error) {
|
||||
var (
|
||||
i interface{}
|
||||
err error
|
||||
)
|
||||
instances := make([]interface{}, 0)
|
||||
for _, t := range ts {
|
||||
if i, err = cr.GetInstance(t); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
instances = append(instances, i)
|
||||
}
|
||||
|
||||
return instances, nil
|
||||
}
|
||||
|
||||
// GetInstancesByAnnotationName returns instance of annotated
|
||||
// n must be name of registered annotation
|
||||
func GetInstancesByAnnotationName(n string) ([]interface{}, error) {
|
||||
return registry.GetInstancesByAnnotationName(n)
|
||||
}
|
||||
func (cr *defaultComponentRegistry) GetInstancesByAnnotationName(n string) ([]interface{}, error) {
|
||||
var (
|
||||
i interface{}
|
||||
err error
|
||||
)
|
||||
instances := make([]interface{}, 0)
|
||||
|
||||
for _, td := range cr.definitionByType {
|
||||
if nil != td.GetAnnotation(n) {
|
||||
if i, err = cr.GetInstance(td.Type); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
instances = append(instances, i)
|
||||
}
|
||||
}
|
||||
|
||||
return instances, nil
|
||||
}
|
||||
|
||||
func (cr *defaultComponentRegistry) buildDefinition(t reflect.Type) (*TypeDefinition, error) {
|
||||
if nil == t {
|
||||
return nil, fmt.Errorf("t[reflect.Type] is nil")
|
||||
}
|
||||
|
||||
rt, pkgName, tName := cur.GetTypeInfo(t)
|
||||
td := &TypeDefinition{}
|
||||
td.FullName = FullName(pkgName, tName)
|
||||
td.PkgName = pkgName
|
||||
td.TypeName = tName
|
||||
td.Type = t
|
||||
td.RealType = rt
|
||||
td.Fields = make([]*FieldDefinition, 0)
|
||||
|
||||
parseFields(rt, td)
|
||||
|
||||
return td, nil
|
||||
}
|
||||
|
||||
func parseFields(t reflect.Type, td *TypeDefinition) {
|
||||
// fields := make([]*FieldDefinition, 0)
|
||||
rt, _, _ := cur.GetTypeInfo(t)
|
||||
if reflect.Struct != rt.Kind() {
|
||||
return
|
||||
}
|
||||
|
||||
for i := 0; i < rt.NumField(); i++ {
|
||||
f := rt.Field(i)
|
||||
|
||||
if f.Anonymous {
|
||||
if parseAnonymousField(f, td) {
|
||||
continue
|
||||
}
|
||||
parseFields(f.Type, td)
|
||||
} else {
|
||||
as, err := cda.ParseAnnotation(f.Tag)
|
||||
if nil != err {
|
||||
return
|
||||
}
|
||||
if nil != as && 0 < len(as) {
|
||||
fRT, fPkgName, fTName := cur.GetTypeInfo(f.Type)
|
||||
|
||||
fd := &FieldDefinition{
|
||||
FieldName: f.Name,
|
||||
PkgName: fPkgName,
|
||||
TypeName: fTName,
|
||||
Type: f.Type,
|
||||
RealType: fRT,
|
||||
Annotations: as,
|
||||
}
|
||||
td.Fields = append(td.Fields, fd)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func parseAnonymousField(f reflect.StructField, td *TypeDefinition) bool {
|
||||
ct := reflect.TypeOf((*cda.TypeAnnotation)(nil)).Elem()
|
||||
|
||||
if f.Type != ct {
|
||||
return false
|
||||
}
|
||||
|
||||
as, err := cda.ParseAnnotation(f.Tag)
|
||||
if nil != err {
|
||||
return false
|
||||
}
|
||||
|
||||
if nil != as && 0 < len(as) {
|
||||
td.TypeAnnotations = as
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
68
registry/registry_test.go
Normal file
68
registry/registry_test.go
Normal file
|
@ -0,0 +1,68 @@
|
|||
package registry
|
||||
|
||||
import (
|
||||
"log"
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"git.loafle.net/commons/di-go/annotation"
|
||||
cdia "git.loafle.net/commons/di-go/injection/annotation"
|
||||
)
|
||||
|
||||
func TestRegisterType(t *testing.T) {
|
||||
var (
|
||||
err error
|
||||
css []interface{}
|
||||
cs interface{}
|
||||
)
|
||||
RegisterType(reflect.TypeOf((*AService)(nil)))
|
||||
RegisterType(reflect.TypeOf((*BService)(nil)))
|
||||
// RegisterType(reflect.TypeOf((*TestStruct2)(nil)), &cda.ComponentAnnotation{
|
||||
// Name: "test1",
|
||||
// })
|
||||
// RegisterType(reflect.TypeOf((*TestStruct3)(nil)), &cda.ComponentAnnotation{
|
||||
// Name: "test2",
|
||||
// })
|
||||
|
||||
// fs := getFields(reflect.TypeOf((*TestStruct3)(nil)))
|
||||
// log.Printf("%v", fs)
|
||||
|
||||
RegisterResource("List", []string{"dfdkf", "skgkfg"})
|
||||
|
||||
if css, err = GetInstancesByAnnotationName(cdia.ComponentTag); nil != err {
|
||||
log.Printf("%v \n", err)
|
||||
}
|
||||
log.Printf("%v", css)
|
||||
|
||||
if css, err = GetInstances([]reflect.Type{
|
||||
reflect.TypeOf((*AService)(nil)),
|
||||
reflect.TypeOf((*BService)(nil)),
|
||||
}); nil != err {
|
||||
log.Printf("%v \n", err)
|
||||
}
|
||||
log.Printf("%v", css)
|
||||
|
||||
if cs, err = GetInstance(reflect.TypeOf((*CService)(nil))); nil != err {
|
||||
log.Printf("%v \n", err)
|
||||
}
|
||||
log.Printf("%v", cs.(*CService))
|
||||
|
||||
}
|
||||
|
||||
type AService struct {
|
||||
annotation.TypeAnnotation `annotation:"@Component(name=dkdkdf)"`
|
||||
|
||||
NameA string
|
||||
}
|
||||
|
||||
type BService struct {
|
||||
annotation.TypeAnnotation `annotation:"@Component()"`
|
||||
|
||||
NameB string
|
||||
}
|
||||
|
||||
type CService struct {
|
||||
AService *AService `annotation:"@Inject()"`
|
||||
BService *BService `annotation:"@Inject()"`
|
||||
List []string `annotation:"@Resource()"`
|
||||
}
|
Loading…
Reference in New Issue
Block a user