More cleanup for events

2024-08-24 17:00:54 +02:00
parent c9be7c9c2c
commit c7a52e8edb
3 changed files with 13 additions and 10 deletions
@@ -0,0 +1,77 @@
+using System.Text;
+using ChatTwo.Http.MessageProtocol;
+using Newtonsoft.Json;
+using WatsonWebserver.Core;
+
+namespace ChatTwo.Http;
+
+public class SSEConnection
+{
+    private long Index;
+    private bool Stopping;
+    private readonly CancellationToken Token;
+
+    public bool Done;
+    public readonly Stack<BaseMessage> OutboundStack = new();
+
+    public SSEConnection(CancellationToken token)
+    {
+        Token = token;
+    }
+
+    public async Task HandleEventLoop(HttpContextBase ctx)
+    {
+        try
+        {
+            ctx.Response.Headers.Add("Content-Type", "text/event-stream");
+            ctx.Response.Headers.Add("Cache-Control", "no-cache");
+            ctx.Response.Headers.Add("Connection", "keep-alive");
+
+            ctx.Response.ChunkedTransfer = true;
+            while (!Token.IsCancellationRequested && !Stopping)
+            {
+                await Task.Delay(10, Token);
+                if (Token.IsCancellationRequested)
+                    return;
+
+                if (!OutboundStack.TryPop(out var message))
+                    continue;
+
+                var data = JsonConvert.SerializeObject(message);
+                await ctx.Response.SendChunk(Encoding.UTF8.GetBytes($"id: {Index}\ndata: {data}\n\n"), Token);
+                Index++;
+            }
+        }
+        catch (TaskCanceledException)
+        {
+            // Ignore
+        }
+        catch (Exception ex)
+        {
+            Plugin.Log.Error(ex, "Event queued failed to continue");
+        }
+        finally
+        {
+            // "No Content" (204) didn't work for Firefox, so manually closing the connection on client side
+            await ctx.Response.SendFinalChunk("data: closing\nevent: close\n\n"u8.ToArray());
+
+            Done = true;
+        }
+    }
+
+    public async ValueTask DisposeAsync()
+    {
+        Stopping = true;
+
+        var timeout = 1000; // 1000ms
+        while (timeout > 0)
+        {
+            if (Done)
+                break;
+
+            timeout -= 100;
+            await Task.Delay(100);
+            Plugin.Log.Debug("Sleeping because EventServer still alive");
+        }
+    }
+}