serein-flow/NodeFlow/Tool/FlowLibraryManagement.cs

using Serein.Library;
using Serein.Library.Api;
using Serein.Library.FlowNode;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using System.Reflection;
using System.Runtime.Loader;
using System.Text;
using System.Threading.Tasks;
using System.Xml.Linq;

namespace Serein.NodeFlow.Tool
{
    /// <summary>
    /// 管理加载在运行环境中的外部程序集
    /// </summary>
    public class FlowLibraryManagement
    {
        public FlowLibraryManagement(IFlowEnvironment flowEnvironment)
        {
            this.flowEnvironment = flowEnvironment;
        }

        private readonly IFlowEnvironment flowEnvironment;

        /// <summary>
        /// 缓存所有加载了的程序集
        /// </summary>
        private readonly ConcurrentDictionary<string, FlowLibrary> _myFlowLibrarys = new ConcurrentDictionary<string, FlowLibrary>();

        public (NodeLibraryInfo, List<MethodDetailsInfo>) LoadLibrary(string libraryfilePath)
        {

            return LoadDllNodeInfo(libraryfilePath);
        }

        public bool UnloadLibrary(string libraryName)
        {
            libraryName = libraryName.Split(',')[0];
            if (_myFlowLibrarys.Remove(libraryName, out var flowLibrary))
            {
                try
                {
                    flowLibrary.Upload();
                    return true;
                }
                catch (Exception ex)
                {
                    Console.WriteLine($"尝试卸载程序集[{libraryName}]发生错误：{ex}");
                    return false;
                }

            }
            else
            {
                return false;
            }
        }

        /// <summary>
        /// 获取方法描述
        /// </summary>
        /// <param name="assemblyName">程序集名称</param>
        /// <param name="methodName">方法名称</param>
        /// <param name="md">返回的方法描述</param>
        /// <returns>是否获取成功</returns>
        public bool TryGetMethodDetails(string assemblyName, string methodName, [MaybeNullWhen(false)] out MethodDetails md)
        {
            if (_myFlowLibrarys.TryGetValue(assemblyName, out var flowLibrary)
                && flowLibrary.MethodDetailss.TryGetValue(methodName, out md))
            {
                return true;
            }
            else
            {
                md = null;
                return false;
            }
        }

        /// <summary>
        /// 获取方法调用的委托
        /// </summary>
        /// <param name="assemblyName">程序集名称</param>
        /// <param name="methodName">方法名称</param>
        /// <param name="dd">返回的委托调用封装类</param>
        /// <returns>是否获取成功</returns>
        public bool TryGetDelegateDetails(string assemblyName, string methodName, [MaybeNullWhen(false)] out DelegateDetails dd)
        {
            if (_myFlowLibrarys.TryGetValue(assemblyName, out var flowLibrary)
               && flowLibrary.DelegateDetailss.TryGetValue(methodName, out dd))
            {
                return true;
            }
            else
            {
                dd = null;
                return false;
            }
        }


        /// <summary>
        /// 获取(初始化/加载时/退出后)相应的节点方法
        /// </summary>
        /// <param name="nodeType"></param>
        /// <returns></returns>
        public List<MethodDetails> GetMdsOnFlowStart(NodeType nodeType)
        {
            List<MethodDetails> mds = [];

            foreach (var library in _myFlowLibrarys.Values)
            {
                var t_mds = library.MethodDetailss.Values.Where(it => it.MethodDynamicType == nodeType).ToList();
                mds.AddRange(t_mds);
            }
            return mds;
        }

        /// <summary>
        /// 获取流程启动时在不同时间点需要自动实例化的类型
        /// </summary>
        /// <returns></returns>
        public Dictionary<RegisterSequence, List<Type>> GetaAutoRegisterType()
        {
            Dictionary<RegisterSequence, List<Type>> rsTypes = new Dictionary<RegisterSequence, List<Type>>();
            foreach (var library in _myFlowLibrarys.Values)
            {
                foreach (var kv in library.RegisterTypes)
                {
                    var @class = kv.Key;
                    var type = kv.Value;
                    if (!rsTypes.TryGetValue(@class, out var tmpTypes))
                    {
                        tmpTypes = new List<Type>();
                        rsTypes.Add(@class, tmpTypes);
                    }
                    tmpTypes.AddRange(type);
                }
            }
            return rsTypes;
        }


        /// <summary>
        /// 获取所有方法信息，用于保存项目时调用
        /// </summary>
        /// <returns></returns>
        public List<LibraryMds> GetAllLibraryMds()
        {
            List<LibraryMds> mds = new List<LibraryMds>();
            foreach (FlowLibrary library in _myFlowLibrarys.Values)
            {
                var tmp = new LibraryMds
                {
                    AssemblyName = library.FullName,
                    Mds = library.MethodDetailss.Values.Select(md => md.ToInfo()).ToArray()
                };
                mds.Add(tmp);
            }
            return mds;
        }





        /// <summary>
        /// 序列化当前项目的依赖信息、节点信息，用于远程登录的场景，需要将依赖信息从本地（受控端）发送到远程（主控端）
        /// </summary>
        /// <returns></returns>
        public List<NodeLibraryInfo> GetAllLibraryInfo()
        {
            return _myFlowLibrarys.Values.Select(library => library.ToInfo()).ToList();
        }


        #region 功能性方法

        /// <summary>
        /// 从文件路径中加载程序集，返回相应的信息
        /// </summary>
        /// <param name="dllFilePath"></param>
        /// <returns></returns>
        private (NodeLibraryInfo, List<MethodDetailsInfo>) LoadDllNodeInfo(string dllFilePath)
        {

#if true
            var fileName = Path.GetFileName(dllFilePath); // 获取文件名
            var dir = Path.GetDirectoryName(dllFilePath); // 获取目录路径
            var sereinFlowLibraryPath = Path.Combine(dir, $"{nameof(Serein)}.{nameof(Serein.Library)}.dll"); 
            // 每个类库下面至少需要有“Serein.Library.dll”类库依赖
            var flowAlc = new FlowLibraryAssemblyContext(sereinFlowLibraryPath, fileName);
            Action actionUnload = () =>
            {
                flowAlc?.Unload(); // 卸载程序集
                flowAlc = null;
                GC.Collect(); // 强制触发GC确保卸载成功
                GC.WaitForPendingFinalizers();
            };
            var assembly = flowAlc.LoadFromAssemblyPath(dllFilePath); // 加载指定路径的程序集
            if (_myFlowLibrarys.ContainsKey(assembly.GetName().Name))
            {
                actionUnload.Invoke();
                throw new Exception($"程序集[{assembly.GetName().FullName}]已经加载过!");
            }

            FlowLibrary flowLibrary = new FlowLibrary(dllFilePath, assembly, actionUnload);
            if (flowLibrary.LoadAssembly(assembly))
            {
                _myFlowLibrarys.TryAdd(assembly.GetName().Name, flowLibrary);
                (NodeLibraryInfo, List<MethodDetailsInfo>) result = (flowLibrary.ToInfo(),
                                                                flowLibrary.MethodDetailss.Values.Select(md => md.ToInfo()).ToList());
                return result;
            }
            else
            {
                throw new Exception($"程序集[{assembly.GetName().FullName}]加载失败");
            }

#else
  var fileName = Path.GetFileName(dllFilePath); // 获取文件名
            Assembly assembly = Assembly.LoadFrom(dllFilePath); // 加载程序集
            FlowLibrary flowLibrary = new FlowLibrary(dllFilePath, assembly, () =>
            {
                Console.WriteLine("暂未实现卸载程序集");
                //flowAlc.Unload(); // 卸载程序集
                //flowAlc = null;
                //GC.Collect(); // 强制触发GC确保卸载成功
                //GC.WaitForPendingFinalizers();
            });
            
            _myFlowLibrarys.TryAdd(assembly.GetName().Name, flowLibrary);

            (NodeLibraryInfo, List<MethodDetailsInfo>) result = (flowLibrary.ToInfo(),
                                                            flowLibrary.MethodDetailss.Values.Select(md => md.ToInfo()).ToList());
            return result;

            
           
#endif

        }



        #endregion
    }

    /// <summary>
    /// 流程依赖加载
    /// </summary>
    public class FlowLibraryAssemblyContext : AssemblyLoadContext
    {
        private readonly AssemblyDependencyResolver _resolver;

        /// <summary>
        ///  创建新的加载上下文
        /// </summary>
        /// <param name="sereinFlowLibraryPath">类库主</param>
        /// <param name="name"></param>
        public FlowLibraryAssemblyContext(string sereinFlowLibraryPath, string name) : base(name, isCollectible: true)
        {
            _resolver = new AssemblyDependencyResolver(sereinFlowLibraryPath);
        }

        protected override Assembly? Load(AssemblyName assemblyName)
        {
            string? assemblyPath = _resolver.ResolveAssemblyToPath(assemblyName);
            if (!string.IsNullOrEmpty(assemblyPath))
            {
                var assembly = AssemblyLoadContext.Default.LoadFromAssemblyPath(assemblyPath);
                //var assembly = LoadFromAssemblyPath(assemblyPath);
                return assembly;
            }
            else
            {
                return Default.Assemblies.FirstOrDefault(x => x.FullName == assemblyName.FullName);
            }


            // 构建依赖项的路径
            //string assemblyPath = Path.Combine(AppContext.BaseDirectory, assemblyName.Name + ".dll");
            //if (File.Exists(assemblyPath))
            //{
            //    return LoadFromAssemblyPath(assemblyPath);
            //}
            //assemblyPath = Path.Combine(filePath, assemblyName.Name + ".dll");
            //if (File.Exists(assemblyPath))
            //{
            //    return LoadFromAssemblyPath(assemblyPath);
            //}

            //return null; // 如果没有找到，返回 null
        }
    }
    public static class PluginAssemblyContextExtensions
    {

        public static Assembly FromAssemblyPath(this AssemblyLoadContext context, string path)
        {

            return context.LoadFromAssemblyPath(path);

        }

    }
}