org.jboss.portletbridge.BufferedRenderResponseWrapper Maven / Gradle / Ivy

Go to download
/**
 * 
 */
package org.jboss.portletbridge;

import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.io.Writer;
import java.util.Locale;

import javax.portlet.PortletURL;
import javax.portlet.RenderResponse;

import org.ajax4jsf.io.FastBufferOutputStream;
import org.ajax4jsf.io.FastBufferWriter;

public class BufferedRenderResponseWrapper implements RenderResponse {

	private static class FastPrintWriter extends PrintWriter {

		private final FastBufferWriter fastBufferWriter;
		
		public FastPrintWriter() {
			this(new FastBufferWriter());
		}
		
		/**
		 * This is to call only by public default constructor
		 * @param writer
		 */
		private FastPrintWriter(FastBufferWriter writer) {
			super(writer);
		
			this.fastBufferWriter = writer;
		}
		
		public void reset() {
			fastBufferWriter.reset();
		}
		
		public void writeTo(Writer writer) throws IOException {
			fastBufferWriter.writeTo(writer);
		}
	}
	
	private final RenderResponse renderResponse;
	
	private FastBufferOutputStream fastBufferStream = null;
	
	private BufferedRenderResponseWrapper.FastPrintWriter fastPrintWriter = null;
	
	public BufferedRenderResponseWrapper(RenderResponse renderResponse) {
		super();
		this.renderResponse = renderResponse;
	}

	/**
	 * Jboss portal workaround, implementation of the Dispatcher needs this method
	 * to get original request.
	 * @return the renderResponse
	 */
	public RenderResponse getResponse() {
		return renderResponse;
	}

	/**
	 * @param arg0
	 * @param arg1
	 * @see javax.portlet.PortletResponse#addProperty(java.lang.String, java.lang.String)
	 */
	public void addProperty(String arg0, String arg1) {
		renderResponse.addProperty(arg0, arg1);
	}

	/**
	 * @param arg0
	 * @param arg1
	 * @see javax.portlet.PortletResponse#setProperty(java.lang.String, java.lang.String)
	 */
	public void setProperty(String arg0, String arg1) {
		renderResponse.setProperty(arg0, arg1);
	}

	/**
	 * @param arg0
	 * @see javax.portlet.RenderResponse#setTitle(java.lang.String)
	 */
	public void setTitle(String arg0) {
		renderResponse.setTitle(arg0);
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#createActionURL()
	 */
	public PortletURL createActionURL() {
		return renderResponse.createActionURL();
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#createRenderURL()
	 */
	public PortletURL createRenderURL() {
		return renderResponse.createRenderURL();
	}

	/**
	 * @param arg0
	 * @return
	 * @see javax.portlet.PortletResponse#encodeURL(java.lang.String)
	 */
	public String encodeURL(String arg0) {
		return renderResponse.encodeURL(arg0);
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#getCharacterEncoding()
	 */
	public String getCharacterEncoding() {
		return renderResponse.getCharacterEncoding();
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#getContentType()
	 */
	public String getContentType() {
		return renderResponse.getContentType();
	}

	/**
	 * @param arg0
	 * @see javax.portlet.RenderResponse#setContentType(java.lang.String)
	 */
	public void setContentType(String arg0) {
		renderResponse.setContentType(arg0);
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#getLocale()
	 */
	public Locale getLocale() {
		return renderResponse.getLocale();
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#getNamespace()
	 */
	public String getNamespace() {
		return renderResponse.getNamespace();
	}

	/**
	 * @return
	 * @throws IOException
	 * @see javax.portlet.RenderResponse#getPortletOutputStream()
	 */
	public OutputStream getPortletOutputStream() throws IOException {
		if (fastBufferStream == null) {
			fastBufferStream = new FastBufferOutputStream();
		}
		
		return fastBufferStream;
	}

	/**
	 * @return
	 * @throws IOException
	 * @see javax.portlet.RenderResponse#getWriter()
	 */
	public PrintWriter getWriter() throws IOException {
		if (fastPrintWriter == null) {
			fastPrintWriter = new FastPrintWriter();
		}
		
		return fastPrintWriter;
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#isCommitted()
	 */
	public boolean isCommitted() {
		return renderResponse.isCommitted();
	}

	private void resetBuffers() {
		if (fastBufferStream != null) {
			fastBufferStream.reset();
		} 
		
		if (fastPrintWriter != null) {
			fastPrintWriter.reset();
		}
	}
	
	/**
	 * 
	 * @see javax.portlet.RenderResponse#reset()
	 */
	public void reset() {
		renderResponse.reset();
		
		resetBuffers();
	}

	/**
	 * 
	 * @see javax.portlet.RenderResponse#resetBuffer()
	 */
	public void resetBuffer() {
		renderResponse.resetBuffer();
	
		resetBuffers();
	}

	/**
	 * @throws IOException
	 * @see javax.portlet.RenderResponse#flushBuffer()
	 */
	public void flushBuffer() throws IOException {
		renderResponse.flushBuffer();
	}

	/**
	 * @param arg0
	 * @see javax.portlet.RenderResponse#setBufferSize(int)
	 */
	public void setBufferSize(int arg0) {
		renderResponse.setBufferSize(arg0);
	}

	/**
	 * @return
	 * @see javax.portlet.RenderResponse#getBufferSize()
	 */
	public int getBufferSize() {
		return renderResponse.getBufferSize();
	}

	public void writeBufferedData() throws IOException {
		if (fastBufferStream != null) {
			fastBufferStream.writeTo(renderResponse.getPortletOutputStream());
		} else if (fastPrintWriter != null) {
			fastPrintWriter.writeTo(renderResponse.getWriter());
		}
	}
	
	public boolean isUseWriter() {
		return fastBufferStream == null;
	}
}