Examples with BindingContext org.springframework.web.reactive.BindingContext used on opensource projects

Example 21 with BindingContext

use of org.springframework.web.reactive.BindingContext in project spring-framework by spring-projects.

the class SessionAttributeMethodArgumentResolverTests method resolveNotRequired.

@Test
public void resolveNotRequired() throws Exception {
    MethodParameter param = initMethodParameter(2);
    Mono<Object> mono = this.resolver.resolveArgument(param, new BindingContext(), this.exchange);
    assertNull(mono.block());
    Foo foo = new Foo();
    when(this.session.getAttribute("foo")).thenReturn(Optional.of(foo));
    mono = this.resolver.resolveArgument(param, new BindingContext(), this.exchange);
    assertSame(foo, mono.block());
}

Example 22 with BindingContext

use of org.springframework.web.reactive.BindingContext in project spring-framework by spring-projects.

the class SessionAttributeMethodArgumentResolverTests method resolveWithName.

@Test
public void resolveWithName() throws Exception {
    MethodParameter param = initMethodParameter(1);
    Foo foo = new Foo();
    when(this.session.getAttribute("specialFoo")).thenReturn(Optional.of(foo));
    Mono<Object> mono = this.resolver.resolveArgument(param, new BindingContext(), this.exchange);
    assertSame(foo, mono.block());
}

Example 23 with BindingContext

use of org.springframework.web.reactive.BindingContext in project spring-framework by spring-projects.

the class AbstractMessageReaderArgumentResolver method readBody.

protected Mono<Object> readBody(MethodParameter bodyParameter, boolean isBodyRequired, BindingContext bindingContext, ServerWebExchange exchange) {
    ResolvableType bodyType = ResolvableType.forMethodParameter(bodyParameter);
    ReactiveAdapter adapter = getAdapterRegistry().getAdapter(bodyType.resolve());
    ResolvableType elementType = (adapter != null ? bodyType.getGeneric(0) : bodyType);
    ServerHttpRequest request = exchange.getRequest();
    ServerHttpResponse response = exchange.getResponse();
    MediaType mediaType = request.getHeaders().getContentType();
    if (mediaType == null) {
        mediaType = MediaType.APPLICATION_OCTET_STREAM;
    }
    for (HttpMessageReader<?> reader : getMessageReaders()) {
        if (reader.canRead(elementType, mediaType)) {
            Map<String, Object> readHints = Collections.emptyMap();
            if (adapter != null && adapter.isMultiValue()) {
                Flux<?> flux;
                if (reader instanceof ServerHttpMessageReader) {
                    ServerHttpMessageReader<?> serverReader = ((ServerHttpMessageReader<?>) reader);
                    flux = serverReader.read(bodyType, elementType, request, response, readHints);
                } else {
                    flux = reader.read(elementType, request, readHints);
                }
                flux = flux.onErrorResumeWith(ex -> Flux.error(getReadError(bodyParameter, ex)));
                if (isBodyRequired || !adapter.supportsEmpty()) {
                    flux = flux.switchIfEmpty(Flux.error(getRequiredBodyError(bodyParameter)));
                }
                Object[] hints = extractValidationHints(bodyParameter);
                if (hints != null) {
                    flux = flux.doOnNext(target -> validate(target, hints, bodyParameter, bindingContext, exchange));
                }
                return Mono.just(adapter.fromPublisher(flux));
            } else {
                Mono<?> mono;
                if (reader instanceof ServerHttpMessageReader) {
                    ServerHttpMessageReader<?> serverReader = (ServerHttpMessageReader<?>) reader;
                    mono = serverReader.readMono(bodyType, elementType, request, response, readHints);
                } else {
                    mono = reader.readMono(elementType, request, readHints);
                }
                mono = mono.otherwise(ex -> Mono.error(getReadError(bodyParameter, ex)));
                if (isBodyRequired || (adapter != null && !adapter.supportsEmpty())) {
                    mono = mono.otherwiseIfEmpty(Mono.error(getRequiredBodyError(bodyParameter)));
                }
                Object[] hints = extractValidationHints(bodyParameter);
                if (hints != null) {
                    mono = mono.doOnNext(target -> validate(target, hints, bodyParameter, bindingContext, exchange));
                }
                if (adapter != null) {
                    return Mono.just(adapter.fromPublisher(mono));
                } else {
                    return Mono.from(mono);
                }
            }
        }
    }
    return Mono.error(new UnsupportedMediaTypeStatusException(mediaType, this.supportedMediaTypes));
}

Example 24 with BindingContext

use of org.springframework.web.reactive.BindingContext in project spring-framework by spring-projects.

the class AbstractNamedValueArgumentResolver method resolveArgument.

@Override
public Mono<Object> resolveArgument(MethodParameter parameter, BindingContext bindingContext, ServerWebExchange exchange) {
    NamedValueInfo namedValueInfo = getNamedValueInfo(parameter);
    MethodParameter nestedParameter = parameter.nestedIfOptional();
    Object resolvedName = resolveStringValue(namedValueInfo.name);
    if (resolvedName == null) {
        return Mono.error(new IllegalArgumentException("Specified name must not resolve to null: [" + namedValueInfo.name + "]"));
    }
    Model model = bindingContext.getModel();
    return resolveName(resolvedName.toString(), nestedParameter, exchange).map(arg -> {
        if ("".equals(arg) && namedValueInfo.defaultValue != null) {
            arg = resolveStringValue(namedValueInfo.defaultValue);
        }
        arg = applyConversion(arg, namedValueInfo, parameter, bindingContext, exchange);
        handleResolvedValue(arg, namedValueInfo.name, parameter, model, exchange);
        return arg;
    }).otherwiseIfEmpty(getDefaultValue(namedValueInfo, parameter, bindingContext, model, exchange));
}

Example 25 with BindingContext

use of org.springframework.web.reactive.BindingContext in project spring-framework by spring-projects.

the class HttpEntityArgumentResolverTests method resolveValueWithEmptyBody.

@SuppressWarnings("unchecked")
private <T> HttpEntity<T> resolveValueWithEmptyBody(ResolvableType type) {
    ServerWebExchange exchange = post("/path").toExchange();
    MethodParameter param = this.testMethod.arg(type);
    Mono<Object> result = this.resolver.resolveArgument(param, new BindingContext(), exchange);
    HttpEntity<String> httpEntity = (HttpEntity<String>) result.block(Duration.ofSeconds(5));
    assertEquals(exchange.getRequest().getHeaders(), httpEntity.getHeaders());
    return (HttpEntity<T>) httpEntity;
}