[Fix] Properly stop polling

AleXr64 · Oct 11, 2023 · b6239d2 · b6239d2
1 parent c19240e
commit b6239d2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/TGBotFramework/BotFramework/Bot.cs b/TGBotFramework/BotFramework/Bot.cs
@@ -31,6 +31,7 @@ public class Bot: IHostedService, ITelegramBot
         private TelegramBotClient client;
         private EventHandlerFactory factory;
 
+        private readonly CancellationTokenSource _stopPollingSource = new();
         private readonly LinkedList<Type> _wares;
 
         private string _userName;
@@ -55,6 +56,7 @@ async Task IHostedService.StopAsync(CancellationToken cancellationToken)
         {
             await client.DeleteWebhookAsync(cancellationToken:cancellationToken);
             await client.CloseAsync(cancellationToken);
+            _stopPollingSource.Cancel();
         }
 
         private async Task StartListen(CancellationToken cancellationToken)
@@ -109,7 +111,7 @@ private async Task StartListen(CancellationToken cancellationToken)
             }
             else
             {
-                client.StartReceiving(new DefaultUpdateHandler(HandleUpdateAsync, HandleErrorAsync), cancellationToken);
+                client.StartReceiving(new DefaultUpdateHandler(HandleUpdateAsync, HandleErrorAsync), _stopPollingSource.Token);
 
             }
         }