com.king.platform.net.http.netty.HttpRequestContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of king-http-client Show documentation
A asyncronous http client built ontop of netty.
There is a newer version: 3.0.26
// 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;

import com.king.platform.net.http.HttpResponse;
import com.king.platform.net.http.ResponseBodyConsumer;
import com.king.platform.net.http.netty.eventbus.RequestEventBus;
import com.king.platform.net.http.netty.metric.TimeStampRecorder;
import com.king.platform.net.http.netty.request.NettyHttpClientRequest;
import com.king.platform.net.http.netty.response.NettyHttpClientResponse;
import io.netty.channel.Channel;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.util.AttributeKey;

public class HttpRequestContext {
	public static final AttributeKey HTTP_REQUEST_ATTRIBUTE_KEY = AttributeKey.valueOf("__HttpRequestContext");

	private final HttpMethod httpMethod;
	private final NettyHttpClientRequest nettyHttpClientRequest;

	private final int idleTimeoutMillis;
	private final int totalRequestTimeoutMillis;
	private final boolean followRedirects;
	private final boolean keepAlive;
	private final TimeStampRecorder timeStampRecorder;
	private final ResponseBodyConsumer responseBodyConsumer;
	private final RequestEventBus requestEventBus;


	private NettyHttpClientResponse nettyHttpClientResponse;
	private int redirectionCount;
	private Channel attachedChannel;
	private boolean hasCompleted;
	private long expectedContentLength;
	private long readBytes;
	private boolean isRedirecting;

	private boolean automaticallyDecompressResponse;

	private WebSocketConf webSocketConf;
	private HttpResponse httpResponse;

	public HttpRequestContext(HttpMethod httpMethod, NettyHttpClientRequest nettyHttpClientRequest, RequestEventBus requestEventBus, ResponseBodyConsumer responseBodyConsumer, int idleTimeoutMillis, int totalRequestTimeoutMillis, boolean followRedirects, boolean keepAlive, TimeStampRecorder timeStampRecorder, boolean automaticallyDecompressResponse, WebSocketConf webSocketConf) {
		this.httpMethod = httpMethod;
		this.nettyHttpClientRequest = nettyHttpClientRequest;
		this.requestEventBus = requestEventBus;
		this.responseBodyConsumer = responseBodyConsumer;
		this.idleTimeoutMillis = idleTimeoutMillis;
		this.totalRequestTimeoutMillis = totalRequestTimeoutMillis;
		this.followRedirects = followRedirects;
		this.keepAlive = keepAlive;
		this.timeStampRecorder = timeStampRecorder;
		this.automaticallyDecompressResponse = automaticallyDecompressResponse;
		this.webSocketConf = webSocketConf;
		timeStampRecorder.recordCreatedRequest();
	}

	public HttpRequestContext createRedirectRequest(ServerInfo redirectServerInfo, String redirectLocation) {
		NettyHttpClientRequest redirectRequest = nettyHttpClientRequest.createRedirectRequest(redirectServerInfo, redirectLocation);
		HttpRequestContext httpRequestContext = new HttpRequestContext(httpMethod, redirectRequest, requestEventBus, responseBodyConsumer,
			idleTimeoutMillis, totalRequestTimeoutMillis, followRedirects, keepAlive, timeStampRecorder, automaticallyDecompressResponse, webSocketConf);
		httpRequestContext.redirectionCount = this.redirectionCount + 1;

		nettyHttpClientRequest.setKeepAlive(keepAlive);
		httpRequestContext.setHttpResponse(httpResponse);
		return httpRequestContext;
	}

	public ServerInfo getServerInfo() {
		return nettyHttpClientRequest.getServerInfo();
	}


	public NettyHttpClientRequest getNettyHttpClientRequest() {
		return nettyHttpClientRequest;
	}


	public int getIdleTimeoutMillis() {
		return idleTimeoutMillis;
	}

	public int getTotalRequestTimeoutMillis() {
		return totalRequestTimeoutMillis;
	}

	public boolean isFollowRedirects() {
		return followRedirects;
	}

	public boolean isKeepAlive() {
		return keepAlive;
	}

	public ResponseBodyConsumer getResponseBodyConsumer() {
		return responseBodyConsumer;
	}

	public NettyHttpClientResponse getNettyHttpClientResponse() {
		return nettyHttpClientResponse;
	}

	public void setNettyHttpClientResponse(NettyHttpClientResponse nettyHttpClientResponse) {
		this.nettyHttpClientResponse = nettyHttpClientResponse;
	}

	public int getRedirectionCount() {
		return redirectionCount;
	}

	public RequestEventBus getRequestEventBus() {
		return requestEventBus;
	}

	public void attachedToChannel(Channel channel) {
		this.attachedChannel = channel;
	}

	public Channel getAndDetachChannel() {
		Channel channel = attachedChannel;
		attachedChannel = null;
		return channel;
	}

	public TimeStampRecorder getTimeRecorder() {
		return timeStampRecorder;
	}

	public void setHasCompleted(boolean hasCompleted) {
		this.hasCompleted = hasCompleted;
	}

	public boolean hasCompletedContent() {
		return hasCompleted;
	}

	public void setExpectedContentLength(long expectedContentLength) {
		this.expectedContentLength = expectedContentLength;
	}

	public long getExpectedContentLength() {
		return expectedContentLength;
	}

	public void addReadBytes(int readBytes) {
		this.readBytes += readBytes;
	}

	public long getReadBytes() {
		return readBytes;
	}

	public boolean isRedirecting() {
		return isRedirecting;
	}

	public void setRedirecting(boolean isRedirecting) {
		this.isRedirecting = isRedirecting;
	}

	public HttpMethod getHttpMethod() {
		return httpMethod;
	}

	public boolean automaticallyDecompressResponse() {
		return automaticallyDecompressResponse;
	}

	public WebSocketConf webSocketConf() {
		return webSocketConf;
	}

	public void setHttpResponse(HttpResponse httpResponse) {
		this.httpResponse = httpResponse;
	}

	public HttpResponse getHttpResponse() {
		return httpResponse;
	}
}