Examples with BodySubscriber java.net.http.HttpResponse.BodySubscriber used on opensource projects

Example 1 with BodySubscriber

use of java.net.http.HttpResponse.BodySubscriber in project kubernetes-client by fabric8io.

the class JdkHttpClientImpl method sendAsync.

@Override
public <T> CompletableFuture<HttpResponse<T>> sendAsync(HttpRequest request, Class<T> type) {
    JdkHttpRequestImpl jdkRequest = (JdkHttpRequestImpl) request;
    JdkHttpRequestImpl.BuilderImpl builderImpl = jdkRequest.newBuilder();
    for (Interceptor interceptor : builder.interceptors.values()) {
        interceptor.before(builderImpl, jdkRequest);
        jdkRequest = builderImpl.build();
    }
    BodyHandler<T> bodyHandler;
    if (type == null) {
        bodyHandler = (BodyHandler<T>) BodyHandlers.discarding();
    } else if (type == InputStream.class) {
        bodyHandler = (BodyHandler<T>) BodyHandlers.ofInputStream();
    } else if (type == String.class) {
        bodyHandler = (BodyHandler<T>) BodyHandlers.ofString();
    } else {
        bodyHandler = responseInfo -> {
            BodySubscriber<InputStream> upstream = BodyHandlers.ofInputStream().apply(responseInfo);
            BodySubscriber<Reader> downstream = BodySubscribers.mapping(upstream, (InputStream is) -> new InputStreamReader(is, StandardCharsets.UTF_8));
            return (BodySubscriber<T>) downstream;
        };
    }
    CompletableFuture<java.net.http.HttpResponse<T>> cf = this.httpClient.sendAsync(builderImpl.build().request, bodyHandler);
    for (Interceptor interceptor : builder.interceptors.values()) {
        cf = cf.thenCompose(response -> {
            if (response != null && !HttpResponse.isSuccessful(response.statusCode()) && interceptor.afterFailure(builderImpl, new JdkHttpResponseImpl<>(response))) {
                return this.httpClient.sendAsync(builderImpl.build().request, bodyHandler);
            }
            return CompletableFuture.completedFuture(response);
        });
    }
    return cf.thenApply(JdkHttpResponseImpl::new);
}