serein-flow/NodeFlow/FlowStarter.cs

using Serein.Library.Api;
using Serein.Library.Core.NodeFlow;
using Serein.Library.Entity;
using Serein.Library.Enums;
using Serein.Library.Utils;
using Serein.Library.Web;
using Serein.NodeFlow.Base;
using Serein.NodeFlow.Model;
using System.ComponentModel.Design;
using System.Runtime.CompilerServices;
using static Serein.Library.Utils.ChannelFlowInterrupt;

namespace Serein.NodeFlow
{

    /// <summary>
    /// 流程启动器
    /// </summary>
    /// <param name="serviceContainer"></param>
    /// <param name="methodDetails"></param>
    public class FlowStarter
    {
        public FlowStarter()
        {
            SereinIOC = new SereinIOC();
        }

        /// <summary>
        /// 流程运行状态
        /// </summary>
        public enum RunState
        {
            /// <summary>
            /// 等待开始
            /// </summary>
            NoStart,
            /// <summary>
            /// 正在运行
            /// </summary>
            Running,
            /// <summary>
            /// 运行完成
            /// </summary>
            Completion,
        }
        /// <summary>
        /// 控制触发器
        /// </summary>
       
        public const string FlipFlopCtsName = "<>.FlowFlipFlopCts";

        public bool IsStopStart = false;
        /// <summary>
        /// 运行状态
        /// </summary>
        public RunState FlowState { get; private set; } = RunState.NoStart;
        public RunState FlipFlopState { get; private set; } = RunState.NoStart;

        /// <summary>
        /// 运行时的IOC容器
        /// </summary>
        private ISereinIOC SereinIOC { get; } = null; 

        /// <summary>
        /// 结束运行时需要执行的方法
        /// </summary>
        private Action ExitAction { get; set; }  = null; 

        /// <summary>
        /// 运行的上下文
        /// </summary>
        private IDynamicContext Context { get; set; }  = null; 

        private void CheckStartState()
        {
            if (IsStopStart)
            {
                throw new Exception("停止启动");

            }
        }


        /// <summary>
        /// 开始运行
        /// </summary>
        /// <param name="startNode">起始节点</param>
        /// <param name="env">运行环境</param>
        /// <param name="runNodeMd">环境中已加载的所有节点方法</param>
        /// <param name="flipflopNodes">触发器节点</param>
        /// <returns></returns>
        public async Task RunAsync(NodeModelBase startNode,
                                   IFlowEnvironment env,
                                   List<MethodDetails> runNodeMd,
                                   List<MethodDetails> initMethods,
                                   List<MethodDetails> loadingMethods,
                                   List<MethodDetails> exitMethods,
                                   List<SingleFlipflopNode> flipflopNodes)
        {
            
            FlowState = RunState.Running; // 开始运行

            if (startNode == null) {
                FlowState = RunState.Completion; // 不存在起点，退出流程
                return; 
            }

            #region 选择运行环境的上下文

            // 判断使用哪一种流程上下文
            var isNetFramework = false;
            if (isNetFramework)
            {
                Context = new Serein.Library.Framework.NodeFlow.DynamicContext(SereinIOC, env);
            }
            else
            {
                Context = new Serein.Library.Core.NodeFlow.DynamicContext(SereinIOC, env); // 从起始节点启动流程时创建上下文
            }
            #endregion

            #region 初始化运行环境的Ioc容器
            // 清除节点使用的对象，筛选出需要初始化的方法描述
            var thisRuningMds = new List<MethodDetails>();
            thisRuningMds.AddRange(runNodeMd.Where(md => md is not null));
            thisRuningMds.AddRange(initMethods.Where(md => md is not null));
            thisRuningMds.AddRange(loadingMethods.Where(md => md is not null));
            thisRuningMds.AddRange(exitMethods.Where(md => md is not null));

            // .AddRange(initMethods).AddRange(loadingMethods).a
            foreach (var nodeMd in thisRuningMds)
            {
                nodeMd.ActingInstance = null;
            }
            SereinIOC.Reset(); // 开始运行时清空ioc中注册的实例
            // 初始化ioc容器中的类型对象
            foreach (var md in thisRuningMds)
            {
                if (md.ActingInstanceType != null)
                {
                    SereinIOC.Register(md.ActingInstanceType);
                }
                else
                {
                    await Console.Out.WriteLineAsync($"{md.MethodName} - 没有类型声明");
                    IsStopStart = true;
                }
            }
            CheckStartState();


            SereinIOC.Build(); // 流程启动前的初始化

            foreach (var md in thisRuningMds)
            {
                md.ActingInstance = SereinIOC.GetOrRegisterInstantiate(md.ActingInstanceType);
                if(md.ActingInstance is null)
                {
                    await Console.Out.WriteLineAsync($"{md.MethodName} - 无法获取类型[{md.ActingInstanceType}]的实例");
                    IsStopStart = true;
                }
            }

            CheckStartState();

            //foreach (var md in flipflopNodes.Select(it => it.MethodDetails).ToArray())
            //{
            //    md.ActingInstance = SereinIoc.GetOrCreateServiceInstance(md.ActingInstanceType);
            //}
            #endregion

            #region 检查并修正初始化、加载时、退出时方法作用的对象，保证后续不会报错（已注释）
            //foreach (var md in initMethods) // 初始化
            //{
            //    md.ActingInstance ??= Context.SereinIoc.GetOrRegisterInstantiate(md.ActingInstanceType);
            //}
            //foreach (var md in loadingMethods) // 加载
            //{
            //    md.ActingInstance ??= Context.SereinIoc.GetOrRegisterInstantiate(md.ActingInstanceType);
            //}
            //foreach (var md in exitMethods) // 初始化
            //{
            //    md.ActingInstance ??= Context.SereinIoc.GetOrRegisterInstantiate(md.ActingInstanceType);
            //}
            #endregion

            #region 执行初始化，绑定IOC容器，再执行加载时

            //object?[]? args = [Context];
            foreach (var md in initMethods) // 初始化
            {
                ((Action<object, object?[]?>)md.MethodDelegate).Invoke(md.ActingInstance, [Context]);
            }
            Context.SereinIoc.Build(); // 绑定初始化时注册的类型
            foreach (var md in loadingMethods) // 加载
            {
                //object?[]? data = [md.ActingInstance, args];
                //md.MethodDelegate.DynamicInvoke(data);
                ((Action<object, object?[]?>)md.MethodDelegate).Invoke(md.ActingInstance, [Context]);
            }
            Context.SereinIoc.Build(); // 预防有人在加载时才注册类型，再绑定一次
            #endregion

            #region 设置流程退出时的回调函数
            ExitAction = () =>
            {
                
                SereinIOC.Run<WebServer>(web => {
                    web?.Stop();
                });

                foreach (MethodDetails? md in exitMethods)
                {
                    ((Action<object, object?[]?>)md.MethodDelegate).Invoke(md.ActingInstance, [Context]);
                }

                //if (Context != null && Context.NodeRunCts != null && !Context.NodeRunCts.IsCancellationRequested)
                //{
                //    Context.NodeRunCts.Cancel();
                //}

                //if (FlipFlopCts != null && !FlipFlopCts.IsCancellationRequested)
                //{
                //    FlipFlopCts?.Cancel();
                //    FlipFlopCts?.Dispose();
                //}
                FlowState = RunState.Completion;
                FlipFlopState = RunState.Completion;
            };
            #endregion

            #region 开始启动流程
            CancellationTokenSource FlipFlopCts = null;
            try
            {
                
                if (flipflopNodes.Count > 0)
                {
                    FlipFlopState = RunState.Running;
                    // 如果存在需要启动的触发器，则开始启动
                    FlipFlopCts = new CancellationTokenSource();
                    SereinIOC.CustomRegisterInstance(FlipFlopCtsName, FlipFlopCts,false);

                    // 使用 TaskCompletionSource 创建未启动的触发器任务
                    var tasks = flipflopNodes.Select(async node =>
                    {
                        await FlipflopExecute(env,node);
                    }).ToArray();
                    _ = Task.WhenAll(tasks);
                }
                await startNode.StartExecute(Context); // 开始运行时从起始节点开始运行
                // 等待结束
                if(FlipFlopState == RunState.Running && FlipFlopCts is not null)
                {
                    while (!FlipFlopCts.IsCancellationRequested)
                    {
                        await Task.Delay(100);
                    }
                }

                //FlipFlopCts?.Dispose();
            }
            catch (Exception ex)
            {
                await Console.Out.WriteLineAsync(ex.ToString());
            }
            finally
            {
                FlipFlopCts?.Dispose();
                FlowState = RunState.Completion;
            }
            #endregion
        }

        public void AddFlipflopInRuning(SingleFlipflopNode singleFlipFlopNode, IFlowEnvironment flowEnvironment)
        {
            _ = Task.Run(async () =>
            {
                // 设置对象
                singleFlipFlopNode.MethodDetails.ActingInstance = SereinIOC.GetOrRegisterInstantiate(singleFlipFlopNode.MethodDetails.ActingInstanceType);
                await FlipflopExecute(flowEnvironment,singleFlipFlopNode); // 启动触发器
            });
        }

        /// <summary>
        /// 启动全局触发器
        /// </summary>
        /// <param name="flowEnvironment">流程运行全局环境</param>
        /// <param name="singleFlipFlopNode">需要全局监听信号的触发器</param>
        /// <returns></returns>
        private async Task FlipflopExecute(IFlowEnvironment flowEnvironment,SingleFlipflopNode singleFlipFlopNode)
        {
            CancellationTokenSource cts = null;

            var context = new DynamicContext(SereinIOC, flowEnvironment); // 启动全局触发器时新建上下文

            MethodDetails md = singleFlipFlopNode.MethodDetails;
            var del = md.MethodDelegate;
            object?[]? parameters = singleFlipFlopNode.GetParameters(context, singleFlipFlopNode.MethodDetails); // 启动全局触发器时获取入参参数
            // 设置委托对象
            var func = md.ExplicitDatas.Length == 0 ?
                (Func<object, object, Task<IFlipflopContext>>)del :
                (Func<object, object[], Task<IFlipflopContext>>)del;

            bool t = md.ExplicitDatas.Length == 0;
            try
            {


                cts = Context.SereinIoc.Get<CancellationTokenSource>(FlipFlopCtsName);

                while (!cts.IsCancellationRequested)
                {

                    singleFlipFlopNode.FlowData = await singleFlipFlopNode.ExecutingAsync(context);
                    if (singleFlipFlopNode.NextOrientation != ConnectionType.None)
                    {
                        var nextNodes = singleFlipFlopNode.SuccessorNodes[singleFlipFlopNode.NextOrientation];
                        for (int i = nextNodes.Count - 1; i >= 0; i--)
                        {
                            if (nextNodes[i].DebugSetting.IsEnable) // 排除未启用的后继节点
                            {
                                nextNodes[i].PreviousNode = singleFlipFlopNode;
                                await nextNodes[i].StartExecute(context); // 执行流程节点的后继分支
                            }
                        }
                    }
                    //if(t)
                    //{
                    //    IFlipflopContext flipflopContext = await ((Func<object, Task<IFlipflopContext>>)del.Clone()).Invoke(md.ActingInstance);// 开始等待全局触发器的触发
                    //    var connectionType = flipflopContext.State.ToContentType();
                    //    if (connectionType != ConnectionType.None)
                    //    {
                    //        await GlobalFlipflopExecute(context, singleFlipFlopNode, connectionType, cts);
                    //    }
                    //}
                    //else
                    //{
                    //    IFlipflopContext flipflopContext = await ((Func<object, object[], Task<IFlipflopContext>>)del.Clone()).Invoke(md.ActingInstance, parameters);// 开始等待全局触发器的触发
                    //    var connectionType = flipflopContext.State.ToContentType();
                    //    if (connectionType != ConnectionType.None)
                    //    {
                    //        await GlobalFlipflopExecute(context, singleFlipFlopNode, connectionType, cts);
                    //    }
                    //}

                }
            }
            catch (Exception ex)
            {
                await Console.Out.WriteLineAsync(ex.ToString());
            }
            finally
            {
                cts?.Cancel();
            }
        }

        /// <summary>
        /// 全局触发器开始执行相关分支
        /// </summary>
        /// <param name="context">上下文</param>
        /// <param name="singleFlipFlopNode">被触发的全局触发器</param>
        /// <param name="connectionType">分支类型</param>
        /// <returns></returns>
        public async Task GlobalFlipflopExecute(IDynamicContext context, SingleFlipflopNode singleFlipFlopNode,
                                                ConnectionType connectionType, CancellationTokenSource cts)
        {
            

            bool skip = true;
            Stack<NodeModelBase> stack = new Stack<NodeModelBase>();
            stack.Push(singleFlipFlopNode);


            while (stack.Count > 0 && !cts.IsCancellationRequested) // 循环中直到栈为空才会退出循环
            {
                // 从栈中弹出一个节点作为当前节点进行处理
                var currentNode = stack.Pop();

                // 设置方法执行的对象
                //if (currentNode.MethodDetails?.ActingInstance == null && currentNode.MethodDetails?.ActingInstanceType is not null)
                //{
                //    currentNode.MethodDetails.ActingInstance ??= context.SereinIoc.GetOrRegisterInstantiate(currentNode.MethodDetails.ActingInstanceType);
                //}

                // 首先执行上游分支
                var upstreamNodes = currentNode.SuccessorNodes[ConnectionType.Upstream];
                for (int i = upstreamNodes.Count - 1; i >= 0; i--)
                {
                    upstreamNodes[i].PreviousNode = currentNode;
                    await upstreamNodes[i].StartExecute(context); // 执行全局触发器的上游分支
                }

                // 当前节点是已经触发了的全局触发器，所以跳过，难道每次都要判断一次？
                if (skip)
                {
                    skip = false;
                }
                else
                {
                    currentNode.FlowData = await currentNode.ExecutingAsync(context);
                    

                    if (currentNode.NextOrientation == ConnectionType.None) 
                    {
                        break;  // 不再执行
                    }
                    connectionType = currentNode.NextOrientation;
                }

                // 获取下一分支
                var nextNodes = currentNode.SuccessorNodes[connectionType];

                // 将下一个节点集合中的所有节点逆序推入栈中
                for (int i = nextNodes.Count - 1; i >= 0; i--)
                {
                    nextNodes[i].PreviousNode = currentNode;
                    stack.Push(nextNodes[i]);
                }
            }
        }


        public void Exit()
        {
            ExitAction?.Invoke();
        }
    }
}