import static io.netty.handler.codec.http.HttpResponseStatus.BAD_REQUEST;
import static io.netty.handler.codec.http.HttpResponseStatus.FORBIDDEN;
import static io.netty.handler.codec.http.HttpResponseStatus.INTERNAL_SERVER_ERROR;
+import static io.netty.handler.codec.http.HttpResponseStatus.OK;
import static io.netty.handler.codec.http.HttpUtil.isKeepAlive;
import static io.netty.handler.codec.http.HttpUtil.setContentLength;
import static io.netty.handler.codec.http.HttpVersion.HTTP_1_1;
-import io.netty.buffer.ByteBuf;
-import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerContext;
* {@link FullHttpRequest} and {@link WebSocketFrame} messages.
*/
public class WebSocketServerHandler extends SimpleChannelInboundHandler<Object> {
-
private static final Logger LOG = LoggerFactory.getLogger(WebSocketServerHandler.class);
private WebSocketServerHandshaker handshaker;
/**
* Checks if HTTP request method is GET and if is possible to decode HTTP result of request.
*
- * @param ctx
- * ChannelHandlerContext
- * @param req
- * FullHttpRequest
+ * @param ctx ChannelHandlerContext
+ * @param req FullHttpRequest
*/
private void handleHttpRequest(final ChannelHandlerContext ctx, final FullHttpRequest req) {
// Handle a bad request.
} else {
this.handshaker.handshake(ctx.channel(), req);
}
-
}
/**
* Checks response status, send response and close connection if necessary.
*
- * @param ctx
- * ChannelHandlerContext
- * @param req
- * HttpRequest
- * @param res
- * FullHttpResponse
+ * @param ctx ChannelHandlerContext
+ * @param req HttpRequest
+ * @param res FullHttpResponse
*/
private static void sendHttpResponse(final ChannelHandlerContext ctx, final HttpRequest req,
final FullHttpResponse res) {
// Generate an error page if response getStatus code is not OK (200).
- if (res.status().code() != 200) {
- final ByteBuf buf = Unpooled.copiedBuffer(res.status().toString(), CharsetUtil.UTF_8);
- res.content().writeBytes(buf);
- buf.release();
+ final boolean notOkay = !OK.equals(res.status());
+ if (notOkay) {
+ res.content().writeCharSequence(res.status().toString(), CharsetUtil.UTF_8);
setContentLength(res, res.content().readableBytes());
}
// Send the response and close the connection if necessary.
final ChannelFuture f = ctx.channel().writeAndFlush(res);
- if (!isKeepAlive(req) || res.status().code() != 200) {
+ if (notOkay || !isKeepAlive(req)) {
f.addListener(ChannelFutureListener.CLOSE);
}
}
/**
* Handles web socket frame.
*
- * @param ctx
- * {@link ChannelHandlerContext}
- * @param frame
- * {@link WebSocketFrame}
+ * @param ctx {@link ChannelHandlerContext}
+ * @param frame {@link WebSocketFrame}
*/
private void handleWebSocketFrame(final ChannelHandlerContext ctx, final WebSocketFrame frame) {
if (frame instanceof CloseWebSocketFrame) {
/**
* Get web socket location from HTTP request.
*
- * @param req
- * HTTP request from which the location will be returned
+ * @param req HTTP request from which the location will be returned
* @return String representation of web socket location.
*/
private static String getWebSocketLocation(final HttpRequest req) {