diff --git a/src/main/java/graphql/servlet/internal/ApolloSubscriptionProtocolHandler.java b/src/main/java/graphql/servlet/internal/ApolloSubscriptionProtocolHandler.java index eab19774..dd0e29b8 100644 --- a/src/main/java/graphql/servlet/internal/ApolloSubscriptionProtocolHandler.java +++ b/src/main/java/graphql/servlet/internal/ApolloSubscriptionProtocolHandler.java @@ -58,7 +58,8 @@ public void onMessage(HandshakeRequest request, Session session, WsSessionSubscr subscriptions, message.id, input.getQueryInvoker().query(input.getInvocationInputFactory().create( - input.getGraphQLObjectMapper().getJacksonMapper().convertValue(message.payload, GraphQLRequest.class) + input.getGraphQLObjectMapper().getJacksonMapper().convertValue(message.payload, GraphQLRequest.class), + (HandshakeRequest) session.getUserProperties().get(HandshakeRequest.class.getName()) )) ); break;