com.king.platform.net.http.netty.requestbuilder.HttpClientRequestHeaderBuilderImpl Maven / Gradle / Ivy

Go to download
// Copyright (C) king.com Ltd 2015
// https://github.com/king/king-http-client
// Author: Magnus Gustafsson
// License: Apache 2.0, https://raw.github.com/king/king-http-client/LICENSE-APACHE

package com.king.platform.net.http.netty.requestbuilder;


import com.king.platform.net.http.ConfKeys;
import com.king.platform.net.http.HttpClientRequestHeaderBuilder;
import com.king.platform.net.http.netty.ConfMap;
import com.king.platform.net.http.netty.HttpClientCaller;
import com.king.platform.net.http.util.Param;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.HttpVersion;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executor;

import static java.util.Objects.requireNonNull;

public abstract class HttpClientRequestHeaderBuilderImpl implements HttpClientRequestHeaderBuilder {
	private final Class implClass;
	protected final HttpClientCaller httpClientCaller;
	protected final HttpVersion httpVersion;
	protected final HttpMethod httpMethod;
	protected final String uri;

	protected final List queryParameters = new ArrayList<>();
	protected final List headerParameters = new ArrayList<>();

	protected final String defaultUserAgent;

	protected int idleTimeoutMillis;
	protected int totalRequestTimeoutMillis;

	protected boolean followRedirects;
	protected boolean acceptCompressedResponse;
	protected boolean keepAlive;
	protected Executor callbackExecutor;


	protected HttpClientRequestHeaderBuilderImpl(Class implClass, HttpClientCaller httpClientCaller, HttpVersion httpVersion, HttpMethod httpMethod, String uri, ConfMap confMap, Executor callbackExecutor) {
		this.implClass = implClass;
		this.httpClientCaller = httpClientCaller;
		this.httpVersion = httpVersion;
		this.httpMethod = httpMethod;
		this.uri = uri;

		idleTimeoutMillis = confMap.get(ConfKeys.IDLE_TIMEOUT_MILLIS);
		totalRequestTimeoutMillis = confMap.get(ConfKeys.TOTAL_REQUEST_TIMEOUT_MILLIS);
		followRedirects = confMap.get(ConfKeys.HTTP_FOLLOW_REDIRECTS);


		acceptCompressedResponse = confMap.get(ConfKeys.ACCEPT_COMPRESSED_RESPONSE);

		keepAlive = confMap.get(ConfKeys.KEEP_ALIVE);

		defaultUserAgent = confMap.get(ConfKeys.USER_AGENT);

		this.callbackExecutor = callbackExecutor;
	}


	@Override
	public T addHeader(CharSequence name, CharSequence value) {
		requireNonNull(name, "name");
		requireNonNull(value, "value");
		headerParameters.add(new Param(name, value));
		return implClass.cast(this);
	}

	@Override
	public T addHeaders(Map headers) {
		for (Map.Entry entry : headers.entrySet()) {
			addHeader(entry.getKey(), entry.getValue());
		}

		return implClass.cast(this);
	}

	@Override
	public T keepAlive(boolean keepAlive) {
		this.keepAlive = keepAlive;
		return implClass.cast(this);
	}

	@Override
	public T acceptCompressedResponse(boolean acceptCompressedResponse) {
		this.acceptCompressedResponse = acceptCompressedResponse;
		return implClass.cast(this);
	}

	@Override
	public T addQueryParameter(String name, String value) {
		requireNonNull(name, "name");
		requireNonNull(value, "value");
		queryParameters.add(new Param(name, value));
		return implClass.cast(this);
	}

	@Override
	public T addQueryParameters(Map parameters) {
		for (Map.Entry entry : parameters.entrySet()) {
			queryParameters.add(new Param(entry.getKey(), entry.getValue()));
		}
		return implClass.cast(this);
	}

	@Override
	public T idleTimeoutMillis(int idleTimeoutMillis) {
		this.idleTimeoutMillis = idleTimeoutMillis;
		return implClass.cast(this);
	}

	@Override
	public T totalRequestTimeoutMillis(int totalRequestTimeoutMillis) {
		this.totalRequestTimeoutMillis = totalRequestTimeoutMillis;
		return implClass.cast(this);
	}

	@Override
	public T followRedirects(boolean followRedirects) {
		this.followRedirects = followRedirects;
		return implClass.cast(this);
	}

	@Override
	public T executingOn(Executor executor) {
		this.callbackExecutor = requireNonNull(executor, "Executor can't be null");
		return implClass.cast(this);
	}
}