diff --git a/Admin.NET/Admin.NET.Application/Startup.cs b/Admin.NET/Admin.NET.Application/Startup.cs index 51be1c52..4847c13a 100644 --- a/Admin.NET/Admin.NET.Application/Startup.cs +++ b/Admin.NET/Admin.NET.Application/Startup.cs @@ -41,6 +41,16 @@ public class Startup : AppStartup { // 比较版本号对数据库进行升级结构、种子数据等 } + /// + /// 构建 WebApplication 对象过程中装载中间件 + /// + /// + /// + /// + public void LoadAppComponent(IApplicationBuilder app, IWebHostEnvironment env, ComponentContext componentContext) + { + WebApplication application = app as WebApplication; + } } /// diff --git a/Admin.NET/Admin.NET.Core/SqlSugar/SqlSugarSetup.cs b/Admin.NET/Admin.NET.Core/SqlSugar/SqlSugarSetup.cs index 9534a09f..1cdce632 100644 --- a/Admin.NET/Admin.NET.Core/SqlSugar/SqlSugarSetup.cs +++ b/Admin.NET/Admin.NET.Core/SqlSugar/SqlSugarSetup.cs @@ -429,7 +429,7 @@ public static class SqlSugarSetup { // 扫描所有继承 AppStartup 的类(排序执行顺序) var startups = App.EffectiveTypes - .Where(u => typeof(AppStartup).IsAssignableFrom(u) && u.IsClass && !u.IsAbstract && !u.IsGenericType && u.GetMethod("BeforeInitTable") != null) + .Where(u => typeof(AppStartup).IsAssignableFrom(u) && u.IsClass && !u.IsAbstract && !u.IsGenericType && (u.GetMethod("BeforeInitTable") != null || u.GetMethod("AfterInitSeed") != null)) .OrderByDescending(u => !u.IsDefined(typeof(AppStartupAttribute), true) ? 0 : u.GetCustomAttribute(true).Order); if (startups == null || !startups.Any()) return (startups, 0, 0); diff --git a/Admin.NET/Admin.NET.Web.Entry/Program.cs b/Admin.NET/Admin.NET.Web.Entry/Program.cs index 8c672803..0713af24 100644 --- a/Admin.NET/Admin.NET.Web.Entry/Program.cs +++ b/Admin.NET/Admin.NET.Web.Entry/Program.cs @@ -5,8 +5,10 @@ // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任! using Microsoft.AspNetCore.Http.Features; +using Furion; +using System.Reflection; -Serve.Run(RunOptions.Default.AddWebComponent()); +Serve.Run(RunOptions.Default.AddWebComponent().UseComponent()); public class WebComponent : IWebComponent { @@ -31,4 +33,35 @@ public class WebComponent : IWebComponent options.MultipartBodyLengthLimit = 1073741824; // 限制大小1GB(默认128MB) }); } +} +public class ApplicationComponent : IApplicationComponent +{ + /// + /// 构建 WebApplication 对象过程中装载中间件 + /// + /// + /// + /// + public void Load(IApplicationBuilder app, IWebHostEnvironment env, ComponentContext componentContext) + { + //WebApplication application = app as WebApplication; + // 扫描所有继承 AppStartup 的类(排序执行顺序) + var startups = App.EffectiveTypes + .Where(u => typeof(AppStartup).IsAssignableFrom(u) && u.IsClass && !u.IsAbstract && !u.IsGenericType && u.GetMethod("LoadAppComponent") != null) + .OrderByDescending(u => !u.IsDefined(typeof(AppStartupAttribute), true) ? 0 : u.GetCustomAttribute(true).Order); + if (startups == null || !startups.Any()) + return; + try + { + foreach (var type in startups) + { + var startup = Activator.CreateInstance(type) as AppStartup; + var initDataMethod = type.GetMethod("LoadAppComponent"); + initDataMethod?.Invoke(startup, [app, env, componentContext]); + } + } + catch (Exception ex) + { + } + } } \ No newline at end of file