di-go/registry/registry.go

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)
	GetInstancesByAnnotationType(t reflect.Type) ([]interface{}, error)

	GetTypeAnnotation(instanceType reflect.Type, annotationType reflect.Type) cda.Annotation
	GetMethodAnnotation(instanceType reflect.Type, annotationType reflect.Type, methodName string) cda.Annotation
	GetMethodAnnotations(instanceType reflect.Type, annotationType reflect.Type) map[string]cda.Annotation
}

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.GetTypeAnnotationByType(cdia.ComponentAnnotationType, true); nil != a {
		ca := a.(*cdia.ComponentAnnotation)
		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(cdia.InjectAnnotationType, false); nil != annotation {
			if fV, err = cr.GetInstance(fd.Type); nil != err {
				return nil, err
			}
		}

		if annotation = fd.GetAnnotationByType(cdia.ResourceAnnotationType, false); nil != annotation {
			n := annotation.(*cdia.ResourceAnnotation).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
}

func GetInstancesByAnnotationType(t reflect.Type) ([]interface{}, error) {
	return registry.GetInstancesByAnnotationType(t)
}
func (cr *defaultComponentRegistry) GetInstancesByAnnotationType(t reflect.Type) ([]interface{}, error) {
	var (
		i   interface{}
		err error
	)
	instances := make([]interface{}, 0)

	for _, td := range cr.definitionByType {
		if nil != td.GetTypeAnnotationByType(t, true) {
			if i, err = cr.GetInstance(td.Type); nil != err {
				return nil, err
			}
			instances = append(instances, i)
		}
	}

	return instances, nil
}

func GetTypeAnnotation(instanceType reflect.Type, annotationType reflect.Type) cda.Annotation {
	return registry.GetTypeAnnotation(instanceType, annotationType)
}
func (cr *defaultComponentRegistry) GetTypeAnnotation(instanceType reflect.Type, annotationType reflect.Type) cda.Annotation {
	def, ok := cr.definitionByType[instanceType]
	if !ok {
		return nil
	}

	return def.GetTypeAnnotationByType(annotationType, false)
}
func GetMethodAnnotation(instanceType reflect.Type, annotationType reflect.Type, methodName string) cda.Annotation {
	return registry.GetMethodAnnotation(instanceType, annotationType, methodName)
}
func (cr *defaultComponentRegistry) GetMethodAnnotation(instanceType reflect.Type, annotationType reflect.Type, methodName string) cda.Annotation {
	rt, _, _ := cur.GetTypeInfo(instanceType)

	def, ok := cr.definitionByType[rt]
	if !ok {
		return nil
	}

	return def.GetMethodAnnotationByType(annotationType, methodName)
}

func GetMethodAnnotations(instanceType reflect.Type, annotationType reflect.Type) map[string]cda.Annotation {
	return registry.GetMethodAnnotations(instanceType, annotationType)
}
func (cr *defaultComponentRegistry) GetMethodAnnotations(instanceType reflect.Type, annotationType reflect.Type) map[string]cda.Annotation {
	rt, _, _ := cur.GetTypeInfo(instanceType)

	def, ok := cr.definitionByType[rt]
	if !ok {
		return nil
	}

	return def.GetMethodAnnotationsByType(annotationType)
}

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 {
			if parseMethodAnnotation(&f, td) {
				continue
			}
			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 {
	if parseTypeAnnotation(f, td) {
		return true
	}
	return false
}

func parseTypeAnnotation(f *reflect.StructField, td *TypeDefinition) bool {
	// if !haveEmbeddingOf(cda.TypeAnnotationType, f.Type) {
	// 	return
	// }
	if cda.TypeAnnotationType != f.Type {
		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
}

func parseMethodAnnotation(f *reflect.StructField, td *TypeDefinition) bool {
	if cda.MethodAnnotationType != f.Type {
		return false
	}

	as, err := cda.ParseAnnotation(f.Tag)
	if nil != err {
		return false
	}

	if nil != as && 0 < len(as) {
		if nil == td.MethodAnnotations {
			td.MethodAnnotations = make(map[string]map[reflect.Type]cda.Annotation, 0)
		}
		td.MethodAnnotations[f.Name[1:]] = as
		return true
	}
	return false
}

func haveEmbeddingOf(t reflect.Type, target reflect.Type) bool {
	if t == target {
		return true
	}

	rt, _, _ := cur.GetTypeInfo(target)
	if reflect.Struct != rt.Kind() {
		return false
	}

	for i := 0; i < rt.NumField(); i++ {
		f := rt.Field(i)

		if f.Anonymous {
			if haveEmbeddingOf(t, f.Type) {
				return true
			}
		}
	}
	return false
}