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

Author: xperiandri

RELEASE_NOTES.md

@@ -235,3 +235,4 @@
 ### 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

@@ -346,25 +346,26 @@ type GraphQLWebSocketMiddleware<'Root>
             return Result.Error <| "{nameof ConnectionInit} timeout"
     }
 
-    member _.InvokeAsync (ctx : HttpContext) = task {
+    member _.InvokeAsync (ctx : HttpContext) =
         if not (ctx.Request.Path = endpointUrl) then
-            do! next.Invoke (ctx)
+            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.")
+            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)
-    }
+            next.Invoke (ctx)