use ApplicationReadyEvent to init error handler

Author: Tad Sanden

graphql-kickstart-spring-support/build.gradle

@@ -2,6 +2,7 @@ dependencies {
     compileOnly "com.graphql-java-kickstart:graphql-java-kickstart:$LIB_GRAPHQL_SERVLET_VER"
     compileOnly "org.springframework:spring-web:$LIB_SPRING_CORE_VER"
     compileOnly "org.springframework:spring-context:$LIB_SPRING_CORE_VER"
+    compileOnly "org.springframework.boot:spring-boot-starter-web:$LIB_SPRING_BOOT_VER"
 
     testCompile "com.graphql-java-kickstart:graphql-java-kickstart:$LIB_GRAPHQL_SERVLET_VER"
     testCompile "com.graphql-java:graphql-java:$LIB_GRAPHQL_JAVA_VER"

graphql-kickstart-spring-support/src/main/java/graphql/kickstart/spring/error/GraphQLErrorStartupListener.java

@@ -1,12 +1,12 @@
 package graphql.kickstart.spring.error;
 
 import graphql.kickstart.execution.error.GraphQLErrorHandler;
+import org.springframework.boot.context.event.ApplicationReadyEvent;
 import org.springframework.context.ApplicationListener;
 import org.springframework.context.ConfigurableApplicationContext;
-import org.springframework.context.event.ContextRefreshedEvent;
 import org.springframework.lang.NonNull;
 
-public class GraphQLErrorStartupListener implements ApplicationListener<ContextRefreshedEvent> {
+public class GraphQLErrorStartupListener implements ApplicationListener<ApplicationReadyEvent> {
 
   private final ErrorHandlerSupplier errorHandlerSupplier;
   private final boolean exceptionHandlersEnabled;
@@ -17,11 +17,10 @@ public GraphQLErrorStartupListener(ErrorHandlerSupplier errorHandlerSupplier, bo
   }
 
   @Override
-  public void onApplicationEvent(@NonNull ContextRefreshedEvent event) {
-    ConfigurableApplicationContext context = (ConfigurableApplicationContext) event.getApplicationContext();
+  public void onApplicationEvent(@NonNull ApplicationReadyEvent event) {
+    ConfigurableApplicationContext context = event.getApplicationContext();
     GraphQLErrorHandler errorHandler = new GraphQLErrorHandlerFactory().create(context, exceptionHandlersEnabled);
     context.getBeanFactory().registerSingleton(errorHandler.getClass().getCanonicalName(), errorHandler);
     errorHandlerSupplier.setErrorHandler(errorHandler);
   }
-
 }