添加项目文件。

Entities/Context/UnitOfWork/UnitOfWorkServiceCollectionExtensions.cs

@@ -0,0 +1,118 @@
+
+
+namespace SharedDATA.Api
+{
+    using Microsoft.EntityFrameworkCore;
+    using Microsoft.Extensions.DependencyInjection;
+
+    /// <summary>
+    /// Extension methods for setting up unit of work related services in an <see cref="IServiceCollection"/>.
+    /// </summary>
+    public static class UnitOfWorkServiceCollectionExtensions
+    {
+        /// <summary>
+        /// Registers the unit of work given context as a service in the <see cref="IServiceCollection"/>.
+        /// </summary>
+        /// <typeparam name="TContext">The type of the db context.</typeparam>
+        /// <param name="services">The <see cref="IServiceCollection"/> to add services to.</param>
+        /// <returns>The same service collection so that multiple calls can be chained.</returns>
+        /// <remarks>
+        /// This method only support one db context, if been called more than once, will throw exception.
+        /// </remarks>
+        public static IServiceCollection AddUnitOfWork<TContext>(this IServiceCollection services) where TContext : DbContext
+        {
+            services.AddScoped<IRepositoryFactory, UnitOfWork<TContext>>();
+            // Following has a issue: IUnitOfWork cannot support multiple dbcontext/database, 
+            // that means cannot call AddUnitOfWork<TContext> multiple times.
+            // Solution: check IUnitOfWork whether or null
+            services.AddScoped<IUnitOfWork, UnitOfWork<TContext>>();
+            services.AddScoped<IUnitOfWork<TContext>, UnitOfWork<TContext>>();
+
+            return services;
+        }
+
+        /// <summary>
+        /// Registers the unit of work given context as a service in the <see cref="IServiceCollection"/>.
+        /// </summary>
+        /// <typeparam name="TContext1">The type of the db context.</typeparam>
+        /// <typeparam name="TContext2">The type of the db context.</typeparam>
+        /// <param name="services">The <see cref="IServiceCollection"/> to add services to.</param>
+        /// <returns>The same service collection so that multiple calls can be chained.</returns>
+        /// <remarks>
+        /// This method only support one db context, if been called more than once, will throw exception.
+        /// </remarks>
+        public static IServiceCollection AddUnitOfWork<TContext1, TContext2>(this IServiceCollection services)
+            where TContext1 : DbContext
+            where TContext2 : DbContext
+        {
+            services.AddScoped<IUnitOfWork<TContext1>, UnitOfWork<TContext1>>();
+            services.AddScoped<IUnitOfWork<TContext2>, UnitOfWork<TContext2>>();
+
+            return services;
+        }
+
+        /// <summary>
+        /// Registers the unit of work given context as a service in the <see cref="IServiceCollection"/>.
+        /// </summary>
+        /// <typeparam name="TContext1">The type of the db context.</typeparam>
+        /// <typeparam name="TContext2">The type of the db context.</typeparam>
+        /// <typeparam name="TContext3">The type of the db context.</typeparam>
+        /// <param name="services">The <see cref="IServiceCollection"/> to add services to.</param>
+        /// <returns>The same service collection so that multiple calls can be chained.</returns>
+        /// <remarks>
+        /// This method only support one db context, if been called more than once, will throw exception.
+        /// </remarks>
+        public static IServiceCollection AddUnitOfWork<TContext1, TContext2, TContext3>(this IServiceCollection services)
+            where TContext1 : DbContext
+            where TContext2 : DbContext
+            where TContext3 : DbContext
+        {
+            services.AddScoped<IUnitOfWork<TContext1>, UnitOfWork<TContext1>>();
+            services.AddScoped<IUnitOfWork<TContext2>, UnitOfWork<TContext2>>();
+            services.AddScoped<IUnitOfWork<TContext3>, UnitOfWork<TContext3>>();
+
+            return services;
+        }
+
+        /// <summary>
+        /// Registers the unit of work given context as a service in the <see cref="IServiceCollection"/>.
+        /// </summary>
+        /// <typeparam name="TContext1">The type of the db context.</typeparam>
+        /// <typeparam name="TContext2">The type of the db context.</typeparam>
+        /// <typeparam name="TContext3">The type of the db context.</typeparam>
+        /// <typeparam name="TContext4">The type of the db context.</typeparam>
+        /// <param name="services">The <see cref="IServiceCollection"/> to add services to.</param>
+        /// <returns>The same service collection so that multiple calls can be chained.</returns>
+        /// <remarks>
+        /// This method only support one db context, if been called more than once, will throw exception.
+        /// </remarks>
+        public static IServiceCollection AddUnitOfWork<TContext1, TContext2, TContext3, TContext4>(this IServiceCollection services)
+            where TContext1 : DbContext
+            where TContext2 : DbContext
+            where TContext3 : DbContext
+            where TContext4 : DbContext
+        {
+            services.AddScoped<IUnitOfWork<TContext1>, UnitOfWork<TContext1>>();
+            services.AddScoped<IUnitOfWork<TContext2>, UnitOfWork<TContext2>>();
+            services.AddScoped<IUnitOfWork<TContext3>, UnitOfWork<TContext3>>();
+            services.AddScoped<IUnitOfWork<TContext4>, UnitOfWork<TContext4>>();
+
+            return services;
+        }
+
+        /// <summary>
+        /// Registers the custom repository as a service in the <see cref="IServiceCollection"/>.
+        /// </summary>
+        /// <typeparam name="TEntity">The type of the entity.</typeparam>
+        /// <typeparam name="TRepository">The type of the custom repositry.</typeparam>
+        /// <param name="services">The <see cref="IServiceCollection"/> to add services to.</param>
+        /// <returns>The same service collection so that multiple calls can be chained.</returns>
+        public static IServiceCollection AddCustomRepository<TEntity, TRepository>(this IServiceCollection services)
+            where TEntity : class
+            where TRepository : class, IRepository<TEntity>
+        {
+            services.AddScoped<IRepository<TEntity>, TRepository>();
+            return services;
+        }
+    }
+}