From 7bf89fd93bc487e8445d6d85bb527251b7371a04 Mon Sep 17 00:00:00 2001
From: alielie <alielie@git.com>
Date: 星期一, 12 五月 2025 13:14:11 +0800
Subject: [PATCH] 优化项目；
---
HMI.Base/PluginLoader.cs | 245 +++++++++++++++++++++++++++++++++---------------
1 files changed, 168 insertions(+), 77 deletions(-)
diff --git a/HMI.Base/PluginLoader.cs b/HMI.Base/PluginLoader.cs
index b3a4952..6b4ba0b 100644
--- a/HMI.Base/PluginLoader.cs
+++ b/HMI.Base/PluginLoader.cs
@@ -1,32 +1,55 @@
-锘縰sing System;
+锘縰sing HMI.Core.Api;
+using HMI.Core.Entity;
+using HMI.Core.Util;
+using log4net;
+using System;
using System.Collections.Generic;
-using System.Collections.ObjectModel;
using System.Data;
using System.IO;
using System.Reflection;
-using System.Windows.Media.Imaging;
-using HMI.Core.Api;
-using HMI.Core.Util;
-using log4net;
+using System.Threading;
+using System.Windows;
+using System.Windows.Threading;
namespace HMI.Base
{
+ public enum LoadMsgType
+ {
+ INIT, LOADING, FINISHED
+ }
+
+ public class LoadMsg
+ {
+ public string Message { get; set; }
+ public bool? Result { get; set; }
+ public LoadMsgType Type { get; set; }
+ public int NumData { get; set; }
+ }
+
public class PluginLoader
{
+ private static ILog log = LogManager.GetLogger("PluginLoader");
private static string PluginFolder = PathHelper.PluginFolder();
public bool Completed { get; private set; } = false;
public SqliteHelper sqliteHelper;
- public ObservableCollection<PluginInfo> Menus { get; private set; }
- public ObservableCollection<PluginInfo> Plugins { get; private set; }
-
+ public List<PluginInfo> Menus { get; private set; }
+ public List<PluginInfo> Plugins { get; private set; }
public static PluginLoader Instance { get; private set; }
- public event EventHandler<string> LoadMessageAppend;
+ public event EventHandler<LoadMsg> LoadMessageAppend;
private PluginLoader()
{
sqliteHelper = new SqliteHelper(PathHelper.DatabasePath);
+ }
+
+ public static void Reset()
+ {
+ if (Instance != null)
+ {
+ Instance = null;
+ }
}
public static PluginLoader GetLoader()
@@ -38,95 +61,163 @@
return Instance;
}
- public void LoadMenuAndPlugin()
+ private IPluginContainer pluginContainer;
+
+ public void LoadMenuAndPlugin(IPluginContainer container, Dispatcher dispatcher)
{
+ pluginContainer = container;
if (Menus == null)
{
- Plugins = new ObservableCollection<PluginInfo>();
- Menus = LoadTreeMenu();
- }
- }
-
- private void LoadChildrenMenu(PluginInfo menu)
- {
- LoadMessageAppend?.Invoke(this, $"姝ｅ湪鍔犺浇{menu.Name}鑿滃崟...");
- try
- {
- if (!string.IsNullOrEmpty(menu.DllPath))
+ Plugins = new List<PluginInfo>();
+ Menus = LoadTreeMenu(dispatcher);
+ sqliteHelper.Close();
+ dispatcher.Invoke(() =>
{
- string dllPath = Path.Combine(PluginFolder, menu.DllPath);
- Assembly ass = Assembly.LoadFrom(dllPath);
- Type ClassType = ass.GetType(menu.ClassName);
- object Instance = Activator.CreateInstance(ClassType);
- menu.Content = Instance;
- if (Instance is IPlugin plugin)
- {
- plugin.SetLogger(LogManager.GetLogger(menu.Code));
- string configFolder = Path.Combine(PathHelper.ConfigFolder, menu.Code);
- if (!Directory.Exists(configFolder))
- {
- Directory.CreateDirectory(configFolder);
- }
- plugin.SetBaseFolder(configFolder);
- plugin.InitFinish();
- }
- Type SettingType = ass.GetType($"{menu.ClassName}_Setting");
- if (SettingType != null)
- {
- object SettingInstance = Activator.CreateInstance(SettingType);
- menu.SettingControl = SettingInstance;
- }
- }
- }
- catch (Exception ex)
- {
- LoadMessageAppend?.Invoke(this, $"{ex.Message} {ex.StackTrace}");
+ LoadMessageAppend?.Invoke(this, new LoadMsg { Type = LoadMsgType.FINISHED });
+ });
}
}
- public ObservableCollection<PluginInfo> LoadTreeMenu()
+ private void LoadChildrenMenu()
{
- ObservableCollection<PluginInfo> data = new ObservableCollection<PluginInfo>();
+ foreach (var menu in Plugins)
+ {
+ CountdownEvent c = new CountdownEvent(1);
+ Thread.Sleep(80);
+ Application.Current.Dispatcher.Invoke(new Action(() =>
+ {
+ try
+ {
+ LoadMessageAppend?.Invoke(this,
+ new LoadMsg
+ {
+ Type = LoadMsgType.LOADING,
+ Message = this.lang("App.IsLoadingMenu", menu.Name)
+ }
+ );
+ if (!string.IsNullOrEmpty(menu.DllPath))
+ {
+ string dllPath = Path.Combine(PluginFolder, menu.DllPath);
+ Assembly ass = Assembly.LoadFrom(dllPath);
+ var tss = ass.GetTypes();
+ Type ClassType = ass.GetType(menu.ClassName);
+ object Instance = Activator.CreateInstance(ClassType);
+ menu.Content = Instance;
+ if (Instance is IPlugin plugin)
+ {
+ }
+ else
+ {
+ plugin = new PluginPackage(Instance);
+ }
+ plugin.SetLogger(LogManager.GetLogger(menu.Code));
+ string configFolder = Path.Combine(PathHelper.ConfigFolder, menu.Code);
+ if (!Directory.Exists(configFolder))
+ {
+ Directory.CreateDirectory(configFolder);
+ }
+ plugin.SetBaseFolder(configFolder);
+ plugin.InitFinish();
+ pluginContainer.OnCloseEvent += plugin.OnClose;
+ pluginContainer.DoCloseEvent += plugin.DoClose;
+ Type SettingType = ass.GetType($"{menu.ClassName}_Setting");
+ if (SettingType != null)
+ {
+ object SettingInstance = Activator.CreateInstance(SettingType);
+ if (SettingInstance is IFolderOwner fo)
+ {
+ fo.SetBaseFolder(configFolder);
+ }
+ menu.SettingControl = SettingInstance;
+ }
+ }
+ LoadMessageAppend?.Invoke(this,
+ new LoadMsg
+ {
+ Type = LoadMsgType.LOADING,
+ Result = true,
+ Message = this.lang("App.MenuLoadSuccess", menu.Name),
+ });
+ }
+ catch (Exception ex)
+ {
+ log.Error($"Load Plugin [{menu.Name}] Failed.", ex);
+ LoadMessageAppend?.Invoke(this, new LoadMsg
+ {
+ Type = LoadMsgType.LOADING,
+ Result = false,
+ Message = this.lang("App.MenuLoadFailed", menu.Name, ex.Message, ex.StackTrace)
+ });
+ }
+ finally
+ {
+ c.Signal();
+ }
+ }));
+ c.Wait();
+ }
+ }
+
+ public List<PluginInfo> LoadTreeMenu(Dispatcher dispatcher)
+ {
+ //LoadMessageAppend?.Invoke(this, "寮�濮嬪姞杞借彍鍗�...");
+ List<PluginInfo> data = new List<PluginInfo>();
Dictionary<string, PluginInfo> cache = new Dictionary<string, PluginInfo>();
try
{
- LoadMessageAppend?.Invoke(this, "寮�濮嬪姞杞借彍鍗�...");
- DataTable dt = sqliteHelper.GetDataTable(@"SELECT * FROM menu ORDER BY level,id;");
+ UserInfo user = UserInfo.User;
+ string sql = $@"SELECT m.* FROM menu AS m INNER JOIN role_menu_rel AS rrel ON m.code = rrel.menuCode
+INNER JOIN role AS r ON rrel.roleCode = r.code
+INNER JOIN user_role_rel AS urel ON r.code = urel.roleCode
+AND urel.username = '{user.Username}' ORDER BY m.level,m.sort, m.id;";
+
+ DataTable dt = sqliteHelper.GetDataTable(sql);
if (dt != null)
{
- LoadMessageAppend?.Invoke(this, $"鍏辨湁{dt.Rows.Count}鏉¤彍鍗曞緟鍔犺浇...");
- foreach (DataRow dr in dt.Rows)
+ Application.Current.Dispatcher.Invoke(() =>
{
- string path = Path.Combine(PathHelper.IconFolder, dr["iconPath"].ToString());
- var menu = new PluginInfo
+ LoadMessageAppend?.Invoke(this,
+ new LoadMsg
+ {
+ Type = LoadMsgType.INIT,
+ NumData = dt.Rows.Count,
+ Message = this.lang("App.MenuCount", dt.Rows.Count),
+ });
+
+ foreach (DataRow dr in dt.Rows)
{
- Code = dr["code"].ToString(),
- Name = dr["name"].ToString(),
- ParentCode = dr["parentCode"].ToString(),
- DllPath = dr["dllPath"].ToString(),
- ClassName = dr["className"].ToString(),
- Icon = new BitmapImage(new Uri(Path.Combine(PathHelper.IconFolder, dr["iconPath"].ToString())))
- };
- cache[menu.Code] = menu;
- if (!cache.ContainsKey(menu.ParentCode))
- {
- data.Add(menu);
+ string path = Path.Combine(PathHelper.IconFolder, dr["iconPath"].ToString());
+ var menu = new PluginInfo
+ {
+ Code = dr["code"].ToString(),
+ Name = dr["name"].ToString(),
+ Type = dr["type"].ToString(),
+ ParentCode = dr["parentCode"].ToString(),
+ DllPath = dr["dllPath"].ToString(),
+ ClassName = dr["className"].ToString(),
+ IconFont = dr["icon"].ToString()
+ //Icon = new BitmapImage(new Uri(Path.Combine(PathHelper.IconFolder, dr["iconPath"].ToString())))
+ };
+ cache[menu.Code] = menu;
+ if (!cache.ContainsKey(menu.ParentCode))
+ {
+ data.Add(menu);
+ }
+ else
+ {
+ cache[menu.ParentCode].Children.Add(menu);
+ }
+ Plugins.Add(menu);
}
- else
- {
- cache[menu.ParentCode].Children.Add(menu);
- }
- Plugins.Add(menu);
- LoadChildrenMenu(menu);
- }
+ });
+ LoadChildrenMenu();
}
}
catch (Exception ex)
{
- LoadMessageAppend?.Invoke(this, ex.Message);
+ //LoadMessageAppend?.Invoke(this, ex.Message);
}
return data;
}
-
}
}
--
Gitblit v1.9.1