From 947e3f36d8b1424775acfedd47f113c1f805fc7f Mon Sep 17 00:00:00 2001
From: Alexandr Daniyelian <5821262@gmail.com>
Date: Tue, 28 Aug 2018 15:33:31 +0300
Subject: [PATCH] #94 No HandshakeRequest in websocket subscription
 DataFetchingEnvironment

---
 .../servlet/internal/ApolloSubscriptionProtocolHandler.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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;