From bcc5716f970e9a1488d2dcc933a0560987cac048 Mon Sep 17 00:00:00 2001
From: Kei-Luna <touhousetsuna@gmail.com>
Date: Thu, 30 Apr 2026 15:46:00 +0900
Subject: [PATCH] Fix for the function to save to the database.

---
 Proxy/ProxyServer.cs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/Proxy/ProxyServer.cs b/Proxy/ProxyServer.cs
index 142c188..68785c4 100644
--- a/Proxy/ProxyServer.cs
+++ b/Proxy/ProxyServer.cs
@@ -74,12 +74,20 @@ public sealed class ProxyServer(
         catch (OperationCanceledException) when (stoppingToken.IsCancellationRequested)
         {
         }
+        catch (SocketException) when (stoppingToken.IsCancellationRequested)
+        {
+        }
+        catch (ObjectDisposedException) when (stoppingToken.IsCancellationRequested)
+        {
+        }
     }
 
-    public override Task StopAsync(CancellationToken cancellationToken)
+    public override async Task StopAsync(CancellationToken cancellationToken)
     {
+        // Cancel the BackgroundService token first so shutdown exceptions are treated as expected.
+        var stopTask = base.StopAsync(cancellationToken);
         _listener?.Stop();
-        return base.StopAsync(cancellationToken);
+        await stopTask;
     }
 
     private async Task HandleClientAsync(TcpClient client, CancellationToken cancellationToken)