PromiseExtensions.cs
        
        
            
                    187 lines
            
             | 6.2 KiB
            
                | text/x-csharp
            
             |
                CSharpLexer
            
          
        
             / Implab / PromiseExtensions.cs
          
          
          
      |  | r72 | using System.Threading; | ||
|  | r75 | using System; | ||
|  | r109 | using Implab.Diagnostics; | ||
|  | r119 | using System.Collections.Generic; | ||
|  | r109 | |||
|  | r75 | #if NET_4_5 | ||
| using System.Threading.Tasks; | ||||
| #endif | ||||
|  | r72 | |||
| namespace Implab { | ||||
| public static class PromiseExtensions { | ||||
| public static IPromise<T> DispatchToCurrentContext<T>(this IPromise<T> that) { | ||||
|  | r75 | Safe.ArgumentNotNull(that, "that"); | ||
|  | r72 | var context = SynchronizationContext.Current; | ||
| if (context == null) | ||||
| return that; | ||||
|  | r119 | var p = new SyncContextPromise<T>(context); | ||
| p.On(that.Cancel, PromiseEventType.Cancelled); | ||||
|  | r72 | |||
|  | r104 | that.On( | ||
|  | r76 | p.Resolve, | ||
| p.Reject, | ||||
| p.Cancel | ||||
|  | r72 | ); | ||
| return p; | ||||
| } | ||||
| public static IPromise<T> DispatchToContext<T>(this IPromise<T> that, SynchronizationContext context) { | ||||
|  | r75 | Safe.ArgumentNotNull(that, "that"); | ||
|  | r72 | Safe.ArgumentNotNull(context, "context"); | ||
|  | r119 | var p = new SyncContextPromise<T>(context); | ||
| p.On(that.Cancel, PromiseEventType.Cancelled); | ||||
|  | r72 | |||
|  | r104 | that.On( | ||
|  | r76 | p.Resolve, | ||
| p.Reject, | ||||
| p.Cancel | ||||
|  | r72 | ); | ||
| return p; | ||||
| } | ||||
|  | r75 | |||
|  | r101 | /// <summary> | ||
| /// Ensures the dispatched. | ||||
| /// </summary> | ||||
| /// <returns>The dispatched.</returns> | ||||
| /// <param name="that">That.</param> | ||||
| /// <param name="head">Head.</param> | ||||
| /// <param name="cleanup">Cleanup.</param> | ||||
| /// <typeparam name="TPromise">The 1st type parameter.</typeparam> | ||||
| /// <typeparam name="T">The 2nd type parameter.</typeparam> | ||||
| public static TPromise EnsureDispatched<TPromise,T>(this TPromise that, IPromise<T> head, Action<T> cleanup) where TPromise : IPromise{ | ||||
| Safe.ArgumentNotNull(that, "that"); | ||||
| Safe.ArgumentNotNull(head, "head"); | ||||
|  | r104 | that.On(null,null,() => head.On(cleanup)); | ||
|  | r101 | |||
| return that; | ||||
| } | ||||
|  | r75 | public static AsyncCallback AsyncCallback<T>(this Promise<T> that, Func<IAsyncResult,T> callback) { | ||
| Safe.ArgumentNotNull(that, "that"); | ||||
| Safe.ArgumentNotNull(callback, "callback"); | ||||
|  | r109 | var op = TraceContext.Instance.CurrentOperation; | ||
|  | r75 | return ar => { | ||
|  | r109 | TraceContext.Instance.EnterLogicalOperation(op,false); | ||
|  | r75 | try { | ||
| that.Resolve(callback(ar)); | ||||
| } catch (Exception err) { | ||||
| that.Reject(err); | ||||
|  | r109 | } finally { | ||
| TraceContext.Instance.Leave(); | ||||
|  | r75 | } | ||
| }; | ||||
| } | ||||
|  | r110 | |||
| static void CancelCallback(object cookie) { | ||||
| ((ICancellable)cookie).Cancel(); | ||||
| } | ||||
| /// <summary> | ||||
| /// Cancells promise after the specified timeout is elapsed. | ||||
| /// </summary> | ||||
| /// <param name="that">The promise to cancel on timeout.</param> | ||||
| /// <param name="milliseconds">The timeout in milliseconds.</param> | ||||
| /// <typeparam name="TPromise">The 1st type parameter.</typeparam> | ||||
| public static TPromise Timeout<TPromise>(this TPromise that, int milliseconds) where TPromise : IPromise { | ||||
| Safe.ArgumentNotNull(that, "that"); | ||||
| var timer = new Timer(CancelCallback, that, milliseconds, -1); | ||||
| that.On(timer.Dispose, PromiseEventType.All); | ||||
| return that; | ||||
| } | ||||
|  | r119 | |||
|  | r124 | public static IPromise Bundle(this ICollection<IPromise> that) { | ||
|  | r119 | Safe.ArgumentNotNull(that, "that"); | ||
| int count = that.Count; | ||||
|  | r124 | int errors = 0; | ||
|  | r119 | var medium = new Promise(); | ||
|  | r124 | medium.On(() => { | ||
| foreach(var p2 in that) | ||||
| p2.Cancel(); | ||||
| }, PromiseEventType.ErrorOrCancel); | ||||
|  | r119 | foreach (var p in that) | ||
| p.On( | ||||
| () => { | ||||
| if (Interlocked.Decrement(ref count) == 0) | ||||
| medium.Resolve(); | ||||
| }, | ||||
| error => { | ||||
|  | r124 | if (Interlocked.Increment(ref errors) == 1) | ||
| medium.Reject( | ||||
| new Exception("The dependency promise is failed", error) | ||||
| ); | ||||
|  | r119 | }, | ||
| () => { | ||||
|  | r124 | if (Interlocked.Increment(ref errors) == 1) | ||
| medium.Reject( | ||||
| new Exception("The dependency promise is cancelled") | ||||
| ); | ||||
|  | r119 | } | ||
| ); | ||||
| return medium; | ||||
| } | ||||
|  | r124 | |||
| public static IPromise<T[]> Bundle<T>(this ICollection<IPromise<T>> that) { | ||||
| Safe.ArgumentNotNull(that, "that"); | ||||
| int count = that.Count; | ||||
| int errors = 0; | ||||
| var medium = new Promise<T[]>(); | ||||
| var results = new T[that.Count]; | ||||
| medium.On(() => { | ||||
| foreach(var p2 in that) | ||||
| p2.Cancel(); | ||||
| }, PromiseEventType.ErrorOrCancel); | ||||
| int i = 0; | ||||
| foreach (var p in that) { | ||||
| var idx = i; | ||||
| p.On( | ||||
| x => { | ||||
| results[idx] = x; | ||||
| if (Interlocked.Decrement(ref count) == 0) | ||||
| medium.Resolve(results); | ||||
| }, | ||||
| error => { | ||||
| if (Interlocked.Increment(ref errors) == 1) | ||||
| medium.Reject( | ||||
| new Exception("The dependency promise is failed", error) | ||||
| ); | ||||
| }, | ||||
| () => { | ||||
| if (Interlocked.Increment(ref errors) == 1) | ||||
| medium.Reject( | ||||
| new Exception("The dependency promise is cancelled") | ||||
| ); | ||||
| } | ||||
| ); | ||||
| i++; | ||||
| } | ||||
| return medium; | ||||
| } | ||||
|  | r75 | |||
| #if NET_4_5 | ||||
| public static Task<T> GetTask<T>(this IPromise<T> that) { | ||||
| Safe.ArgumentNotNull(that, "that"); | ||||
| var tcs = new TaskCompletionSource<T>(); | ||||
|  | r104 | that.On(tcs.SetResult, tcs.SetException, tcs.SetCanceled); | ||
|  | r75 | |||
| return tcs.Task; | ||||
| } | ||||
| #endif | ||||
|  | r72 | } | ||
| } | ||||
