Examples with WebInput org.springframework.graphql.web.WebInput used on opensource projects

Example 1 with WebInput

use of org.springframework.graphql.web.WebInput in project spring-graphql by spring-projects.

the class WebGraphQlClientBuilderTests method mutateDocumentSource.

@ParameterizedTest
@MethodSource("argumentSource")
void mutateDocumentSource(ClientBuilderSetup builderSetup) {
    DocumentSource documentSource = name -> name.equals("name") ? Mono.just(DOCUMENT) : Mono.error(new IllegalArgumentException());
    // Original
    WebGraphQlClient.Builder<?> builder = builderSetup.initBuilder().documentSource(documentSource);
    WebGraphQlClient client = builder.build();
    client.documentName("name").execute().block(TIMEOUT);
    WebInput input = builderSetup.getWebInput();
    assertThat(input.getDocument()).isEqualTo(DOCUMENT);
    // Mutate
    client = client.mutate().build();
    client.documentName("name").execute().block(TIMEOUT);
    input = builderSetup.getWebInput();
    assertThat(input.getDocument()).isEqualTo(DOCUMENT);
}

Example 2 with WebInput

use of org.springframework.graphql.web.WebInput in project spring-graphql by spring-projects.

the class WebGraphQlTesterBuilderTests method mutateDocumentSource.

@ParameterizedTest
@MethodSource("argumentSource")
void mutateDocumentSource(TesterBuilderSetup builderSetup) {
    DocumentSource documentSource = name -> name.equals("name") ? Mono.just(DOCUMENT) : Mono.error(new IllegalArgumentException());
    // Original
    WebGraphQlTester.Builder<?> builder = builderSetup.initBuilder().documentSource(documentSource);
    WebGraphQlTester tester = builder.build();
    tester.documentName("name").execute();
    WebInput input = builderSetup.getWebInput();
    assertThat(input.getDocument()).isEqualTo(DOCUMENT);
    // Mutate
    tester = tester.mutate().build();
    tester.documentName("name").execute();
    input = builderSetup.getWebInput();
    assertThat(input.getDocument()).isEqualTo(DOCUMENT);
}

Example 3 with WebInput

use of org.springframework.graphql.web.WebInput in project spring-graphql by spring-projects.

the class GraphQlWebSocketHandler method handle.

@Override
public Mono<Void> handle(WebSocketSession session) {
    HandshakeInfo handshakeInfo = session.getHandshakeInfo();
    if ("graphql-ws".equalsIgnoreCase(handshakeInfo.getSubProtocol())) {
        if (logger.isDebugEnabled()) {
            logger.debug("apollographql/subscriptions-transport-ws is not supported, nor maintained. " + "Please, use https://github.com/enisdenjo/graphql-ws.");
        }
        return session.close(GraphQlStatus.INVALID_MESSAGE_STATUS);
    }
    // Session state
    AtomicReference<Map<String, Object>> connectionInitPayloadRef = new AtomicReference<>();
    Map<String, Subscription> subscriptions = new ConcurrentHashMap<>();
    Mono.delay(this.initTimeoutDuration).then(Mono.defer(() -> connectionInitPayloadRef.compareAndSet(null, Collections.emptyMap()) ? session.close(GraphQlStatus.INIT_TIMEOUT_STATUS) : Mono.empty())).subscribe();
    session.closeStatus().doOnSuccess(closeStatus -> {
        Map<String, Object> connectionInitPayload = connectionInitPayloadRef.get();
        if (connectionInitPayload == null) {
            return;
        }
        int statusCode = (closeStatus != null ? closeStatus.getCode() : 1005);
        this.webSocketInterceptor.handleConnectionClosed(session.getId(), statusCode, connectionInitPayload);
    }).subscribe();
    return session.send(session.receive().flatMap(webSocketMessage -> {
        GraphQlMessage message = this.codecDelegate.decode(webSocketMessage);
        String id = message.getId();
        Map<String, Object> payload = message.getPayload();
        switch(message.resolvedType()) {
            case SUBSCRIBE:
                if (connectionInitPayloadRef.get() == null) {
                    return GraphQlStatus.close(session, GraphQlStatus.UNAUTHORIZED_STATUS);
                }
                if (id == null) {
                    return GraphQlStatus.close(session, GraphQlStatus.INVALID_MESSAGE_STATUS);
                }
                WebInput input = new WebInput(handshakeInfo.getUri(), handshakeInfo.getHeaders(), payload, id, null);
                if (logger.isDebugEnabled()) {
                    logger.debug("Executing: " + input);
                }
                return this.graphQlHandler.handleRequest(input).flatMapMany((output) -> handleWebOutput(session, id, subscriptions, output)).doOnTerminate(() -> subscriptions.remove(id));
            case PING:
                return Flux.just(this.codecDelegate.encode(session, GraphQlMessage.pong(null)));
            case COMPLETE:
                if (id != null) {
                    Subscription subscription = subscriptions.remove(id);
                    if (subscription != null) {
                        subscription.cancel();
                    }
                    return this.webSocketInterceptor.handleCancelledSubscription(session.getId(), id).thenMany(Flux.empty());
                }
                return Flux.empty();
            case CONNECTION_INIT:
                if (!connectionInitPayloadRef.compareAndSet(null, payload)) {
                    return GraphQlStatus.close(session, GraphQlStatus.TOO_MANY_INIT_REQUESTS_STATUS);
                }
                return this.webSocketInterceptor.handleConnectionInitialization(session.getId(), payload).defaultIfEmpty(Collections.emptyMap()).map(ackPayload -> this.codecDelegate.encodeConnectionAck(session, ackPayload)).flux().onErrorResume(ex -> GraphQlStatus.close(session, GraphQlStatus.UNAUTHORIZED_STATUS));
            default:
                return GraphQlStatus.close(session, GraphQlStatus.INVALID_MESSAGE_STATUS);
        }
    }));
}

Example 4 with WebInput

use of org.springframework.graphql.web.WebInput in project spring-graphql by spring-projects.

the class GraphQlHttpHandler method handleRequest.

/**
 * Handle GraphQL requests over HTTP.
 * @param request the incoming HTTP request
 * @return the HTTP response
 * @throws ServletException may be raised when reading the request body, e.g.
 * {@link HttpMediaTypeNotSupportedException}.
 */
public ServerResponse handleRequest(ServerRequest request) throws ServletException {
    WebInput input = new WebInput(request.uri(), request.headers().asHttpHeaders(), readBody(request), this.idGenerator.generateId().toString(), LocaleContextHolder.getLocale());
    if (logger.isDebugEnabled()) {
        logger.debug("Executing: " + input);
    }
    Mono<ServerResponse> responseMono = this.graphQlHandler.handleRequest(input).map((output) -> {
        if (logger.isDebugEnabled()) {
            logger.debug("Execution complete");
        }
        ServerResponse.BodyBuilder builder = ServerResponse.ok();
        builder.headers(headers -> headers.putAll(output.getResponseHeaders()));
        return builder.body(output.toSpecification());
    });
    return ServerResponse.async(responseMono);
}

Example 5 with WebInput

use of org.springframework.graphql.web.WebInput in project spring-graphql by spring-projects.

the class GraphQlWebSocketHandler method handleTextMessage.

@Override
protected void handleTextMessage(WebSocketSession session, TextMessage webSocketMessage) throws Exception {
    GraphQlMessage message = decode(webSocketMessage);
    String id = message.getId();
    Map<String, Object> payload = message.getPayload();
    SessionState sessionState = getSessionInfo(session);
    switch(message.resolvedType()) {
        case SUBSCRIBE:
            if (sessionState.getConnectionInitPayload() == null) {
                GraphQlStatus.closeSession(session, GraphQlStatus.UNAUTHORIZED_STATUS);
                return;
            }
            if (id == null) {
                GraphQlStatus.closeSession(session, GraphQlStatus.INVALID_MESSAGE_STATUS);
                return;
            }
            URI uri = session.getUri();
            Assert.notNull(uri, "Expected handshake url");
            HttpHeaders headers = session.getHandshakeHeaders();
            WebInput input = new WebInput(uri, headers, payload, id, null);
            if (logger.isDebugEnabled()) {
                logger.debug("Executing: " + input);
            }
            this.graphQlHandler.handleRequest(input).flatMapMany((output) -> handleWebOutput(session, input.getId(), output)).publishOn(// Serial blocking send via single thread
            sessionState.getScheduler()).subscribe(new SendMessageSubscriber(id, session, sessionState));
            return;
        case PING:
            session.sendMessage(encode(GraphQlMessage.pong(null)));
            return;
        case COMPLETE:
            if (id != null) {
                Subscription subscription = sessionState.getSubscriptions().remove(id);
                if (subscription != null) {
                    subscription.cancel();
                }
                this.webSocketInterceptor.handleCancelledSubscription(session.getId(), id).block(Duration.ofSeconds(10));
            }
            return;
        case CONNECTION_INIT:
            if (!sessionState.setConnectionInitPayload(payload)) {
                GraphQlStatus.closeSession(session, GraphQlStatus.TOO_MANY_INIT_REQUESTS_STATUS);
                return;
            }
            this.webSocketInterceptor.handleConnectionInitialization(session.getId(), payload).defaultIfEmpty(Collections.emptyMap()).publishOn(// Serial blocking send via single thread
            sessionState.getScheduler()).doOnNext(ackPayload -> {
                TextMessage outputMessage = encode(GraphQlMessage.connectionAck(ackPayload));
                try {
                    session.sendMessage(outputMessage);
                } catch (IOException ex) {
                    throw new IllegalStateException(ex);
                }
            }).onErrorResume(ex -> {
                GraphQlStatus.closeSession(session, GraphQlStatus.UNAUTHORIZED_STATUS);
                return Mono.empty();
            }).block(Duration.ofSeconds(10));
            return;
        default:
            GraphQlStatus.closeSession(session, GraphQlStatus.INVALID_MESSAGE_STATUS);
    }
}