Java BodyExtractors类代码示例

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
@RequestMapping(path = REQUEST_MAPPING_PATH, method = {RequestMethod.GET, RequestMethod.HEAD, RequestMethod.POST,
        RequestMethod.PUT, RequestMethod.PATCH, RequestMethod.DELETE, RequestMethod.OPTIONS})
public Mono<Void> endpointProxy(@PathVariable("instanceId") String instanceId,
                                ServerHttpRequest request,
                                ServerHttpResponse response) {
    String endpointLocalPath = getEndpointLocalPath(request.getPath().pathWithinApplication().value());
    URI uri = UriComponentsBuilder.fromPath(endpointLocalPath)
                                  .query(request.getURI().getRawQuery())
                                  .build(true)
                                  .toUri();

    return super.forward(instanceId, uri, request.getMethod(), request.getHeaders(),
            () -> BodyInserters.fromDataBuffers(request.getBody())).flatMap(clientResponse -> {
        response.setStatusCode(clientResponse.statusCode());
        response.getHeaders().addAll(filterHeaders(clientResponse.headers().asHttpHeaders()));
        return response.writeWith(clientResponse.body(BodyExtractors.toDataBuffers()));
    });
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
	// NOTICE: nothing in "pre" filter stage as CLIENT_RESPONSE_ATTR is not added
	// until the WebHandler is run
	return chain.filter(exchange).then(Mono.defer(() -> {
		ClientResponse clientResponse = exchange.getAttribute(CLIENT_RESPONSE_ATTR);
		if (clientResponse == null) {
			return Mono.empty();
		}
		log.trace("WebClientWriteResponseFilter start");
		ServerHttpResponse response = exchange.getResponse();

		return response.writeWith(clientResponse.body(BodyExtractors.toDataBuffers())).log("webClient response");
	}));
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
@ResponseBody
@RequestMapping(path = REQUEST_MAPPING_PATH, method = {RequestMethod.GET, RequestMethod.HEAD, RequestMethod.POST,
        RequestMethod.PUT, RequestMethod.PATCH, RequestMethod.DELETE, RequestMethod.OPTIONS})
public Mono<Void> endpointProxy(@PathVariable("instanceId") String instanceId,
                                HttpServletRequest servletRequest,
                                HttpServletResponse servletResponse) {
    ServerHttpRequest request = new ServletServerHttpRequest(servletRequest);
    ServerHttpResponse response = new ServletServerHttpResponse(servletResponse);

    String pathWithinApplication = servletRequest.getAttribute(HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE)
                                                 .toString();
    String endpointLocalPath = getEndpointLocalPath(pathWithinApplication);

    URI uri = UriComponentsBuilder.fromPath(endpointLocalPath)
                                  .query(request.getURI().getRawQuery())
                                  .build(true)
                                  .toUri();

    //We need to explicitly block until the headers are recieved.
    // otherwise the FrameworkServlet will add wrong Allow header for OPTIONS request
    ClientResponse clientResponse = super.forward(instanceId, uri, request.getMethod(), request.getHeaders(),
            () -> BodyInserters.fromDataBuffers(
                    DataBufferUtils.readInputStream(request::getBody, this.bufferFactory, 16384))).block();

    response.setStatusCode(clientResponse.statusCode());
    response.getHeaders().addAll(filterHeaders(clientResponse.headers().asHttpHeaders()));

    try {
        return DataBufferUtils.write(clientResponse.body(BodyExtractors.toDataBuffers()), response.getBody())
                              .doOnNext(DataBufferUtils::release)
                              .then();
    } catch (IOException ex) {
        return Mono.error(new ResponseStatusException(HttpStatus.INTERNAL_SERVER_ERROR, null, ex));
    }
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
@Override
public <T> Mono<T> bodyToMono(Class<? extends T> elementClass) {
    if (Void.class.isAssignableFrom(elementClass)) {
        return response.bodyToMono(elementClass);
    } else {
        return body(BodyExtractors.toMono(elementClass));
    }
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
@Override
public <T> Flux<T> bodyToFlux(Class<? extends T> elementClass) {
    if (Void.class.isAssignableFrom(elementClass)) {
        return response.bodyToFlux(elementClass);
    } else {
        return body(BodyExtractors.toFlux(elementClass));
    }
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
public Mono<ServerResponse> create(ServerRequest req) {
    return req.body(BodyExtractors.toMono(Post.class))
        .flatMap(post -> this.posts.save(post))
        .flatMap(p -> created(URI.create("/posts/" + p.getId())).build());
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
public Mono<ServerResponse> create(ServerRequest req) {
    return req.body(BodyExtractors.toMono(Post.class))
        .flatMap(post -> this.posts.save(post))
        .flatMap(p -> ServerResponse.created(URI.create("/posts/" + p.getId())).build());
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
private <T> Mono<T> toMono(Mono<ClientResponse> monoResponse, Class<T> monoContentType) {
    return monoResponse
            .flatMap(response -> bodyToPublisher(response,
                            BodyExtractors.toMono(monoContentType),
                            ErrorBodyExtractors.toMono(httpErrorReaders)));
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
private <T> Flux<T> toFlux(Mono<ClientResponse> monoResponse, Class<T> fluxContentType) {
    return monoResponse
            .flatMapMany(response -> bodyToPublisher(response,
                            BodyExtractors.toFlux(fluxContentType),
                            ErrorBodyExtractors.toFlux(httpErrorReaders)));
}
 

import org.springframework.web.reactive.function.BodyExtractors; //导入依赖的package包/类
private static <T> Function<ClientResponse, Mono<? extends T>> mapResponse(
		ParameterizedTypeReference<T> typeReference, String path) {

	return response -> isSuccess(response) ? response.body(BodyExtractors
			.toMono(typeReference)) : mapOtherwise(response, path);
}