using System; using System.Diagnostics; using System.Threading; using System.Threading.Tasks; namespace Polly.Timeout { public class TimeoutPolicy : Policy, ITimeoutPolicy, IsPolicy { private Func<Context, TimeSpan> _timeoutProvider; private TimeoutStrategy _timeoutStrategy; private Action<Context, TimeSpan, Task, Exception> _onTimeout; internal TimeoutPolicy(Func<Context, TimeSpan> timeoutProvider, TimeoutStrategy timeoutStrategy, Action<Context, TimeSpan, Task, Exception> onTimeout) : base((PolicyBuilder)null) { if (timeoutProvider == null) throw new ArgumentNullException("timeoutProvider"); _timeoutProvider = timeoutProvider; _timeoutStrategy = timeoutStrategy; if (onTimeout == null) throw new ArgumentNullException("onTimeout"); _onTimeout = onTimeout; } [DebuggerStepThrough] protected override TResult Implementation<TResult>(Func<Context, CancellationToken, TResult> action, Context context, CancellationToken cancellationToken) { return TimeoutEngine.Implementation(action, context, cancellationToken, _timeoutProvider, _timeoutStrategy, _onTimeout); } } }