initial commit

2025-12-29 19:05:47 +08:00 · 2024-11-13 18:18:28 +08:00
commit 013f35e296
1500 changed files with 443723 additions and 0 deletions
--- a/FreeSql.DbContext/Extensions/DependencyInjection.cs
+++ b/FreeSql.DbContext/Extensions/DependencyInjection.cs
@@ -0,0 +1,56 @@
+#if netcoreapp
+using FreeSql;
+using System;
+using System.Linq;
+using System.Reflection;
+
+namespace Microsoft.Extensions.DependencyInjection
+{
+    public static class FreeSqlDbContextDependencyInjection
+    {
+        static IServiceCollection AddFreeDbContext(this IServiceCollection services, Type dbContextType, Action<DbContextOptionsBuilder> options)
+        {
+            services.AddScoped(dbContextType, sp =>
+            {
+                DbContext ctx = null;
+                try
+                {
+                    var ctor = dbContextType. GetConstructors().FirstOrDefault();
+                    var ctorParams = ctor.GetParameters().Select(a => sp.GetService(a.ParameterType)).ToArray();
+                    ctx = Activator.CreateInstance(dbContextType, ctorParams) as DbContext;
+                }
+                catch(Exception ex)
+                {
+                    throw new Exception(DbContextStrings.AddFreeDbContextError_CheckConstruction(dbContextType.Name), ex);
+                }
+                if (ctx != null && ctx._ormScoped == null)
+                {
+                    var builder = new DbContextOptionsBuilder();
+                    options(builder);
+                    ctx._ormScoped = DbContextScopedFreeSql.Create(builder._fsql, () => ctx, () => ctx.UnitOfWork);
+                    ctx._optionsPriv = builder._options;
+
+                    if (ctx._ormScoped == null)
+                        throw new Exception(DbContextStrings.ConfigureUseFreeSql);
+
+                    ctx.InitPropSets();
+                }
+                return ctx;
+            });
+            return services;
+        }
+
+        public static IServiceCollection AddFreeDbContext<TDbContext>(this IServiceCollection services, Action<DbContextOptionsBuilder> options) where TDbContext : DbContext => 
+            AddFreeDbContext(services, typeof(TDbContext), options);
+
+        public static IServiceCollection AddFreeDbContext(this IServiceCollection services, Action<DbContextOptionsBuilder> options, Assembly[] assemblies)
+        {
+            if (assemblies?.Any() == true)
+                foreach (var asse in assemblies)
+                    foreach (var dbType in asse.GetTypes().Where(a => a.IsAbstract == false && typeof(DbContext).IsAssignableFrom(a)))
+                        AddFreeDbContext(services, dbType, options);
+            return services;
+        }
+    }
+}
+#endif