FastGithub/FastGithub.Core/PipelineBuilderExtensions.cs

using System;

namespace FastGithub
{
    /// <summary>
    /// 中间件创建者扩展
    /// </summary>
    public static class PipelineBuilderExtensions
    {
        /// <summary>
        /// 中断执行中间件
        /// </summary>
        /// <typeparam name="TContext"></typeparam>
        /// <param name="builder"></param>
        /// <param name="handler">处理者</param>
        /// <returns></returns>
        public static IPipelineBuilder<TContext> Run<TContext>(this IPipelineBuilder<TContext> builder, InvokeDelegate<TContext> handler)
        {
            return builder.Use(_ => handler);
        }

        /// <summary>
        /// 条件中间件
        /// </summary>
        /// <typeparam name="TContext"></typeparam>
        /// <param name="builder"></param>
        /// <param name="predicate"></param>
        /// <param name="handler"></param> 
        /// <returns></returns>
        public static IPipelineBuilder<TContext> When<TContext>(this IPipelineBuilder<TContext> builder, Func<TContext, bool> predicate, InvokeDelegate<TContext> handler)
        {
            return builder.Use(next => async context =>
            {
                if (predicate.Invoke(context) == true)
                {
                    await handler.Invoke(context);
                }
                else
                {
                    await next(context);
                }
            });
        }


        /// <summary>
        /// 条件中间件
        /// </summary>
        /// <typeparam name="TContext"></typeparam>
        /// <param name="builder"></param>
        /// <param name="predicate"></param>
        /// <param name="configureAction"></param>
        /// <returns></returns>
        public static IPipelineBuilder<TContext> When<TContext>(this IPipelineBuilder<TContext> builder, Func<TContext, bool> predicate, Action<IPipelineBuilder<TContext>> configureAction)
        {
            return builder.Use(next => async context =>
            {
                if (predicate.Invoke(context) == true)
                {
                    var branchBuilder = builder.New();
                    configureAction(branchBuilder);
                    await branchBuilder.Build().Invoke(context);
                }
                else
                {
                    await next(context);
                }
            });
        } 
    }
}