Excluded GraphQLWebSocketMiddleware from exception stack trace if request is not a Web Socket

Author: xperiandri

RELEASE_NOTES.md

@@ -232,6 +232,7 @@
 * Migrated all solutions to SLNX format
 * Various performance optimizations and bug fixes
 
-### Unreleased
+### 3.1.0 - Unreleased
 
 * Added `TimeOnly` GraphQL type
+* Excluded `GraphQLWebSocketMiddleware` from exception stack trace if request not a Web Socket

src/FSharp.Data.GraphQL.Server.AspNetCore/GraphQLWebsocketMiddleware.fs

@@ -25,7 +25,6 @@ open FSharp.Data.GraphQL.Shared.WebSockets
 
 type GraphQLWebSocketMiddleware<'Root>
     (
-        next : RequestDelegate,
         applicationLifetime : IHostApplicationLifetime,
         serviceProvider : IServiceProvider,
         logger : ILogger<GraphQLWebSocketMiddleware<'Root>>,
@@ -35,7 +34,6 @@ type GraphQLWebSocketMiddleware<'Root>
     let options = options.Value
     let serializerOptions = options.SerializerOptions
     let pingHandler = options.WebsocketOptions.CustomPingHandler
-    let endpointUrl = PathString options.WebsocketOptions.EndpointUrl
     let connectionInitTimeout = options.WebsocketOptions.ConnectionInitTimeout
 
     let serializeServerMessage (jsonSerializerOptions : JsonSerializerOptions) (serverMessage : ServerMessage) = task {
@@ -346,25 +344,29 @@ type GraphQLWebSocketMiddleware<'Root>
             return Result.Error <| "{nameof ConnectionInit} timeout"
     }
 
-    member _.InvokeAsync (ctx : HttpContext) = task {
-        if not (ctx.Request.Path = endpointUrl) then
-            do! next.Invoke (ctx)
-        else if ctx.WebSockets.IsWebSocketRequest then
-            use! socket = ctx.WebSockets.AcceptWebSocketAsync ("graphql-transport-ws")
-            let! connectionInitResult = socket |> waitForConnectionInitAndRespondToClient
-            match connectionInitResult with
-            | Result.Error errMsg -> logger.LogWarning errMsg
-            | Ok _ ->
-                let longRunningCancellationToken =
-                    (CancellationTokenSource
-                        .CreateLinkedTokenSource(ctx.RequestAborted, applicationLifetime.ApplicationStopping)
-                        .Token)
-                longRunningCancellationToken.Register (fun _ -> (socket |> tryToGracefullyCloseSocketWithDefaultBehavior).Wait ())
-                |> ignore
-                try
-                    do! socket |> handleMessages longRunningCancellationToken ctx
-                with ex ->
-                    logger.LogError (ex, "Cannot handle WebSocket message.")
+    member _.InvokeAsync (ctx : HttpContext) : Task =
+        if ctx.WebSockets.IsWebSocketRequest then
+            task {
+                use! socket = ctx.WebSockets.AcceptWebSocketAsync ("graphql-transport-ws")
+                let! connectionInitResult = socket |> waitForConnectionInitAndRespondToClient
+                match connectionInitResult with
+                | Result.Error errMsg -> logger.LogWarning errMsg
+                | Ok _ ->
+                    let longRunningCancellationToken =
+                        (CancellationTokenSource
+                            .CreateLinkedTokenSource(ctx.RequestAborted, applicationLifetime.ApplicationStopping)
+                            .Token)
+                    longRunningCancellationToken.Register (fun _ -> (socket |> tryToGracefullyCloseSocketWithDefaultBehavior).Wait ())
+                    |> ignore
+                    try
+                        do! socket |> handleMessages longRunningCancellationToken ctx
+                    with ex ->
+                        logger.LogError (ex, "Cannot handle WebSocket message.")
+            }
         else
-            do! next.Invoke (ctx)
-    }
+            TypedResults.Problem (
+                title = "WebSocket connection expected.",
+                detail = $"'{options.WebsocketOptions.EndpointUrl}' endpoint only accepts WebSocket connections.",
+                statusCode = StatusCodes.Status400BadRequest
+            ) :> IResult
+            |> _.ExecuteAsync(ctx)

src/FSharp.Data.GraphQL.Server.AspNetCore/ServiceCollectionExtensions.fs

@@ -329,5 +329,18 @@ module ApplicationBuilderExtensions =
 
     type IApplicationBuilder with
 
+        /// <summary>
+        /// Registers the GraphQL WebSocket middleware to handle WebSocket connections at the configured endpoint.
+        /// The middleware will only be applied to requests matching the endpoint path configured in <see cref="GraphQLOptions" />.
+        /// </summary>
         [<Extension; CompiledName "UseWebSocketsForGraphQL">]
-        member builder.UseWebSocketsForGraphQL<'Root> () = builder.UseMiddleware<GraphQLWebSocketMiddleware<'Root>> ()
+        member builder.UseWebSocketsForGraphQL<'Root> () =
+
+            let options = builder.ApplicationServices.GetRequiredService<IOptions<GraphQLOptions<'Root>>>()
+            let endpointPath = PathString options.Value.WebsocketOptions.EndpointUrl
+
+            builder.UseWhen(
+                (fun ctx -> ctx.Request.Path = endpointPath),
+                fun appBuilder ->
+                    appBuilder.UseMiddleware<GraphQLWebSocketMiddleware<'Root>>() |> ignore
+            )