From 3dedc315ca4114ccc8ddef4f27b50a8d2fb90e11 Mon Sep 17 00:00:00 2001
From: Infi <infiziert@protonmail.ch>
Date: Tue, 27 Aug 2024 16:40:36 +0200
Subject: [PATCH] Add SSE disconnect logic

---
 ChatTwo/Http/SSEConnection.cs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/ChatTwo/Http/SSEConnection.cs b/ChatTwo/Http/SSEConnection.cs
index f495c32..1dfd7e6 100644
--- a/ChatTwo/Http/SSEConnection.cs
+++ b/ChatTwo/Http/SSEConnection.cs
@@ -36,7 +36,13 @@ public class SSEConnection
                 if (!OutboundQueue.TryDequeue(out var outgoingEvent))
                     continue;
 
-                await ctx.Response.SendChunk(outgoingEvent.Build(), Token);
+                if (!await ctx.Response.SendChunk(outgoingEvent.Build(), Token))
+                {
+                    Plugin.Log.Information("SSE connection was unable to send new data");
+                    Plugin.Log.Information($"Client disconnected: {ctx.Guid}");
+                    return;
+                }
+
                 Index++;
             }
         }