org.davidmoten.oa3.codegen.http.service.internal.ApacheHttpClientHttpConnection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of openapi-codegen-http Show documentation
The newest version!
package org.davidmoten.oa3.codegen.http.service.internal;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;

import org.apache.hc.client5.http.classic.methods.HttpUriRequestBase;
import org.apache.hc.client5.http.config.RequestConfig;
import org.apache.hc.client5.http.config.RequestConfig.Builder;
import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;
import org.apache.hc.client5.http.impl.classic.HttpClients;
import org.apache.hc.core5.http.ContentType;
import org.apache.hc.core5.http.Header;
import org.apache.hc.core5.http.io.HttpClientResponseHandler;
import org.apache.hc.core5.http.io.entity.EntityTemplate;
import org.davidmoten.oa3.codegen.http.service.HttpConnection;
import org.davidmoten.oa3.codegen.http.service.Response;
import org.davidmoten.oa3.codegen.util.Util;

public class ApacheHttpClientHttpConnection implements HttpConnection {

    private final HttpUriRequestBase request;
	private Optional connectTimeoutMs = Optional.empty();
	private Optional readTimeoutMs = Optional.empty();

    public ApacheHttpClientHttpConnection(HttpUriRequestBase request) {
        this.request = request;
    }

    @Override
    public void header(String key, String value) {
        request.addHeader(key, value);
    }
    
	@Override
	public void setConnectTimeoutMs(long connectTimeoutMs) {
		this.connectTimeoutMs = Optional.of(connectTimeoutMs);
	}

	@Override
	public void setReadTimeoutMs(long readTimeoutMs) {
		this.readTimeoutMs = Optional.of(readTimeoutMs);
	}

    @Override
    public void output(Consumer consumer, String contentType, Optional contentEncoding,
            boolean chunked) {
        NoCopyByteArrayOutputStream bytes = new NoCopyByteArrayOutputStream(256);
        consumer.accept(bytes);
        
        request.setEntity(new EntityTemplate(bytes.size(), ContentType.parse(contentType),
                contentEncoding.orElse(null), out -> out.write(bytes.buffer(), 0, bytes.size())));
    }

    @Override
    public Response response() throws IOException {
    	Builder b = RequestConfig.custom();
    	connectTimeoutMs.ifPresent(timeoutMs -> b.setConnectionRequestTimeout(0, TimeUnit.MILLISECONDS));
    	readTimeoutMs.ifPresent(timeoutMs -> b.setResponseTimeout(timeoutMs, TimeUnit.MILLISECONDS));
    	RequestConfig config = b.build();
        try (CloseableHttpClient client = HttpClients.custom().setDefaultRequestConfig(config).build()) {
            HttpClientResponseHandler handler = r -> {
                final InputStream in;
                if (r.getEntity() != null) {
                    in = new ByteArrayInputStream(Util.read(r.getEntity().getContent()));
                } else {
                    in = null;
                }
                Map> map = new HashMap<>();
                for (Header header : r.getHeaders()) {
                    List list = map.get(header.getName());
                    if (list == null) {
                        list = new ArrayList<>();
                        map.put(header.getName(), list);
                    }
                    list.add(header.getValue());
                }
                return new ApacheHttpClientResponse(r.getCode(), in, map);
            };
            return client.execute(request, handler);
        }
    }

    @Override
    public void close() throws IOException {
        // TODO what here?
    }

}