Examples with Mono reactor.core.publisher.Mono used on opensource projects

Example 91 with Mono

use of reactor.core.publisher.Mono in project spring-framework by spring-projects.

the class RequestMappingHandlerAdapter method handle.

@Override
public Mono<HandlerResult> handle(ServerWebExchange exchange, Object handler) {
    HandlerMethod handlerMethod = (HandlerMethod) handler;
    Assert.state(this.methodResolver != null && this.modelInitializer != null, "Not initialized");
    InitBinderBindingContext bindingContext = new InitBinderBindingContext(getWebBindingInitializer(), this.methodResolver.getInitBinderMethods(handlerMethod));
    InvocableHandlerMethod invocableMethod = this.methodResolver.getRequestMappingMethod(handlerMethod);
    Function<Throwable, Mono<HandlerResult>> exceptionHandler = ex -> handleException(ex, handlerMethod, bindingContext, exchange);
    return this.modelInitializer.initModel(handlerMethod, bindingContext, exchange).then(Mono.defer(() -> invocableMethod.invoke(exchange, bindingContext))).doOnNext(result -> result.setExceptionHandler(exceptionHandler)).doOnNext(result -> bindingContext.saveModel()).onErrorResume(exceptionHandler);
}

Example 92 with Mono

use of reactor.core.publisher.Mono in project spring-framework by spring-projects.

the class RequestPartMethodArgumentResolver method decode.

@SuppressWarnings("unchecked")
private <T> Mono<T> decode(Part part, MethodParameter elementType, BindingContext bindingContext, ServerWebExchange exchange, boolean isRequired) {
    ServerHttpRequest partRequest = new PartServerHttpRequest(exchange.getRequest(), part);
    ServerWebExchange partExchange = exchange.mutate().request(partRequest).build();
    if (logger.isDebugEnabled()) {
        logger.debug(exchange.getLogPrefix() + "Decoding part '" + part.name() + "'");
    }
    return (Mono<T>) readBody(elementType, isRequired, bindingContext, partExchange);
}

Example 93 with Mono

use of reactor.core.publisher.Mono in project spring-framework by spring-projects.

the class RequestPartMethodArgumentResolver method resolveArgument.

@Override
public Mono<Object> resolveArgument(MethodParameter parameter, BindingContext bindingContext, ServerWebExchange exchange) {
    RequestPart requestPart = parameter.getParameterAnnotation(RequestPart.class);
    boolean isRequired = (requestPart == null || requestPart.required());
    Class<?> paramType = parameter.getParameterType();
    Flux<Part> partValues = getPartValues(parameter, requestPart, isRequired, exchange);
    if (Part.class.isAssignableFrom(paramType)) {
        return partValues.next().cast(Object.class);
    }
    if (Collection.class.isAssignableFrom(paramType) || List.class.isAssignableFrom(paramType)) {
        MethodParameter elementType = parameter.nested();
        if (Part.class.isAssignableFrom(elementType.getNestedParameterType())) {
            return partValues.collectList().cast(Object.class);
        } else {
            return partValues.next().flatMap(part -> decode(part, parameter, bindingContext, exchange, isRequired)).defaultIfEmpty(Collections.emptyList());
        }
    }
    ReactiveAdapter adapter = getAdapterRegistry().getAdapter(paramType);
    if (adapter == null) {
        return partValues.next().flatMap(part -> decode(part, parameter, bindingContext, exchange, isRequired));
    }
    MethodParameter elementType = parameter.nested();
    if (Part.class.isAssignableFrom(elementType.getNestedParameterType())) {
        return Mono.just(adapter.fromPublisher(partValues));
    }
    Flux<?> flux = partValues.flatMap(part -> decode(part, elementType, bindingContext, exchange, isRequired));
    return Mono.just(adapter.fromPublisher(flux));
}

Example 94 with Mono

use of reactor.core.publisher.Mono in project spring-framework by spring-projects.

the class MultipartHttpMessageWriter method writeMultipart.

private Mono<Void> writeMultipart(MultiValueMap<String, ?> map, ReactiveHttpOutputMessage outputMessage, @Nullable MediaType mediaType, Map<String, Object> hints) {
    byte[] boundary = generateMultipartBoundary();
    mediaType = getMultipartMediaType(mediaType, boundary);
    outputMessage.getHeaders().setContentType(mediaType);
    LogFormatUtils.traceDebug(logger, traceOn -> Hints.getLogPrefix(hints) + "Encoding " + (isEnableLoggingRequestDetails() ? LogFormatUtils.formatValue(map, !traceOn) : "parts " + map.keySet() + " (content masked)"));
    DataBufferFactory bufferFactory = outputMessage.bufferFactory();
    Flux<DataBuffer> body = Flux.fromIterable(map.entrySet()).concatMap(entry -> encodePartValues(boundary, entry.getKey(), entry.getValue(), bufferFactory)).concatWith(generateLastLine(boundary, bufferFactory)).doOnDiscard(PooledDataBuffer.class, DataBufferUtils::release);
    if (logger.isDebugEnabled()) {
        body = body.doOnNext(buffer -> Hints.touchDataBuffer(buffer, hints, logger));
    }
    return outputMessage.writeWith(body);
}

Example 95 with Mono

use of reactor.core.publisher.Mono in project spring-framework by spring-projects.

the class ResourceHttpMessageWriter method writeResource.

private Mono<Void> writeResource(Resource resource, ResolvableType type, @Nullable MediaType mediaType, ReactiveHttpOutputMessage message, Map<String, Object> hints) {
    HttpHeaders headers = message.getHeaders();
    MediaType resourceMediaType = getResourceMediaType(mediaType, resource, hints);
    headers.setContentType(resourceMediaType);
    if (headers.getContentLength() < 0) {
        long length = lengthOf(resource);
        if (length != -1) {
            headers.setContentLength(length);
        }
    }
    return zeroCopy(resource, null, message, hints).orElseGet(() -> {
        Mono<Resource> input = Mono.just(resource);
        DataBufferFactory factory = message.bufferFactory();
        Flux<DataBuffer> body = this.encoder.encode(input, factory, type, resourceMediaType, hints);
        if (logger.isDebugEnabled()) {
            body = body.doOnNext(buffer -> Hints.touchDataBuffer(buffer, hints, logger));
        }
        return message.writeWith(body);
    });
}