使用Unity拦截应用程序中的所有类

Tsahi 发表于 Dev

沙希

我想在我的应用程序中使用拦截器来进行日志记录。当然，我可以向拦截器注册每种类型，如下所示：

container.AddNewExtension<Interception>();
container.RegisterType<ITenantStore, TenantStore>(
new Interceptor<InterfaceInterceptor>(),
new InterceptionBehavior<LoggingInterceptionBehavior>());

但是由于我有很多类型，所以我想知道是否有一种更干净的方法可以对所有班级做到这一点？

泰勒·奥尔森

这是一种可以对所有已注册类型进行拦截的方法。因此，请在注册完成后致电此网站。然后，您可以使用策略注入来连接要使用的拦截器。

public static class UnityContainerInterceptionExtensions
{
    public static IUnityContainer EnableInterceptionForAllRegisteredTypes
        (this IUnityContainer unityContainer)
    {
        unityContainer
            .Registrations
            .ForEach(r => unityContainer.EnableInterception(r.RegisteredType, r.Name));

        return unityContainer;
    }

    public static IUnityContainer EnableInterception<T>
        (this IUnityContainer unityContainer, string name = null)
    {
        return unityContainer.EnableInterception(typeof (T), name);
    }

    public static IUnityContainer EnableInterception
        (this IUnityContainer unityContainer, Type registrationType, string name = null)
    {
        // Don't allow interception of unity types
        if (registrationType.Namespace.StartsWith(typeof(IUnityContainer).Namespace))
            return unityContainer;

        // Don't allow interception on the intercepting call handlers
        if (typeof (ICallHandler).IsAssignableFrom(registrationType))
            return unityContainer;

        var interception = unityContainer.Configure<Interception>();

        var interfaceInterceptor = new InterfaceInterceptor();
        if (interfaceInterceptor.CanIntercept(registrationType))
        {
            interception
                .SetInterceptorFor(registrationType, name, interfaceInterceptor);
            return unityContainer;
        }

        var virtualMethodInterceptor = new VirtualMethodInterceptor();
        if (virtualMethodInterceptor.CanIntercept(registrationType))
        {
            interception
                .SetInterceptorFor(registrationType, name, virtualMethodInterceptor);
            return unityContainer;
        }

        var transparentProxyInterceptor = new TransparentProxyInterceptor();
        if (transparentProxyInterceptor.CanIntercept(registrationType))
        {
            interception
                .SetInterceptorFor(registrationType, name, transparentProxyInterceptor);
            return unityContainer;
        }

        return unityContainer;
    }
}