Added user state to socket interceptor

ChilliCream · Feb 3, 2023 · 981104d · 981104d
1 parent 967365a
commit 981104d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/HotChocolate/AspNetCore/src/AspNetCore/DefaultSocketSessionInterceptor.cs b/src/HotChocolate/AspNetCore/src/AspNetCore/DefaultSocketSessionInterceptor.cs
@@ -21,13 +21,17 @@ public class DefaultSocketSessionInterceptor : ISocketSessionInterceptor
         CancellationToken cancellationToken = default)
     {
         var context = session.Connection.HttpContext;
+        var userState = new UserState(context.User);
+
         requestBuilder.TrySetServices(session.Connection.RequestServices);
         requestBuilder.TryAddGlobalState(nameof(CancellationToken), session.Connection.RequestAborted);
         requestBuilder.TryAddGlobalState(nameof(HttpContext), context);
-        requestBuilder.TryAddGlobalState(nameof(ClaimsPrincipal), context.User);
         requestBuilder.TryAddGlobalState(nameof(ISocketSession), session);
         requestBuilder.TryAddGlobalState(OperationSessionId, operationSessionId);
 
+        requestBuilder.TryAddGlobalState(nameof(ClaimsPrincipal), userState.User);
+        requestBuilder.TryAddGlobalState(WellKnownContextData.UserState, userState);
+
         if (context.IsTracingEnabled())
         {
             requestBuilder.TryAddGlobalState(EnableTracing, true);