From 2bcaf6375de07a39fe309feb2c5216880a89ef52 Mon Sep 17 00:00:00 2001
From: FunCoder <qzzdy@foxmail.com>
Date: Tue, 22 Apr 2025 16:01:08 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9C=A8=E6=9E=84=E5=BB=BAWebApplication?=
 =?UTF-8?q?=E5=AF=B9=E8=B1=A1=E8=BF=87=E7=A8=8B=E4=B8=AD=E8=A3=85=E8=BD=BD?=
 =?UTF-8?q?=E4=B8=AD=E9=97=B4=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Admin.NET/Admin.NET.Application/Startup.cs    | 10 ++++++
 .../Admin.NET.Core/SqlSugar/SqlSugarSetup.cs  |  2 +-
 Admin.NET/Admin.NET.Web.Entry/Program.cs      | 35 ++++++++++++++++++-
 3 files changed, 45 insertions(+), 2 deletions(-)

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
     {
         // 比较版本号对数据库进行升级结构、种子数据等
     }
+    /// <summary>
+    /// 构建 WebApplication 对象过程中装载中间件
+    /// </summary>
+    /// <param name="app"></param>
+    /// <param name="env"></param>
+    /// <param name="componentContext"></param>
+    public void LoadAppComponent(IApplicationBuilder app, IWebHostEnvironment env, ComponentContext componentContext)
+    {
+        WebApplication application = app as WebApplication;
+    }
 }
 
 /// <summary>
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<AppStartupAttribute>(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<WebComponent>());
+Serve.Run(RunOptions.Default.AddWebComponent<WebComponent>().UseComponent<ApplicationComponent>());
 
 public class WebComponent : IWebComponent
 {
@@ -31,4 +33,35 @@ public class WebComponent : IWebComponent
             options.MultipartBodyLengthLimit = 1073741824; // 限制大小1GB（默认128MB）
         });
     }
+}
+public class ApplicationComponent : IApplicationComponent
+{
+    /// <summary>
+    /// 构建 WebApplication 对象过程中装载中间件
+    /// </summary>
+    /// <param name="app"></param>
+    /// <param name="env"></param>
+    /// <param name="componentContext"></param>
+    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<AppStartupAttribute>(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