fix: Handle TaskCanceledException when flush is called

AB#47227

Author: andrew-hardy

docs/4.4.3-release-notes.md

@@ -0,0 +1,2 @@
+### Fixes
+- Handle TaskCanceledException when stop/starting a stream

src/Connector.DataLake.Common/Buffer.cs

@@ -75,7 +75,14 @@ private async Task Idle()
         {
             while (true)
             {
-                await Task.Delay(100, _idleCancellationTokenSource.Token);
+                try
+                {
+                    await Task.Delay(100, _idleCancellationTokenSource.Token);
+                }
+                catch (TaskCanceledException taskCanceledException)
+                {
+                    // TODO add log
+                }
 
                 if (!_idleCancellationTokenSource.IsCancellationRequested && DateTime.Now.Subtract(_lastAdded).TotalMilliseconds < _timeout)
                 {
@@ -159,7 +166,9 @@ public async Task Flush()
             var t = _idleTask;
             if (t != null)
             {
+                Console.WriteLine("Canceling");
                 _idleCancellationTokenSource.Cancel();
+                Console.WriteLine("Canceled");
 
                 await t;
             }