samples/DotNetCoreSample/PipelineTest.cs

﻿using WeihanLi.Common.Helpers;

// ReSharper disable LocalizableElement

namespace DotNetCoreSample;

public class PipelineTest
{
    public static void Test()
    {
        var requestContext = new RequestContext()
        {
            RequesterName = "Kangkang",
            Hour = 12,
        };

        var builder = PipelineBuilder.Create<RequestContext>(context =>
                {
                    Console.WriteLine($"{context.RequesterName} {context.Hour}h apply failed");
                })
                .Use((context, next) =>
                {
                    if (context.Hour <= 2)
                    {
                        Console.WriteLine("pass 1");
                    }
                    else
                    {
                        next();
                    }
                })
                .Use((context, next) =>
                {
                    if (context.Hour <= 4)
                    {
                        Console.WriteLine("pass 2");
                    }
                    else
                    {
                        next();
                    }
                })
                .Use((context, next) =>
                {
                    if (context.Hour <= 6)
                    {
                        Console.WriteLine("pass 3");
                    }
                    else
                    {
                        next();
                    }
                })
            ;
        var requestPipeline = builder.Build();
        Console.WriteLine();
        foreach (var i in Enumerable.Range(1, 8))
        {
            Console.WriteLine($"--------- h:{i} apply Pipeline------------------");
            requestContext.Hour = i;
            requestPipeline.Invoke(requestContext);
            Console.WriteLine("----------------------------");
        }
    }

    public static void MiddlewareTest()
    {
        var context = new RequestContext();
        var pipeline = PipelineBuilder.Create<RequestContext>()
            .UseMiddleware<RequestContext, FooMiddleware>()
            .UseMiddleware<RequestContext, BarMiddleware>()
            .Build();
        pipeline(context);
    }

    public static void TestV2()
    {
        var requestContext = new RequestContext()
        {
            RequesterName = "Kangkang",
            Hour = 12,
        };

        var builder = PipelineBuilder.Create<RequestContext>(context =>
                {
                    Console.WriteLine($"{context.RequesterName} {context.Hour}h apply failed");
                })
                .Use((context, next) =>
                {
                    Console.WriteLine("Initialize");
                    next(context);
                })
                .When(context => context.Hour <= 2, pipeline =>
                       {
                           pipeline.Use((_, next) =>
                           {
                               Console.WriteLine("This should be invoked");
                               next();
                           });
                           pipeline.Run(_ => Console.WriteLine("pass 1"));
                           pipeline.Use((_, next) =>
                           {
                               Console.WriteLine("This should not be invoked");
                               next();
                               Console.WriteLine("will this invoke?");
                           });
                       })
                .UseWhen(context => context.Hour > 2, pipeline =>
                  {
                      pipeline.Use((c, next) =>
                      {
                          Console.WriteLine("Use when middleware before");
                          next();
                          Console.WriteLine("Use when middleware after");
                      });
                  })
                .When(context => context.Hour <= 4, pipeline =>
                   {
                       pipeline.Run(_ => Console.WriteLine("pass 2"));
                   })
                .When(context => context.Hour <= 6, pipeline =>
                   {
                       pipeline.Run(_ => Console.WriteLine("pass 3"));
                   })

            ;
        var requestPipeline = builder.Build();
        Console.WriteLine();
        foreach (var i in Enumerable.Range(1, 8))
        {
            Console.WriteLine($"--------- h:{i} apply Pipeline------------------");
            requestContext.Hour = i;
            requestPipeline.Invoke(requestContext);
            Console.WriteLine("----------------------------");
        }
    }

    public static async Task AsyncPipelineBuilderTest()
    {
        var requestContext = new RequestContext()
        {
            RequesterName = "Michael",
            Hour = 12,
        };

        var builder = PipelineBuilder.CreateAsync<RequestContext>(context =>
                {
                    Console.WriteLine($"{context.RequesterName} {context.Hour}h apply failed");
                    return Task.CompletedTask;
                })
                .Use(async (context, next) =>
                {
                    if (context.Hour <= 2)
                    {
                        Console.WriteLine("pass 1");
                    }
                    else
                    {
                        await next();
                    }
                })
                .Use(async (context, next) =>
                {
                    if (context.Hour <= 4)
                    {
                        Console.WriteLine("pass 2");
                    }
                    else
                    {
                        await next();
                    }
                })
                .Use(async (context, next) =>
                {
                    if (context.Hour <= 6)
                    {
                        Console.WriteLine("pass 3");
                    }
                    else
                    {
                        await next();
                    }
                })
            ;
        var requestPipeline = builder.Build();
        Console.WriteLine();
        foreach (var i in Enumerable.Range(1, 8))
        {
            Console.WriteLine($"--------- h:{i} apply AsyncPipeline------------------");
            requestContext.Hour = i;
            await requestPipeline.Invoke(requestContext);
            Console.WriteLine("----------------------------");
        }
    }

    public static async Task AsyncPipelineBuilderTestV2()
    {
        var requestContext = new RequestContext()
        {
            RequesterName = "Michael",
            Hour = 12,
        };

        var builder = PipelineBuilder.CreateAsync<RequestContext>(context =>
                {
                    Console.WriteLine($"{context.RequesterName} {context.Hour}h apply failed");
                    return Task.CompletedTask;
                })
                .When(context => context.Hour <= 2, pipeline =>
                {
                    pipeline.Run(_ =>
                    {
                        Console.WriteLine("pass 1");
                        return Task.CompletedTask;
                    });
                })
                .When(context => context.Hour <= 4, pipeline =>
                {
                    pipeline.Run(_ =>
                    {
                        Console.WriteLine("pass 2");
                        return Task.CompletedTask;
                    });
                })
                .When(context => context.Hour <= 6, pipeline =>
                {
                    pipeline.Run(_ =>
                    {
                        Console.WriteLine("pass 3");
                        return Task.CompletedTask;
                    });
                })
            ;
        var requestPipeline = builder.Build();
        Console.WriteLine();
        foreach (var i in Enumerable.Range(1, 8))
        {
            Console.WriteLine($"--------- h:{i} apply AsyncPipeline------------------");
            requestContext.Hour = i;
            await requestPipeline.Invoke(requestContext);
            Console.WriteLine("----------------------------");
        }
    }
}
file class RequestContext
{
    public string? RequesterName { get; set; }

    public int Hour { get; set; }
}

file class FooMiddleware : IPipelineMiddleware<RequestContext>, IAsyncPipelineMiddleware<RequestContext>
{
    public void Invoke(RequestContext context, Action<RequestContext> next)
    {
        Console.WriteLine("I'm foo");
        next(context);
    }

    public Task InvokeAsync(RequestContext context, Func<RequestContext, Task> next)
    {
        Console.WriteLine("I'm foo");
        return next(context); ;
    }
}

file class BarMiddleware : IPipelineMiddleware<RequestContext>, IAsyncPipelineMiddleware<RequestContext>
{
    public void Invoke(RequestContext context, Action<RequestContext> next)
    {
        Console.WriteLine("I'm bar");
        next(context);
    }

    public Task InvokeAsync(RequestContext context, Func<RequestContext, Task> next)
    {
        Console.WriteLine("I'm bar");
        return next(context); ;
    }
}