using System;

using System.Linq;

using System.Collections.Generic;

using Unity;

using Unity.Interception;

using Unity.Interception.ContainerIntegration;

using Unity.Interception.Interceptors.TypeInterceptors.VirtualMethodInterception;

using Unity.Interception.InterceptionBehaviors;

using Unity.Interception.PolicyInjection.Pipeline;

​

public class Program

{

    public static void Main()

    {

        var container = new UnityContainer();

        container.AddNewExtension<Interception>();

        container.RegisterType<IService,Service>(new Interceptor<VirtualMethodInterceptor>(),new InterceptionBehavior<ServiceInterceptor>());

        container.RegisterType<IService2,Service2>();

        Console.WriteLine("Resolve and call SayHello after interception registered");

        var s1 = container.Resolve<IService>();

        var s2 = container.Resolve<IService2>();

        //s1.SayHello();

        //s1.SayHello(5);

        s2.SayHello();

        Console.WriteLine();

    }

}

​

public interface IService 

{

    void SayHello();

    void SayHello(int prova);

}

public class Service: IService 

{

    public virtual void SayHello()

    {

        Console.WriteLine("Hello from Service");

    }

    public virtual void SayHello(int prova)

    {

        Console.WriteLine(String.Format("Hello from Service{0}",prova));

    }

}

public interface IService2 

{

    void SayHello();

}

public class Service2: IService2

{

    private readonly IService _service;

    public Service2(IService Service)

    {

        _service =Service;

    }

​

    public virtual void SayHello()

    {

        _service.SayHello();

    }

​

}

​

​

public class ServiceInterceptor : IInterceptionBehavior

{

     public IMethodReturn Invoke(IMethodInvocation input, GetNextInterceptionBehaviorDelegate getNext)

    {

        // Before invoking the method on the original target.

        Console.WriteLine(String.Format("Invoking method {0} at {1}",input.MethodBase, DateTime.Now.ToLongTimeString()));

        // Invoke the next behavior in the chain.

        var result = getNext()(input, getNext);

        return result;

    }

​

    public IEnumerable<Type> GetRequiredInterfaces()

    {

        return Type.EmptyTypes;

    }

​

    public bool WillExecute

    {

        get { return true; }

    }

}

​

public class ExeInterceptor : IInterceptionBehavior

{

    public IMethodReturn Invoke(IMethodInvocation input, GetNextInterceptionBehaviorDelegate getNext)

    {

        // Before invoking the method on the original target.

        Console.WriteLine(String.Format("Invoking method {0} at {1}",input.MethodBase, DateTime.Now.ToLongTimeString()));

        // Invoke the next behavior in the chain.

        var result = getNext()(input, getNext);

        return result;

    }

​

    public IEnumerable<Type> GetRequiredInterfaces()

    {

        return Type.EmptyTypes;

    }

​

    public bool WillExecute

    {

        get { return true; }

    }

}