org.restexpress.response.DefaultHttpResponseWriter Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010-2014, Strategic Gains, Inc.  All rights reserved.
 */
package org.restexpress.response;

import static org.jboss.netty.handler.codec.http.HttpHeaders.Names.CONNECTION;
import static org.jboss.netty.handler.codec.http.HttpHeaders.Names.CONTENT_LENGTH;

import org.jboss.netty.buffer.ChannelBuffer;
import org.jboss.netty.buffer.ChannelBuffers;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelFutureListener;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.handler.codec.http.DefaultHttpResponse;
import org.jboss.netty.handler.codec.http.HttpResponse;
import org.jboss.netty.handler.codec.http.HttpResponseStatus;
import org.restexpress.ContentType;
import org.restexpress.Parameters;
import org.restexpress.Request;
import org.restexpress.Response;
import org.restexpress.util.HttpSpecification;

/**
 * @author toddf
 * @since Aug 26, 2010
 */
public class DefaultHttpResponseWriter
implements HttpResponseWriter
{
	@Override
	public void write(ChannelHandlerContext ctx, Request request, Response response)
	{
		HttpResponse httpResponse = createHttpResponse(request, response);
		addHeaders(response, httpResponse);		

		if (response.hasBody() && HttpSpecification.isContentAllowed(response))
		{
			// If the response body already contains a ChannelBuffer, use it.
			if (ChannelBuffer.class.isAssignableFrom(response.getBody().getClass()))
			{
				httpResponse.setContent((ChannelBuffer) response.getBody());
			}
			else // response body is assumed to be a string (e.g. raw JSON or XML).
			{
				httpResponse.setContent(ChannelBuffers.copiedBuffer(response.getBody().toString(), ContentType.CHARSET));
			}
		}

		Channel channel = ctx.getChannel();

		if (request.isKeepAlive())
	  	{
	  		// Add 'Content-Length' header only for a keep-alive connection.
			if (HttpSpecification.isContentLengthAllowed(response))
	  		{
				httpResponse.headers().set(CONTENT_LENGTH, String.valueOf(httpResponse.getContent().readableBytes()));
	  		}

			// Support "Connection: Keep-Alive" for HTTP 1.0 requests.
			if (request.isHttpVersion1_0()) 
			{
				httpResponse.headers().add(CONNECTION, "Keep-Alive");
			}

			channel.write(httpResponse).addListener(ChannelFutureListener.CLOSE_ON_FAILURE);
	  	}
		else
		{
			httpResponse.headers().set(CONNECTION, "close");

			// Close the connection as soon as the message is sent.
			channel.write(httpResponse).addListener(ChannelFutureListener.CLOSE);
		}
	}

	private HttpResponse createHttpResponse(Request request, Response response)
    {
		if (request.getHeader(Parameters.Query.IGNORE_HTTP_STATUS) == null)
		{
			return new DefaultHttpResponse(request.getHttpVersion(), response.getResponseStatus());
		}

		return new DefaultHttpResponse(request.getHttpVersion(), HttpResponseStatus.OK);
    }

	/**
     * @param response
     * @param httpResponse
     */
    private void addHeaders(Response response, HttpResponse httpResponse)
    {
    	for (String name : response.getHeaderNames())
    	{
    		for (String value : response.getHeaders(name))
    		{
    			httpResponse.headers().add(name, value);
    		}
    	}
    }
}