use completion source

Gitii · Apr 17, 2022 · e205c9c · e205c9c
1 parent e4feb2c
commit e205c9c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/Community.Wsl.Sdk/Strategies/Commands/StreamDataReader.cs b/Community.Wsl.Sdk/Strategies/Commands/StreamDataReader.cs
@@ -45,7 +45,10 @@ public void Fetch()
                 var data = stream.ToArray();
                 Finished(data);
             }
-        ) { IsBackground = true };
+        )
+        {
+            IsBackground = true
+        };
 
         _thread.Start();
     }
@@ -57,7 +60,7 @@ public virtual void CopyResultTo(ref CommandResult result, bool isStdOut)
             throw new Exception("Data hasn't been fetched, yet!");
         }
 
-        if (_thread.ThreadState != ThreadState.Stopped)
+        if (!_completionSource?.Task.IsCompleted ?? false)
         {
             throw new Exception("Fetching hasn't been finished, yet!");
         }
@@ -84,6 +87,6 @@ public Task WaitAsync()
             return Task.CompletedTask;
         }
 
-        return _completionSource.Task.ContinueWith(task => Task.Delay(1), TaskScheduler.Default);
+        return _completionSource.Task;
     }
 }