feat: refactor spring contexts to use new map of contexts instead

Author: oliemansm

graphql-kickstart-spring-support/src/main/java/graphql/kickstart/spring/GraphQLSpringServerWebExchangeContext.java

@@ -1,28 +1,31 @@
 package graphql.kickstart.spring;
 
 import graphql.kickstart.execution.context.DefaultGraphQLContext;
-import java.util.Objects;
+import lombok.NonNull;
 import org.dataloader.DataLoaderRegistry;
 import org.springframework.web.server.ServerWebExchange;
 
 public class GraphQLSpringServerWebExchangeContext extends DefaultGraphQLContext
     implements GraphQLSpringContext {
 
-  private final ServerWebExchange serverWebExchange;
-
   public GraphQLSpringServerWebExchangeContext(ServerWebExchange serverWebExchange) {
     this(new DataLoaderRegistry(), serverWebExchange);
   }
 
   public GraphQLSpringServerWebExchangeContext(
-      DataLoaderRegistry dataLoaderRegistry, ServerWebExchange serverWebExchange) {
-    super(dataLoaderRegistry, null);
-    this.serverWebExchange =
-        Objects.requireNonNull(serverWebExchange, "Server web exchange cannot be null");
+      DataLoaderRegistry dataLoaderRegistry, @NonNull ServerWebExchange serverWebExchange) {
+    super(dataLoaderRegistry);
+    put(ServerWebExchange.class, serverWebExchange);
   }
 
+  /**
+   * @deprecated Use {@code
+   *     dataFetchingEnvironment.getGraphQlContext().get(ServerWebExchange.class)} instead. Since
+   *     13.0.0
+   */
   @Override
+  @Deprecated
   public ServerWebExchange getServerWebExchange() {
-    return serverWebExchange;
+    return (ServerWebExchange) getMapOfContext().get(ServerWebExchange.class);
   }
 }

graphql-kickstart-spring-webflux/src/main/java/graphql/kickstart/spring/webflux/DefaultGraphQLSpringWebSocketSessionContext.java

@@ -1,28 +1,29 @@
 package graphql.kickstart.spring.webflux;
 
 import graphql.kickstart.execution.context.DefaultGraphQLContext;
-import java.util.Objects;
+import lombok.NonNull;
 import org.dataloader.DataLoaderRegistry;
 import org.springframework.web.reactive.socket.WebSocketSession;
 
 public class DefaultGraphQLSpringWebSocketSessionContext extends DefaultGraphQLContext
     implements GraphQLSpringWebSocketSessionContext {
 
-  private final WebSocketSession webSocketSession;
-
   public DefaultGraphQLSpringWebSocketSessionContext(WebSocketSession webSocketSession) {
     this(new DataLoaderRegistry(), webSocketSession);
   }
 
   public DefaultGraphQLSpringWebSocketSessionContext(
-      DataLoaderRegistry dataLoaderRegistry, WebSocketSession webSocketSession) {
-    super(dataLoaderRegistry, null);
-    this.webSocketSession =
-        Objects.requireNonNull(webSocketSession, "WebSocketSession is required");
+      DataLoaderRegistry dataLoaderRegistry, @NonNull WebSocketSession webSocketSession) {
+    super(dataLoaderRegistry);
+    put(WebSocketSession.class, webSocketSession);
   }
 
+  /**
+   * @deprecated Use {@code dataFetchingEnvironment.getGraphQlContext().get(WebSocketSession.class)}
+   *     instead. Since 13.0.0
+   */
   @Override
   public WebSocketSession getWebSocketSession() {
-    return webSocketSession;
+    return (WebSocketSession) getMapOfContext().get(WebSocketSession.class);
   }
 }