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 interface IWorker

{

    void DoWork();

}

​

public class Worker : IWorker

{

    [Decorate(typeof(LoggingDecorator))]

    public void DoWork()

    {

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

    }

}

​

public class Program

{

    static void Main()

    {

        var serviceProvider = new ServiceCollection()

            .AddSingleton<IUnityContainer,UnityContainer>()

            .AddTransient<LoggingDecorator>()

            .AddSingleton<IWorker, Worker>()

            .Decorate<IWorker>()

            .BuildServiceProvider();

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

        worker.DoWork();

    }

}