Examples with NettyDirectBody org.asynchttpclient.netty.request.body.NettyDirectBody used on opensource projects

Example 1 with NettyDirectBody

use of org.asynchttpclient.netty.request.body.NettyDirectBody in project async-http-client by AsyncHttpClient.

the class NettyRequestFactory method newNettyRequest.

public NettyRequest newNettyRequest(Request request, boolean forceConnect, ProxyServer proxyServer, Realm realm, Realm proxyRealm) {
    Uri uri = request.getUri();
    HttpMethod method = forceConnect ? HttpMethod.CONNECT : HttpMethod.valueOf(request.getMethod());
    boolean connect = method == HttpMethod.CONNECT;
    HttpVersion httpVersion = HttpVersion.HTTP_1_1;
    String requestUri = requestUri(uri, proxyServer, connect);
    NettyBody body = body(request, connect);
    HttpRequest httpRequest;
    NettyRequest nettyRequest;
    if (body instanceof NettyDirectBody) {
        ByteBuf buf = NettyDirectBody.class.cast(body).byteBuf();
        httpRequest = new DefaultFullHttpRequest(httpVersion, method, requestUri, buf);
        // body is passed as null as it's written directly with the request
        nettyRequest = new NettyRequest(httpRequest, null);
    } else if (body == null) {
        httpRequest = new DefaultFullHttpRequest(httpVersion, method, requestUri, Unpooled.EMPTY_BUFFER);
        nettyRequest = new NettyRequest(httpRequest, null);
    } else {
        httpRequest = new DefaultHttpRequest(httpVersion, method, requestUri);
        nettyRequest = new NettyRequest(httpRequest, body);
    }
    HttpHeaders headers = httpRequest.headers();
    if (connect) {
        // assign proxy-auth as configured on request
        headers.set(PROXY_AUTHORIZATION, request.getHeaders().getAll(PROXY_AUTHORIZATION));
    } else {
        // assign headers as configured on request
        headers.set(request.getHeaders());
        if (isNonEmpty(request.getCookies()))
            headers.set(COOKIE, ClientCookieEncoder.STRICT.encode(request.getCookies()));
        String userDefinedAcceptEncoding = headers.get(ACCEPT_ENCODING);
        if (userDefinedAcceptEncoding != null) {
            // we don't support Brotly ATM
            if (userDefinedAcceptEncoding.endsWith(BROTLY_ACCEPT_ENCODING_SUFFIX)) {
                headers.set(ACCEPT_ENCODING, userDefinedAcceptEncoding.subSequence(0, userDefinedAcceptEncoding.length() - BROTLY_ACCEPT_ENCODING_SUFFIX.length()));
            }
        } else if (config.isCompressionEnforced()) {
            headers.set(ACCEPT_ENCODING, GZIP_DEFLATE);
        }
    }
    if (body != null) {
        if (body.getContentLength() < 0)
            headers.set(TRANSFER_ENCODING, HttpHeaderValues.CHUNKED);
        else
            headers.set(CONTENT_LENGTH, body.getContentLength());
        if (body.getContentType() != null)
            headers.set(CONTENT_TYPE, body.getContentType());
    }
    // connection header and friends
    if (!connect && uri.isWebSocket()) {
        //
        headers.set(UPGRADE, HttpHeaderValues.WEBSOCKET).set(CONNECTION, //
        HttpHeaderValues.UPGRADE).set(ORIGIN, //
        "http://" + uri.getHost() + ":" + uri.getExplicitPort()).set(SEC_WEBSOCKET_KEY, //
        getKey()).set(SEC_WEBSOCKET_VERSION, "13");
    } else if (!headers.contains(CONNECTION)) {
        CharSequence connectionHeaderValue = connectionHeader(config.isKeepAlive(), httpVersion);
        if (connectionHeaderValue != null)
            headers.set(CONNECTION, connectionHeaderValue);
    }
    if (!headers.contains(HOST))
        headers.set(HOST, hostHeader(request, uri));
    // don't override authorization but append
    addAuthorizationHeader(headers, perRequestAuthorizationHeader(realm));
    // only set proxy auth on request over plain HTTP, or when performing CONNECT
    if (!uri.isSecured() || connect) {
        setProxyAuthorizationHeader(headers, perRequestProxyAuthorizationHeader(proxyRealm));
    }
    // Add default accept headers
    if (!headers.contains(ACCEPT))
        headers.set(ACCEPT, "*/*");
    // Add default user agent
    if (!headers.contains(USER_AGENT) && config.getUserAgent() != null)
        headers.set(USER_AGENT, config.getUserAgent());
    return nettyRequest;
}