Mercurial > pub > ImplabNet
diff Implab/ActionChainTaskBase.cs @ 187:dd4a3590f9c6 ref20160224
Reworked cancelation handling, if the cancel handler isn't specified the OperationCanceledException will be handled by the error handler
Any unhandled OperationCanceledException will cause the promise cancelation
author | cin |
---|---|
date | Tue, 19 Apr 2016 17:35:20 +0300 |
parents | 75103928da09 |
children | 40d7fed4a09e |
line wrap: on
line diff
--- a/Implab/ActionChainTaskBase.cs Tue Apr 19 00:50:14 2016 +0300 +++ b/Implab/ActionChainTaskBase.cs Tue Apr 19 17:35:20 2016 +0300 @@ -2,12 +2,10 @@ using System.Threading; namespace Implab { - public class ActionChainTaskBase : AbstractPromise { + public class ActionChainTaskBase : AbstractTask { readonly Func<Exception, IPromise> m_error; readonly Func<Exception, IPromise> m_cancel; - int m_cancelationLock; - protected ActionChainTaskBase(Func<Exception, IPromise> error, Func<Exception, IPromise> cancel, bool autoCancellable) { m_error = error; m_cancel = cancel; @@ -21,55 +19,44 @@ } public override void CancelOperation(Exception reason) { - if (LockCancelation()) { - if (!(reason is OperationCanceledException)) - reason = reason != null ? new OperationCanceledException(null, reason) : new OperationCanceledException(); - - if (m_cancel != null) { - try { - m_cancel(reason).On(SetResult, HandleErrorInternal, HandleCancelInternal); - } catch (Exception err) { - HandleErrorInternal(err); - } - } else { - HandleErrorInternal(reason); + if (LockCancelation()) + // отмена вызвана до начала выполнения задачи + HandleCancelInternal(reason); + } + + protected void HandleCancelInternal(Exception reason) { + if (m_cancel != null) { + try { + // вызываем обработчик отмены + var p = m_cancel(reason); + p.On(SetResult, HandleErrorInternal, SetCancelledInternal); + // сообщаем асинхронной операции, что клиент уже не хочет получать результат + // т.е. если он инициировал отмену, задача отменилась, вызвался обрабочик отмены + // отбработчику сообщили, что результат уже не нужен и уже сам обработчик решает + // отдавать ли результат или подтвердить отмену (или вернуть ошибку). + CancellationRequested(p.Cancel); + } catch (Exception err) { + HandleErrorInternal(err); } + } else { + HandleErrorInternal(reason ?? new OperationCanceledException()); } } - void HandleCancelInternal(Exception reason) { - if (!(reason is OperationCanceledException)) - reason = reason != null ? new OperationCanceledException(null, reason) : new OperationCanceledException(); - HandleErrorInternal(reason); - } - - void HandleErrorInternal(Exception error) { + protected void HandleErrorInternal(Exception error) { if (m_error != null) { try { var p = m_error(error); - p.On(SetResult, SetError, SetCancelled); + p.On(SetResult, SetErrorInternal, SetCancelledInternal); CancellationRequested(p.Cancel); } catch (Exception err) { - error = err; + SetErrorInternal(error); } } else { SetErrorInternal(error); } } - void SetErrorInternal(Exception error) { - while (error is PromiseTransientException) - error = error.InnerException; - - if (error is OperationCanceledException) - SetCancelled(error); - else - SetError(error); - } - - protected bool LockCancelation() { - return 0 == Interlocked.CompareExchange(ref m_cancelationLock, 1, 0); - } } }