using System; using System.Threading; using System.Threading.Tasks; namespace Polly.Wrap { public class PolicyWrap : Policy, IPolicyWrap, IsPolicy { internal override void SetPolicyContext(Context executionContext) { if (executionContext.PolicyWrapKey == null) executionContext.PolicyWrapKey = base.PolicyKey; base.SetPolicyContext(executionContext); } internal PolicyWrap(Action<Action<Context, CancellationToken>, Context, CancellationToken> policyAction, Policy outer, Policy inner) : base(policyAction, outer.ExceptionPredicates) { } internal PolicyWrap(Func<Func<Context, CancellationToken, Task>, Context, CancellationToken, bool, Task> policyAction, Policy outer, Policy inner) : base(policyAction, outer.ExceptionPredicates) { } } }