Changeset 499 for TFP-WebServer/WebServer/src/SSE

Timestamp:

Jul 23, 2024, 6:49:56 PM (4 months ago)

Author:

alloc

Message:

*Fixed: Chat code
*Fixed: SSE connection counting, added connection set up logging

Location:

TFP-WebServer/WebServer/src/SSE

Files:

• AbsEvent.cs (modified) (3 diffs)
• SseClient.cs (modified) (2 diffs)

TFP-WebServer/WebServer/src/SSE/AbsEvent.cs

@@ -36 +36 @@
 
                         openClients.Add (_client);
+                        
+                        logConnectionState ("Connection opened", _client);
                 }
 
@@ -64 +66 @@
                                                 bytesToSend = Encoding.UTF8.GetBytes (output, 0, output.Length, buf, 0);
                                         } catch (ArgumentException e) {
-                                                logError ("Exception while encoding data for output, most likely exceeding buffer size:", false);
+                                                Log.Error ($"[Web] [SSE] '{Name}': Exception while encoding data for output, most likely exceeding buffer size:");
                                                 Log.Exception (e);
                                                 return;
@@ -79 +81 @@
                 private void sendBufToListeners (byte[] _bytes, int _bytesToSend) {
                         for (int i = openClients.Count - 1; i >= 0; i--) {
-                                ESseClientWriteResult writeResult = openClients [i].Write (_bytes, _bytesToSend);
-                                if (writeResult == ESseClientWriteResult.Ok) {
-                                        continue;
-                                }
-
-                                currentlyOpen--;
-                                totalClosed++;
-
-                                if (writeResult == ESseClientWriteResult.Error) {
-                                        logError ("Can not write to endpoint, closing", true);
-                                }
+                                openClients [i].Write (_bytes, _bytesToSend);
                         }
-                }
-
-                protected void logError (string _message, bool _printConnections) {
-                        Log.Error (_printConnections
-                                ? $"[Web] [SSE] '{Name}': {_message} (Left open: {currentlyOpen}, total opened: {totalOpened}, closed: {totalClosed})"
-                                : $"[Web] [SSE] '{Name}': {_message}");
                 }
 
                 public virtual int DefaultPermissionLevel () => 0;
 
+                private void logConnectionState (string _message, SseClient _client) {
+                        Log.Out ($"[Web] [SSE] '{Name}': {_message} from {_client.RemoteEndpoint} (Left open: {currentlyOpen}, total opened: {totalOpened}, closed: {totalClosed})");
+                }
+
                 public void ClientClosed (SseClient _client) {
-                        openClients.Remove (_client);
+                        if (openClients.Remove (_client)) {
+                                currentlyOpen--;
+                                totalClosed++;
+
+                                logConnectionState ("Closed connection", _client);
+                        }
                 }
         }

TFP-WebServer/WebServer/src/SSE/SseClient.cs

@@ -1 +1 @@
 using System;
 using System.IO;
+using System.Net;
 using System.Net.Sockets;
 using System.Text;
-using SpaceWizards.HttpListener;
 using Webserver.UrlHandlers;
+using HttpListenerResponse = SpaceWizards.HttpListener.HttpListenerResponse;
 
 namespace Webserver.SSE {
@@ -17 +18 @@
         private static readonly byte[] keepAliveData = Encoding.UTF8.GetBytes (": KeepAlive\n\n");
 
+        public readonly IPEndPoint RemoteEndpoint;
         private readonly SseHandler parent;
         private readonly HttpListenerResponse response;
         private DateTime lastMessageSent = DateTime.Now;
 
-        public SseClient (SseHandler _parent, HttpListenerResponse _response) {
+        public SseClient (SseHandler _parent, RequestContext _context) {
             parent = _parent;
-            response = _response;
+            response = _context.Response;
+            RemoteEndpoint = _context.Request.RemoteEndPoint;
             
             // Keep the request open
-            _response.SendChunked = true;
+            response.SendChunked = true;
 
-            _response.AddHeader ("Content-Type", "text/event-stream");
-            _response.OutputStream.Flush ();
+            response.AddHeader ("Content-Type", "text/event-stream");
+            response.OutputStream.Flush ();
         }