Examples with BodyGenerator io.airlift.http.client.BodyGenerator used on opensource projects

Example 1 with BodyGenerator

use of io.airlift.http.client.BodyGenerator in project airlift by airlift.

the class JettyHttpClient method buildJettyRequest.

private HttpRequest buildJettyRequest(Request finalRequest) {
    HttpRequest jettyRequest = (HttpRequest) httpClient.newRequest(finalRequest.getUri());
    JettyRequestListener listener = new JettyRequestListener(finalRequest.getUri());
    jettyRequest.onRequestBegin(request -> listener.onRequestBegin());
    jettyRequest.onRequestSuccess(request -> listener.onRequestEnd());
    jettyRequest.onResponseBegin(response -> listener.onResponseBegin());
    jettyRequest.onComplete(result -> listener.onFinish());
    jettyRequest.attribute(PRESTO_STATS_KEY, listener);
    // jetty client always adds the user agent header
    // todo should there be a default?
    jettyRequest.getHeaders().remove(HttpHeader.USER_AGENT);
    jettyRequest.method(finalRequest.getMethod());
    for (Entry<String, String> entry : finalRequest.getHeaders().entries()) {
        jettyRequest.header(entry.getKey(), entry.getValue());
    }
    BodyGenerator bodyGenerator = finalRequest.getBodyGenerator();
    if (bodyGenerator != null) {
        if (bodyGenerator instanceof StaticBodyGenerator) {
            StaticBodyGenerator staticBodyGenerator = (StaticBodyGenerator) bodyGenerator;
            jettyRequest.content(new BytesContentProvider(staticBodyGenerator.getBody()));
        } else if (bodyGenerator instanceof FileBodyGenerator) {
            Path path = ((FileBodyGenerator) bodyGenerator).getPath();
            jettyRequest.content(fileContentProvider(path));
        } else {
            jettyRequest.content(new BodyGeneratorContentProvider(bodyGenerator, httpClient.getExecutor()));
        }
    }
    // timeouts
    jettyRequest.timeout(requestTimeoutMillis, MILLISECONDS);
    jettyRequest.idleTimeout(idleTimeoutMillis, MILLISECONDS);
    return jettyRequest;
}

Example 2 with BodyGenerator

use of io.airlift.http.client.BodyGenerator in project airlift by airlift.

the class JettyHttpClient method buildJettyRequest.

private HttpRequest buildJettyRequest(Request finalRequest, JettyRequestListener listener) {
    HttpRequest jettyRequest = (HttpRequest) httpClient.newRequest(finalRequest.getUri());
    jettyRequest.onRequestBegin(request -> listener.onRequestBegin());
    jettyRequest.onRequestSuccess(request -> listener.onRequestEnd());
    jettyRequest.onResponseBegin(response -> listener.onResponseBegin());
    jettyRequest.onComplete(result -> listener.onFinish());
    jettyRequest.onComplete(result -> {
        if (result.isFailed() && result.getFailure() instanceof TimeoutException) {
            clientDiagnostics.logDiagnosticsInfo(httpClient);
        }
    });
    jettyRequest.attribute(STATS_KEY, listener);
    jettyRequest.method(finalRequest.getMethod());
    for (Entry<String, String> entry : finalRequest.getHeaders().entries()) {
        jettyRequest.header(entry.getKey(), entry.getValue());
    }
    BodyGenerator bodyGenerator = finalRequest.getBodyGenerator();
    if (bodyGenerator != null) {
        if (bodyGenerator instanceof StaticBodyGenerator) {
            StaticBodyGenerator staticBodyGenerator = (StaticBodyGenerator) bodyGenerator;
            jettyRequest.content(new BytesContentProvider(staticBodyGenerator.getBody()));
        } else if (bodyGenerator instanceof FileBodyGenerator) {
            Path path = ((FileBodyGenerator) bodyGenerator).getPath();
            jettyRequest.content(fileContentProvider(path));
        } else {
            jettyRequest.content(new BodyGeneratorContentProvider(bodyGenerator, httpClient.getExecutor()));
        }
    }
    jettyRequest.followRedirects(finalRequest.isFollowRedirects());
    setPreserveAuthorization(jettyRequest, finalRequest.isPreserveAuthorizationOnRedirect());
    // timeouts
    jettyRequest.timeout(requestTimeoutMillis, MILLISECONDS);
    jettyRequest.idleTimeout(idleTimeoutMillis, MILLISECONDS);
    return jettyRequest;
}