SynchronizationContext Awaiter Extension Method

SynchronizationContextExtensions.cs

public static class SynchronizationContextExtensions {
    public struct SynchronizationContextAwaiter : INotifyCompletion {
        private readonly SynchronizationContext context;

        public SynchronizationContextAwaiter(SynchronizationContext context) {
            this.context = context;
        }

        public bool IsCompleted {
            get { return (context == null); }
        }

        public void GetResult() { }

        public void OnCompleted(Action continuation) {
            if (context == null) {
                continuation();
            }
            else {
                context.Send(state => continuation(), null);
            }
        }
    }

    public static SynchronizationContextAwaiter GetAwaiter(this SynchronizationContext context) {
        return new SynchronizationContextAwaiter(context);
    }
}