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

Go to download
// Copyright (C) king.com Ltd 2016
// 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.*;
import com.king.platform.net.http.netty.ConfMap;
import com.king.platform.net.http.netty.HttpClientCaller;
import com.king.platform.net.http.netty.ServerInfoValidator;
import com.king.platform.net.http.netty.WebSocketConf;
import com.king.platform.net.http.netty.sse.VoidResponseConsumer;
import com.king.platform.net.http.netty.websocket.WebSocketClientImpl;
import io.netty.handler.codec.http.HttpHeaderNames;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.HttpVersion;

import java.time.Duration;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executor;


public class HttpClientWebSocketRequestBuilderImpl extends HttpClientRequestHeaderBuilderImpl implements HttpClientWebSocketRequestBuilder {
	private final Executor defaultCallbackExecutor;
	private boolean autoPong;
	private Duration pingEveryDuration;
	private boolean autoCloseFrame;
	private int maxFrameSize;
	private boolean aggregateFrames;
	private int maxAggregateBufferSize;
	private int maxOutgoingFrameSize;
	private int maxIncomingFrameSize;
	private boolean splitLargeFrames;

	public HttpClientWebSocketRequestBuilderImpl(HttpClientCaller httpClientCaller, String uri, ConfMap confMap,
												 Executor callbackExecutor) {
		super(HttpClientWebSocketRequestBuilder.class, httpClientCaller, HttpVersion.HTTP_1_1, HttpMethod.GET, uri, confMap, callbackExecutor);
		this.defaultCallbackExecutor = callbackExecutor;
		autoPong(confMap.get(ConfKeys.WEB_SOCKET_AUTO_PONG));
		autoCloseFrame(confMap.get(ConfKeys.WEB_SOCKET_AUTO_CLOSE_FRAME));

		aggregateFrames(confMap.get(ConfKeys.WEB_SOCKET_AGGREGATE_FRAMES));
		maxAggregateBufferSize(confMap.get(ConfKeys.WEB_SOCKET_MAX_AGGREGATE_BUFFER_SIZE));

		maxOutgoingFrameSize(confMap.get(ConfKeys.WEB_SOCKET_MAX_OUTGOING_FRAME_SIZE));
		maxIncomingFrameSize(confMap.get(ConfKeys.WEB_SOCKET_MAX_INCOMING_FRAME_SIZE));

		maxFrameSize(confMap.get(ConfKeys.WEB_SOCKET_MAX_FRAME_SIZE));
		splitLargeFrames(confMap.get(ConfKeys.WEB_SOCKET_SPLIT_FRAMES));

	}

	@Override
	public HttpClientWebSocketRequestBuilder subProtocols(String subProtocols) {
		addHeader(HttpHeaderNames.SEC_WEBSOCKET_PROTOCOL, subProtocols);
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder pingEvery(Duration duration) {
		this.pingEveryDuration = duration;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder autoPong(boolean autoPong) {
		this.autoPong = autoPong;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder autoCloseFrame(boolean autoCloseFrame) {
		this.autoCloseFrame = autoCloseFrame;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder maxFrameSize(int maxFrameSize) {
		this.maxFrameSize = maxFrameSize;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder aggregateFrames(boolean aggregateFrames) {
		this.aggregateFrames = aggregateFrames;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder maxAggregateBufferSize(int maxAggregateBufferSize) {
		this.maxAggregateBufferSize = maxAggregateBufferSize;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder maxOutgoingFrameSize(int maxOutgoingFrameSize) {
		this.maxOutgoingFrameSize = maxOutgoingFrameSize;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder maxIncomingFrameSize(int maxIncomingFrameSize) {
		this.maxIncomingFrameSize = maxIncomingFrameSize;
		return this;
	}

	@Override
	public HttpClientWebSocketRequestBuilder splitLargeFrames(boolean splitLargeFrames) {
		this.splitLargeFrames = splitLargeFrames;
		return this;
	}

	@Override
	public BuiltWebSocketRequest build() {

		totalRequestTimeoutMillis(0); //disable total timeouts

		WebSocketConf webSocketConf = new WebSocketConf(maxFrameSize, aggregateFrames, maxAggregateBufferSize, splitLargeFrames, maxIncomingFrameSize, maxOutgoingFrameSize);

		final BuiltNettyClientRequest builtNettyClientRequest = new BuiltNettyClientRequest<>(httpClientCaller, httpVersion, httpMethod, uri, defaultUserAgent,
			idleTimeoutMillis, totalRequestTimeoutMillis, followRedirects, acceptCompressedResponse, keepAlive, automaticallyDecompressResponse, null, null, null, queryParameters,
			headerParameters, callbackExecutor, VoidResponseConsumer::new, webSocketConf, ServerInfoValidator.WEB_SOCKET);


		return new BuiltWebSocketRequest() {

			@Override
			public CompletableFuture execute(WebSocketListener webSocketListener) {
				WebSocketClientImpl webSocketClient = create();
				webSocketClient.addListener(webSocketListener);
				return webSocketClient.connect();

			}

			@Override
			public CompletableFuture execute(WebSocketFrameListener webSocketFrameListener) {
				WebSocketClientImpl webSocketClient = create();
				webSocketClient.addListener(webSocketFrameListener);
				return webSocketClient.connect();
			}

			@Override
			public CompletableFuture execute(WebSocketMessageListener webSocketMessageListener) {
				WebSocketClientImpl webSocketClient = create();
				webSocketClient.addListener(webSocketMessageListener);
				return webSocketClient.connect();
			}

			@Override
			public WebSocketClient build() {
				return create();
			}

			private WebSocketClientImpl create() {
				Executor listenerExecutor = null;
				if (defaultCallbackExecutor != HttpClientWebSocketRequestBuilderImpl.this.callbackExecutor) {
					listenerExecutor = HttpClientWebSocketRequestBuilderImpl.this.callbackExecutor;
				} else {
					listenerExecutor = Runnable::run; //if no executor has been supplied (ie, still on default executor), run on calling thread
				}

				return new WebSocketClientImpl(builtNettyClientRequest, listenerExecutor, listenerExecutor, autoCloseFrame, autoPong, pingEveryDuration, webSocketConf);
			}


		};

	}




}