MES / IOT · Files

Blame view

sys/Hh.Mes.Service/Repository/StartUpExtenions.cs 3.15 KB
Edit Raw Normal View History
----
5c8a582c
 
赖素文 authored 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
using Microsoft.Extensions.DependencyInjection;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;

namespace Hh.Mes.Service.Repository
{
    /// <summary>
    /// DI实现批量注册
    /// </summary>
    public static class StartUpExtenions
    {
        /// <summary>
        /// 批量注册服务
        /// </summary>
        /// <param name="services">DI服务</param>
        /// <param name="assemblys">需要批量注册的程序集集合 如果有过多头号隔开</param>
        /// <param name="suffix"> 业务都有固定的后缀，可以自己定义命名规则，例如XXXReository,  XXXService等</param>
        /// <param name="serviceLifetime">服务生命周期</param>
        /// <returns></returns>
        public static IServiceCollection BatchRegisterService(this IServiceCollection services, string assemblys, string suffix, ServiceLifetime serviceLifetime = ServiceLifetime.Scoped)
        {
            if (string.IsNullOrEmpty(assemblys)) return null;
            //待注册集合
调整服务类
fc956062
 
唐召明 authored 
26
 
            var typeDic = GetImpleAndInterfaces(assemblys, suffix);
----
5c8a582c
 
赖素文 authored 
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
            //var index = 0;
            //foreach (var temp in typeDic)
            //{
            //    foreach (var v in temp)
            //    {
            //        index++;
            //        Console.WriteLine($"class={v.FullName}");
            //    }
            //}
            //Console.WriteLine($"{suffix}接口及实现个数为：{index}");
            if (!typeDic.Any()) return services;

            foreach (var instanceType in typeDic)
            {
                foreach (var tempClass in instanceType)
                {
                    //根据指定的生命周期进行注册
                    switch (serviceLifetime)
                    {
                        case ServiceLifetime.Scoped:
                            services.AddScoped(tempClass);
                            break;
                        case ServiceLifetime.Singleton:
                            services.AddSingleton(tempClass);
                            break;
                        case ServiceLifetime.Transient:
                            services.AddTransient(tempClass);
                            break;
                    }
                }
            }
            return services;
        }


        /// <summary>
        /// 载入指定的DLL（根据DLL名字载入）
        /// </summary>
        /// <param name="assemblyName"></param>
        /// <param name="suffix"></param>
        /// <returns></returns>
        private static List<List<Type>> GetImpleAndInterfaces(string assemblyName, string suffix)
        {
            var result = new List<List<Type>>();
调整服务类
fc956062
 
唐召明 authored 
71
 
            var assemblyArr = assemblyName.Split(new[] { ',' }, StringSplitOptions.RemoveEmptyEntries);
----
5c8a582c
 
赖素文 authored 
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 

            foreach (var temp in assemblyArr)
            {
                Assembly assembly = Assembly.Load(temp);
                var types = assembly
                    .GetTypes()
                    .Where(x => x.IsClass && x.Name.Contains(suffix))
                    .ToList();

                result.Add(types);
            }
            return result;
        }

    }
}