[FIX] Properly stop polling

AleXr64 · Oct 11, 2023 · 6a84e6d · 6a84e6d
1 parent 0346f4e
commit 6a84e6d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/TGBotFramework/BotFramework/Bot.cs b/TGBotFramework/BotFramework/Bot.cs
@@ -33,6 +33,8 @@ public class Bot: IHostedService, IBotInstance
 
         private EventHandlerFactory factory;
 
+        private CancellationTokenSource _recieveToken = new CancellationTokenSource();
+
         public Bot(IServiceProvider services, IServiceScopeFactory scopeFactory, Type startupType = null)
         {
             this.services = services;
@@ -50,6 +52,7 @@ async Task IHostedService.StopAsync(CancellationToken cancellationToken)
         {
             await client.DeleteWebhookAsync(cancellationToken: cancellationToken);
             await client.CloseAsync(cancellationToken);
+            _recieveToken.Cancel();
         }
 
         public string UserName { get; private set; }
@@ -120,7 +123,7 @@ private async Task StartListen(CancellationToken cancellationToken)
                 await client.DeleteWebhookAsync(false, cancellationToken);
 
                 var receiverOptions = new ReceiverOptions { AllowedUpdates = { }, ThrowPendingUpdates = false };
-                client.StartReceiving(HandleUpdateAsync, HandleErrorAsync, receiverOptions, cancellationToken);
+                client.StartReceiving(HandleUpdateAsync, HandleErrorAsync, receiverOptions, _recieveToken.Token);
             }
         }