C#.NET Socket Task Extension for DisconnectAsync

DisconnectAsyncTaskExtension.cs

namespace DisconnectAsyncTaskExtension
{
    static class DisconnectAsyncTaskExtension
    {
        public static Task<SocketAsyncEventArgs> DisconnectAsyncTask(this Socket that, SocketAsyncEventArgs disconnectArgs)
        {
            TaskCompletionSource<SocketAsyncEventArgs> tcs = new TaskCompletionSource<SocketAsyncEventArgs>();
            
            disconnectArgs.Completed += (object sender, SocketAsyncEventArgs completedArgs) => {
                try
                {
                    tcs.SetResult(completedArgs);
                }
                catch (Exception e) { tcs.TrySetException(e); }
            };

            try
            {
                bool isIOPending = that.DisconnectAsync(disconnectArgs);
                if (!isIOPending)
                {
                    tcs.SetResult(disconnectArgs);
                }
            }
            catch (Exception e) { tcs.TrySetException(e); }

            return tcs.Task;
        }
    }
}