com.king.platform.net.http.netty.websocket.WebSocketResponseHandler 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
package com.king.platform.net.http.netty.websocket;

import com.king.platform.net.http.netty.HttpClientHandler;
import com.king.platform.net.http.netty.HttpRequestContext;
import com.king.platform.net.http.netty.ResponseHandler;
import com.king.platform.net.http.netty.eventbus.Event;
import com.king.platform.net.http.netty.eventbus.RequestEventBus;
import com.king.platform.net.http.netty.response.HttpRedirector;
import com.king.platform.net.http.netty.response.NettyHttpClientResponse;
import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.*;
import io.netty.handler.codec.http.websocketx.WebSocketFrame;
import org.slf4j.Logger;

import static io.netty.handler.codec.http.HttpHeaderNames.SEC_WEBSOCKET_ACCEPT;
import static io.netty.handler.codec.http.HttpHeaderNames.SEC_WEBSOCKET_KEY;
import static org.slf4j.LoggerFactory.getLogger;

public class WebSocketResponseHandler implements ResponseHandler {
	private final Logger logger = getLogger(getClass());

	@Override
	public void handleResponse(ChannelHandlerContext ctx, Object msg) throws Exception {

		HttpRequestContext httpRequestContext = ctx.channel().attr(HttpRequestContext.HTTP_REQUEST_ATTRIBUTE_KEY).get();

		if (httpRequestContext == null) {
			logger.trace("httpRequestContext is null, msg was {}", msg);
			return;
		}

		if (ctx.channel().attr(HttpClientHandler.HTTP_CLIENT_HANDLER_TRIGGERED_ERROR).get()) {
			logger.trace("This channel has already triggered error, ignoring this invocation");
			return;
		}

		NettyHttpClientResponse nettyHttpClientResponse = httpRequestContext.getNettyHttpClientResponse();

		RequestEventBus requestEventBus = nettyHttpClientResponse.getRequestEventBus();

		requestEventBus.triggerEvent(Event.TOUCH);

		if (msg instanceof LastHttpContent) {

			return;
		}

		if (msg instanceof HttpResponse) {

			HttpResponse response = (HttpResponse) msg;

			if (HttpRedirector.isRedirectResponse(response.status())) {
				HttpRedirector httpRedirector = new HttpRedirector();
				httpRedirector.redirectRequest(httpRequestContext, response.headers());
			} else {
				if (response.status().equals(HttpResponseStatus.SWITCHING_PROTOCOLS)) {
					upgrade(response, requestEventBus, ctx.channel(), httpRequestContext);
				} else {
					abort(requestEventBus, httpRequestContext);
				}
			}


		} else if (msg instanceof WebSocketFrame) {

			WebSocketFrame frame = (WebSocketFrame) msg;
			handleFrame(frame, requestEventBus, httpRequestContext, ctx);
		} else {
			logger.error("Invalid message {}", msg);
		}
	}

	private void handleFrame(WebSocketFrame frame, RequestEventBus requestEventBus, HttpRequestContext httpRequestContext, ChannelHandlerContext ctx) {
		requestEventBus.triggerEvent(Event.onWsFrame, frame);

		//if (frame instanceof CloseWebSocketFrame) {
		//	CloseWebSocketFrame closeWebSocketFrame = (CloseWebSocketFrame) frame;
		//	ctx.channel().writeAndFlush(new CloseWebSocketFrame(closeWebSocketFrame.statusCode(), closeWebSocketFrame.reasonText()));
		//	requestEventBus.triggerEvent(Event.COMPLETED, httpRequestContext);
		//}

	}

	private void abort(RequestEventBus requestEventBus, HttpRequestContext httpRequestContext) {
		requestEventBus.triggerEvent(Event.ERROR, httpRequestContext, new WebSocketException("Invalid upgrade - aborting!"));
	}

	private void upgrade(HttpResponse response, RequestEventBus requestEventBus, Channel channel, HttpRequestContext httpRequestContext) {

		boolean validStatus = response.status().equals(HttpResponseStatus.SWITCHING_PROTOCOLS);
		boolean validUpgrade = response.headers().get(HttpHeaderNames.UPGRADE) != null;
		String connection = response.headers().get(HttpHeaderNames.CONNECTION);

		boolean validConnection = HttpHeaderValues.UPGRADE.contentEqualsIgnoreCase(connection);

		if (!validStatus || !validUpgrade || !validConnection) {
			requestEventBus.triggerEvent(Event.ERROR, httpRequestContext, new WebSocketException("Invalid handshake!"));
			return;
		}


		String receivedKey = response.headers().get(SEC_WEBSOCKET_ACCEPT);
		String sentKey = httpRequestContext.getNettyHttpClientRequest().getNettyHeaders().get(SEC_WEBSOCKET_KEY);
		String expectedKey = WebSocketUtil.getAcceptKey(sentKey);
		if (receivedKey == null || !expectedKey.equals(receivedKey)) {
			requestEventBus.triggerEvent(Event.ERROR, httpRequestContext, new WebSocketException("Invalid Challenge! Received key: " + receivedKey + ", Expected key: " + expectedKey));
			return;
		}


		requestEventBus.triggerEvent(Event.WS_UPGRADE_PIPELINE, channel.pipeline());

		requestEventBus.triggerEvent(Event.onWsOpen, channel, response.headers());

	}

	@Override
	public void handleChannelInactive(ChannelHandlerContext ctx) throws Exception {
		HttpRequestContext httpRequestContext = ctx.channel().attr(HttpRequestContext.HTTP_REQUEST_ATTRIBUTE_KEY).get();

		if (httpRequestContext == null) {
			logger.trace("httpRequestContext is null");
			return;
		}

		if (ctx.channel().attr(HttpClientHandler.HTTP_CLIENT_HANDLER_TRIGGERED_ERROR).get()) {
			logger.trace("This channel has already triggered error, ignoring this invocation");
			return;
		}

		NettyHttpClientResponse nettyHttpClientResponse = httpRequestContext.getNettyHttpClientResponse();

		RequestEventBus requestEventBus = nettyHttpClientResponse.getRequestEventBus();

		requestEventBus.triggerEvent(Event.COMPLETED, httpRequestContext);
	}
}