From 219bb2f0c9d1bb9412c8190d958aeb179b7abdb5 Mon Sep 17 00:00:00 2001
From: crusader <crusader@loafle.com>
Date: Thu, 15 Mar 2018 23:32:45 +0900
Subject: [PATCH] ing

---
 registry/definition.go | 45 ++++++++++++++++++++++++++++++++++++++++++
 registry/registry.go   | 13 ++++++------
 2 files changed, 51 insertions(+), 7 deletions(-)

diff --git a/registry/definition.go b/registry/definition.go
index 0719b10..6f468b4 100644
--- a/registry/definition.go
+++ b/registry/definition.go
@@ -5,6 +5,7 @@ import (
 	"reflect"
 
 	cda "git.loafle.net/commons_go/di/annotation"
+	cdur "git.loafle.net/commons_go/di/util/reflect"
 )
 
 type TypeDefinition struct {
@@ -22,9 +23,53 @@ 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 includeEmbedding {
+			if checkAnnotation(reflect.TypeOf(v), at) {
+				return v
+			}
+		} else {
+			if at == reflect.TypeOf(v) {
+				return v
+			}
+		}
+
+	}
+
+	return nil
+}
+
+func checkAnnotation(t reflect.Type, st reflect.Type) bool {
+	rt, _, _ := cdur.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
diff --git a/registry/registry.go b/registry/registry.go
index 6fc00a4..2dbbba7 100644
--- a/registry/registry.go
+++ b/registry/registry.go
@@ -73,14 +73,13 @@ func (cr *defaultComponentRegistry) RegisterType(t reflect.Type) {
 	}
 	cr.definitionByType[td.RealType] = td
 
-	// _ca := ca.(*cdia.ComponentAnnotation)
-	ca := td.GetAnnotation(cdia.ComponentTag).(*cdia.Component)
+	name := td.TypeName
 
-	name := ""
-	if nil != ca && "" != strings.Trim(ca.Name, " ") {
-		name = ca.Name
-	} else {
-		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 {