org.analogweb.netty.FullHttpResponseContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of analogweb-netty Show documentation
There is a newer version: 0.12.1
package org.analogweb.netty;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.*;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Locale;
import java.util.TimeZone;

import io.netty.handler.codec.http2.HttpConversionUtil;
import org.analogweb.*;
import org.analogweb.core.*;
import org.analogweb.util.StringUtils;

/**
 * @author y2k2mt
 */
public class FullHttpResponseContext extends AbstractResponseContext {

	private final FullHttpRequest exc;
	private int status = HttpURLConnection.HTTP_OK;
	private final Headers headers;
	private final ChannelHandlerContext context;

	public FullHttpResponseContext(FullHttpRequest request,
			ChannelHandlerContext context) {
		this.exc = request;
		this.headers = new MapHeaders();
		this.context = context;
	}

	@Override
	public void commit(RequestContext context, Response response) {
		try {
			final ResponseEntity entity = response.getEntity();
			final ByteBuf buffer = Unpooled.buffer();
			final WritableBuffer writeBuffer = ByteBufWritableBuffer
					.writeBuffer(buffer);
			if (entity instanceof DefaultResponseEntity) {
				writeBuffer.writeBytes(((DefaultResponseEntity) entity)
						.entity());
			} else if (entity instanceof ReadableBufferResponseEntity) {
				((ReadableBufferResponseEntity) entity).entity()
						.to(writeBuffer);
			}
			final FullHttpResponse fullResponse = new DefaultFullHttpResponse(
					HttpVersion.HTTP_1_1,
					HttpResponseStatus.valueOf(getStatus()), buffer);
			HttpUtil.setContentLength(fullResponse, fullResponse.content()
					.readableBytes());
			final FullHttpRequest request = getFullHttpRequest();
			String streamId = request.headers().get(
					HttpConversionUtil.ExtensionHeaderNames.STREAM_ID.text());
			if (StringUtils.isNotEmpty(streamId)) {
				fullResponse
						.headers()
						.set(HttpConversionUtil.ExtensionHeaderNames.STREAM_ID
								.text(),
								streamId);
			}
			final boolean close = request.headers().contains("Connection",
					HttpHeaderValues.CLOSE, true)
					|| request.protocolVersion().equals(HttpVersion.HTTP_1_0)
					&& !request.headers().contains("Connection",
							HttpHeaderValues.KEEP_ALIVE, true);
			if (!close) {
				fullResponse.headers().set("Content-Length",
						buffer.readableBytes());
			}
			final HttpHeaders headers = serverHeader(dateHeader(fullResponse
					.headers()));
			final Headers analogHeaders = getResponseHeaders();
			for (final String headerName : analogHeaders.getNames()) {
				headers.set(headerName, analogHeaders.getValues(headerName));
			}

			final Channel channel = getChannelHandlerContext().channel();
			final ChannelFuture future = channel.writeAndFlush(fullResponse);
			if (close) {
				future.addListener(ChannelFutureListener.CLOSE);
			}
		} catch (final IOException e) {
			throw new ApplicationRuntimeException(e) {

				private static final long serialVersionUID = 1L;
			};
		}
	}

	private HttpHeaders serverHeader(HttpHeaders headers) {
		return headers.add("Server", new StringBuilder().append("analogweb")
				.append(Properties.instance().getVersion()).toString());
	}

	private HttpHeaders dateHeader(HttpHeaders headers) {
		SimpleDateFormat dateFormat = new SimpleDateFormat(
				"EEE, dd MMM yyyy HH:mm:ss z", Locale.US);
		dateFormat.setTimeZone(TimeZone.getTimeZone("GMT"));
		return headers.add("Date",
				dateFormat.format(Calendar.getInstance().getTime()));
	}

	@Override
	public Headers getResponseHeaders() {
		return this.headers;
	}

	@Override
	public void setStatus(int status) {
		this.status = status;
	}

	protected int getStatus() {
		return this.status;
	}

	protected FullHttpRequest getFullHttpRequest() {
		return this.exc;
	}

	protected ChannelHandlerContext getChannelHandlerContext() {
		return this.context;
	}
}