在此期间,我找到了解决方案。这是一个简单的修复,但它要求您覆盖 Interception 和 TypeInterceptionStrategy。修复是一个单行程序,仅涉及检查策略列表中出现的类型。
这是来自 TypeInterceptionStrategy 的代码:
if (originalConstructorSelectorPolicy is DefaultUnityConstructorSelectorPolicy)
{
containingPolicyList.Set<IConstructorSelectorPolicy>(new CustomDerivedTypeConstructorSelectorPolicy(proxyType, originalConstructorSelectorPolicy), context.BuildKey);
}
当然,为了改变它,你必须复制 TypeInterceptionStrategy 并做与修复相同的事情,它不能简单地通过覆盖 PreBuildUp 方法来修复。
我将整个修复粘贴在这里,以防其他人遇到问题。
public class CustomTypeInterceptionStrategy : BuilderStrategy
{
public override void PreBuildUp(IBuilderContext context)
{
Guard.ArgumentNotNull(context, "context");
if (context.Existing != null)
{
return;
}
Type type = context.BuildKey.Type;
ITypeInterceptionPolicy typePolicy = FindInterceptionPolicy<ITypeInterceptionPolicy>(context);
if (typePolicy == null)
{
return;
}
ITypeInterceptor interceptor = typePolicy.GetInterceptor(context);
if (!interceptor.CanIntercept(type))
{
return;
}
IInterceptionBehaviorsPolicy behaviorPolicy = FindInterceptionPolicy<IInterceptionBehaviorsPolicy>(context);
IEnumerable<IInterceptionBehavior> interceptionBehaviors = behaviorPolicy == null
? Enumerable.Empty<IInterceptionBehavior>()
: behaviorPolicy.GetEffectiveBehaviors(context, interceptor, type, type).Where(ib => ib.WillExecute);
IAdditionalInterfacesPolicy interceptionPolicy3 = FindInterceptionPolicy<IAdditionalInterfacesPolicy>(context);
IEnumerable<Type> additionalInterfaces1 = interceptionPolicy3 != null ? interceptionPolicy3.AdditionalInterfaces : Type.EmptyTypes;
context.Policies.Set(new CustomEffectiveInterceptionBehaviorsPolicy() { Behaviors = interceptionBehaviors }, context.BuildKey);
Type[] additionalInterfaces2 = Intercept.GetAllAdditionalInterfaces(interceptionBehaviors, additionalInterfaces1);
Type proxyType = interceptor.CreateProxyType(type, additionalInterfaces2);
IPolicyList containingPolicyList;
IConstructorSelectorPolicy originalConstructorSelectorPolicy = context.Policies.Get<IConstructorSelectorPolicy>(context.BuildKey, out containingPolicyList);
if (originalConstructorSelectorPolicy is DefaultUnityConstructorSelectorPolicy)
{
containingPolicyList.Set<IConstructorSelectorPolicy>(new CustomDerivedTypeConstructorSelectorPolicy(proxyType, originalConstructorSelectorPolicy), context.BuildKey);
}
}
public override void PostBuildUp(IBuilderContext context)
{
Guard.ArgumentNotNull(context, "context");
IInterceptingProxy interceptingProxy = context.Existing as IInterceptingProxy;
if (interceptingProxy == null)
{
return;
}
CustomEffectiveInterceptionBehaviorsPolicy interceptionBehaviorsPolicy = context.Policies.Get<CustomEffectiveInterceptionBehaviorsPolicy>(context.BuildKey, true);
if (interceptionBehaviorsPolicy == null)
{
return;
}
foreach (IInterceptionBehavior interceptor in interceptionBehaviorsPolicy.Behaviors)
{
interceptingProxy.AddInterceptionBehavior(interceptor);
}
}
private static TPolicy FindInterceptionPolicy<TPolicy>(IBuilderContext context) where TPolicy : class, IBuilderPolicy
{
TPolicy policy = context.Policies.Get<TPolicy>(context.BuildKey, false);
if (policy != null)
{
return policy;
}
return context.Policies.Get<TPolicy>(context.BuildKey.Type, false);
}
private class CustomEffectiveInterceptionBehaviorsPolicy : IBuilderPolicy
{
public CustomEffectiveInterceptionBehaviorsPolicy()
{
this.Behaviors = new List<IInterceptionBehavior>();
}
public IEnumerable<IInterceptionBehavior> Behaviors { get; set; }
}
private class CustomDerivedTypeConstructorSelectorPolicy : IConstructorSelectorPolicy
{
private readonly Type interceptingType;
private readonly IConstructorSelectorPolicy originalConstructorSelectorPolicy;
public CustomDerivedTypeConstructorSelectorPolicy(Type interceptingType, IConstructorSelectorPolicy originalConstructorSelectorPolicy)
{
this.interceptingType = interceptingType;
this.originalConstructorSelectorPolicy = originalConstructorSelectorPolicy;
}
public SelectedConstructor SelectConstructor(IBuilderContext context, IPolicyList resolverPolicyDestination)
{
return FindNewConstructor(this.originalConstructorSelectorPolicy.SelectConstructor(context, resolverPolicyDestination), this.interceptingType);
}
private static SelectedConstructor FindNewConstructor(SelectedConstructor originalConstructor, Type interceptingType)
{
ParameterInfo[] parameters = originalConstructor.Constructor.GetParameters();
SelectedConstructor selectedConstructor = new SelectedConstructor(interceptingType.GetConstructor(parameters.Select(pi => pi.ParameterType).ToArray()));
foreach (string newKey in originalConstructor.GetParameterKeys())
{
selectedConstructor.AddParameterKey(newKey);
}
return selectedConstructor;
}
}
}
public class CustomInterception : Interception
{
protected override void Initialize()
{
this.Context.Strategies.AddNew<InstanceInterceptionStrategy>(UnityBuildStage.Setup);
this.Context.Strategies.AddNew<CustomTypeInterceptionStrategy>(UnityBuildStage.PreCreation);
this.Context.Container.RegisterInstance(typeof(AttributeDrivenPolicy).AssemblyQualifiedName, (InjectionPolicy)new AttributeDrivenPolicy());
}
}