-
Notifications
You must be signed in to change notification settings - Fork 65
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
affc4c0
commit 908a466
Showing
10 changed files
with
170 additions
and
477 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
82 changes: 82 additions & 0 deletions
82
...-callback/src/main/java/com/apollographql/subscription/CallbackWebGraphQLInterceptor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,82 @@ | ||
| package com.apollographql.subscription; | ||
|
|
||
| import static com.apollographql.subscription.callback.SubscriptionCallback.parseSubscriptionCallbackExtension; | ||
| import static com.apollographql.subscription.callback.SubscriptionCallbackHandler.SUBSCRIPTION_PROTOCOL_HEADER; | ||
| import static com.apollographql.subscription.callback.SubscriptionCallbackHandler.SUBSCRIPTION_PROTOCOL_HEADER_VALUE; | ||
|
|
||
| import com.apollographql.subscription.callback.SubscriptionCallbackHandler; | ||
| import graphql.ExecutionResult; | ||
| import graphql.GraphQLError; | ||
| import org.apache.commons.logging.Log; | ||
| import org.apache.commons.logging.LogFactory; | ||
| import org.springframework.graphql.server.WebGraphQlInterceptor; | ||
| import org.springframework.graphql.server.WebGraphQlRequest; | ||
| import org.springframework.graphql.server.WebGraphQlResponse; | ||
| import org.springframework.graphql.server.WebSocketGraphQlRequest; | ||
| import org.springframework.graphql.support.DefaultExecutionGraphQlResponse; | ||
| import reactor.core.publisher.Mono; | ||
|
|
||
| public class CallbackWebGraphQLInterceptor implements WebGraphQlInterceptor { | ||
|
|
||
| private static final Log logger = LogFactory.getLog(CallbackWebGraphQLInterceptor.class); | ||
|
|
||
| private final SubscriptionCallbackHandler subscriptionCallbackHandler; | ||
|
|
||
| public CallbackWebGraphQLInterceptor(SubscriptionCallbackHandler subscriptionCallbackHandler) { | ||
| this.subscriptionCallbackHandler = subscriptionCallbackHandler; | ||
| } | ||
|
|
||
| @Override | ||
| public Mono<WebGraphQlResponse> intercept(WebGraphQlRequest request, Chain chain) { | ||
| // in order to correctly handle parsing of ANY requests (i.e. it is valid to define a | ||
| // document with query fragments first) we would need to parse it which is a much heavier | ||
| // operation, we may opt to do it in the future releases | ||
| if (!isWebSocketRequest(request) && request.getDocument().startsWith("subscription")) { | ||
| return parseSubscriptionCallbackExtension(request.getExtensions()) | ||
| .flatMap( | ||
| callback -> { | ||
| if (logger.isDebugEnabled()) { | ||
| logger.debug("Starting subscription using callback: " + callback); | ||
| } | ||
| return this.subscriptionCallbackHandler | ||
| .handleSubscriptionUsingCallback(request, callback) | ||
| .map(response -> callbackResponse(request, response)); | ||
| }) | ||
| .onErrorResume( | ||
| (error) -> { | ||
| if (logger.isErrorEnabled()) { | ||
| logger.error("Unable to start subscription using callback protocol", error); | ||
| } | ||
| return Mono.just(errorCallbackResponse(request)); | ||
| }); | ||
| } else { | ||
| return chain.next(request); | ||
| } | ||
| } | ||
|
|
||
| private boolean isWebSocketRequest(WebGraphQlRequest request) { | ||
| return request instanceof WebSocketGraphQlRequest; | ||
| } | ||
|
|
||
| private WebGraphQlResponse callbackResponse( | ||
| WebGraphQlRequest request, ExecutionResult callbackResult) { | ||
| var callbackExecutionResponse = | ||
| new DefaultExecutionGraphQlResponse(request.toExecutionInput(), callbackResult); | ||
| var callbackGraphQLResponse = new WebGraphQlResponse(callbackExecutionResponse); | ||
| callbackGraphQLResponse | ||
| .getResponseHeaders() | ||
| .add(SUBSCRIPTION_PROTOCOL_HEADER, SUBSCRIPTION_PROTOCOL_HEADER_VALUE); | ||
| return callbackGraphQLResponse; | ||
| } | ||
|
|
||
| private WebGraphQlResponse errorCallbackResponse(WebGraphQlRequest request) { | ||
| var errorCallbackResult = | ||
| ExecutionResult.newExecutionResult() | ||
| .addError( | ||
| GraphQLError.newError() | ||
| .message("Unable to start subscription using callback protocol") | ||
| .build()) | ||
| .build(); | ||
| return callbackResponse(request, errorCallbackResult); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
15 changes: 15 additions & 0 deletions
15
.../java/com/apollographql/subscription/exception/CallbackInitializationFailedException.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,15 @@ | ||
| package com.apollographql.subscription.exception; | ||
|
|
||
| import com.apollographql.subscription.callback.SubscriptionCallback; | ||
|
|
||
| /** Exception thrown when callback initialization fails. */ | ||
| public class CallbackInitializationFailedException extends RuntimeException { | ||
|
|
||
| public CallbackInitializationFailedException(SubscriptionCallback callback, int statusCode) { | ||
| super( | ||
| "Subscription callback failed initialization: " | ||
| + callback | ||
| + ", server responded with: " | ||
| + statusCode); | ||
| } | ||
| } |
Oops, something went wrong.