コード例 #1
0
ファイル: ContainerExtentions.cs プロジェクト: shuk/Cortoxa
//        public static IToolContainer InstallData<T>(this IToolContainer container, Action<T> action) where T : DataAccess, new ()
//        {
////            container.Register(r => r.Service<T>().ToSelf().With(container));
////            var builder = container.Resolve<T>();
//            var builder = (T)Activator.CreateInstance(typeof(T), container);
//            action(builder);
//            builder.Build();
//            return container;
//        }

        public static IToolContainer InstallInjections(this IToolContainer container, Assembly assembly = null)
        {
            if (assembly == null)
            {
                assembly = Assembly.GetCallingAssembly();
            }

            var attributeType = typeof(InjectAttribte);
            var types         = assembly.GetTypes().Where(t => Attribute.IsDefined(t, attributeType)).ToArray();

            foreach (var type in types)
            {
                var attribute = type.GetCustomAttributes(attributeType).First() as InjectAttribte;
                if (attribute != null)
                {
                    var  intefaces     = type.GetInterfaces();
                    Type injectionType = type;
                    if (injectionType.IsClass)
                    {
                        if (intefaces.Any())
                        {
                            var serviceType = intefaces.FirstOrDefault();
                            container.Register(r => r.For(serviceType).To(injectionType).LifeTime(attribute.LifeTime));
                        }
                    }

                    if (injectionType.IsInterface)
                    {
                        container.Register(r => r.For(injectionType).ToSelf().LifeTime(attribute.LifeTime));
                    }
                }
            }
            return(container);
        }