Merge pull request #128 from darkl/master

Small refactor for #126
statianzo · Feb 24, 2015 · 5c09198 · 5c09198
2 parents 788daae + 6701509
commit 5c09198
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/Fleck/WebSocketConnection.cs b/src/Fleck/WebSocketConnection.cs
@@ -86,7 +86,10 @@ private Task Send<T>(T message, Func<T, byte[]> createFrame)
       {
           const string errorMessage = "Data sent while closing or after close. Ignoring.";
           FleckLog.Warn(errorMessage);
-          return Task.Factory.StartNew(() => { throw new ConnectionNotAvailableException(errorMessage); });
+
+          var taskForException = new TaskCompletionSource<object>();
+          taskForException.SetException(new ConnectionNotAvailableException(errorMessage));
+          return taskForException.Task;
       }
 
       var bytes = createFrame(message);