using Microsoft.Extensions.DependencyInjection;

using System.Threading.Tasks;

using System;

using Decor;

​

public class LoggingDecorator : IDecorator

{

    public async Task OnInvoke(Call call)

    {

        Console.WriteLine("Will do some work!");

        await call.Next();

        Console.WriteLine("Work is finished!");

    }

}

​

public class Worker

{

    [Decorate(typeof(LoggingDecorator))]

    public virtual void DoWork()

    {

        Console.WriteLine("Working...");

    }

}

​

public class Program

{

    static void Main()

    {

        var serviceProvider = new ServiceCollection()

            .AddTransient<LoggingDecorator>()

            .AddSingleton<Worker>()

            .Decorate<Worker>()

            .BuildServiceProvider();

        var worker = serviceProvider.GetRequiredService<Worker>();

        worker.DoWork();

    }

}