feat: support SSE (#2)

atmina · May 7, 2024 · 1ce74f0 · 1ce74f0
1 parent eebee43
commit 1ce74f0
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/Fluss.HotChocolate/AddExtensionMiddleware.cs b/src/Fluss.HotChocolate/AddExtensionMiddleware.cs
@@ -37,8 +37,10 @@ public async ValueTask InvokeAsync(IRequestContext context)
             return;
         }
 
-        if (true != context.Services.GetRequiredService<IHttpContextAccessor>().HttpContext?.WebSockets
-                .IsWebSocketRequest)
+        var httpContext = context.Services.GetRequiredService<IHttpContextAccessor>().HttpContext;
+        var isWebsocket = true == httpContext?.WebSockets.IsWebSocketRequest;
+        var isSse = httpContext?.Request.Headers.Accept.ToString() == "text/event-stream";
+        if (!isWebsocket && !isSse)
         {
             return;
         }
@@ -76,16 +78,18 @@ private async IAsyncEnumerable<IQueryResult> LiveResults(IReadOnlyDictionary<str
             yield break;
         }
 
+        contextData.TryGetValue(nameof(HttpContext), out var httpContext);
+        var isWebsocket = (httpContext as HttpContext)?.WebSockets.IsWebSocketRequest ?? false;
         var foundSocketSession = contextData.TryGetValue(nameof(ISocketSession), out var contextSocketSession); // as ISocketSession
         var foundOperationId = contextData.TryGetValue("HotChocolate.Execution.Transport.OperationSessionId", out var operationId); // as string
 
-        if (!foundSocketSession || !foundOperationId)
+        if (isWebsocket && (!foundSocketSession || !foundOperationId))
         {
             _logger.LogWarning("Trying to add live results but {SocketSession} or {OperationId} is not present in context!", nameof(contextSocketSession), nameof(operationId));
             yield break;
         }
 
-        if (contextSocketSession is not ISocketSession socketSession)
+        if (isWebsocket && contextSocketSession is not ISocketSession)
         {
             _logger.LogWarning("{ContextSocketSession} key present in context but not an {ISocketSession}!", contextSocketSession?.GetType().FullName, nameof(ISocketSession));
             yield break;
@@ -108,7 +112,7 @@ private async IAsyncEnumerable<IQueryResult> LiveResults(IReadOnlyDictionary<str
                 unitOfWork.ReadModels
             );
 
-            if (socketSession.Operations.All(operationSession => operationSession.Id != operationId?.ToString()))
+            if (isWebsocket && contextSocketSession is ISocketSession socketSession && socketSession.Operations.All(operationSession => operationSession.Id != operationId?.ToString()))
             {
                 break;
             }