/******************************************************************************
* JBoss, a division of Red Hat *
* Copyright 2006, Red Hat Middleware, LLC, and individual *
* contributors as indicated by the @authors tag. See the *
* copyright.txt in the distribution for a full listing of *
* individual contributors. *
* *
* This is free software; you can redistribute it and/or modify it *
* under the terms of the GNU Lesser General Public License as *
* published by the Free Software Foundation; either version 2.1 of *
* the License, or (at your option) any later version. *
* *
* This software is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU *
* Lesser General Public License for more details. *
* *
* You should have received a copy of the GNU Lesser General Public *
* License along with this software; if not, write to the Free *
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA *
* 02110-1301 USA, or see the FSF site: http://www.fsf.org. *
******************************************************************************/
package org.jboss.portal.portlet.impl.jsr168.api;
import org.jboss.portal.portlet.invocation.PortletInvocation;
import org.jboss.portal.portlet.invocation.response.PortletInvocationResponse;
import org.jboss.portal.portlet.invocation.response.RevalidateMarkupResponse;
import org.jboss.portal.portlet.invocation.response.ContentResponse;
import org.jboss.portal.portlet.invocation.response.ResponseProperties;
import org.jboss.portal.portlet.info.PortletInfo;
import org.jboss.portal.portlet.info.CacheInfo;
import org.jboss.portal.portlet.cache.CacheScope;
import org.jboss.portal.portlet.impl.jsr168.ContentBuffer;
import javax.portlet.MimeResponse;
import javax.portlet.PortletURL;
import javax.portlet.ResourceURL;
import javax.portlet.CacheControl;
import java.io.PrintWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Locale;
import java.util.Map;
/**
* @author <a href="mailto:julien@jboss.org">Julien Viet</a>
* @version $Revision: 630 $
*/
public abstract class MimeResponseImpl extends PortletResponseImpl implements MimeResponse
{
/** Not really used but we need it to memorize what the client set optionally. */
protected int bufferSize;
/** The cache control. */
protected CacheControlImpl cacheControl;
/** . */
private boolean contentTypeSet;
/** . */
private ContentBuffer responseContent;
public MimeResponseImpl(PortletInvocation invocation, PortletRequestImpl preq)
{
super(invocation, preq);
// Configure expiration value
PortletInfo info = preq.container.getInfo();
// 0 means no buffering - we say no buffering
this.bufferSize = 0;
this.contentTypeSet = false;
this.responseContent = new ContentBuffer();
}
protected abstract ContentResponse createMarkupResponse(
ResponseProperties properties,
Map<String, Object> attributeMap,
String contentType,
byte[] bytes,
String chars,
org.jboss.portal.portlet.cache.CacheControl cacheControl
);
public PortletInvocationResponse getResponse()
{
org.jboss.portal.portlet.cache.CacheControl cc;
if (cacheControl != null)
{
cc = new org.jboss.portal.portlet.cache.CacheControl(
cacheControl.getExpirationTime(),
cacheControl.isPublicScope() ? CacheScope.PUBLIC : CacheScope.PRIVATE,
cacheControl.getETag());
}
else
{
PortletInfo info = preq.container.getInfo();
CacheInfo cacheInfo = info.getCache();
cc = new org.jboss.portal.portlet.cache.CacheControl(cacheInfo.getExpirationSecs(), CacheScope.PRIVATE, null);
}
//
if (cacheControl != null && cacheControl.useCachedContent())
{
return new RevalidateMarkupResponse(cc);
}
else
{
return createMarkupResponse(
getProperties(false),
preq.attributes.getAttributeMap(),
responseContent.getContentType(),
responseContent.getBytes(),
responseContent.getChars(),
cc);
}
}
public String getContentType()
{
return contentTypeSet ? responseContent.getContentType() : null;
}
public void setContentType(String contentType)
{
if (responseContent.getContentType() == null)
{
responseContent.setContentType(contentType);
contentTypeSet = true;
}
}
public PrintWriter getWriter() throws IOException
{
if (responseContent.getContentType() == null)
{
responseContent.setContentType(preq.getResponseContentType());
}
//
return responseContent.getWriter();
}
public OutputStream getPortletOutputStream() throws IOException
{
if (responseContent.getContentType() == null)
{
responseContent.setContentType(preq.getResponseContentType());
}
//
return responseContent.getOutputStream();
}
public PortletURL createRenderURL()
{
return PortletURLImpl.createRenderURL(invocation, preq);
}
public PortletURL createActionURL()
{
return PortletURLImpl.createActionURL(invocation, preq);
}
public String getCharacterEncoding()
{
return invocation.getContext().getMarkupInfo().getCharset();
}
public Locale getLocale()
{
return invocation.getUserContext().getLocale();
}
public void setBufferSize(int bufferSize)
{
if (bufferSize > -0)
{
this.bufferSize = bufferSize;
}
}
public int getBufferSize()
{
return bufferSize;
}
public void flushBuffer() throws IOException
{
responseContent.commit();
}
public void resetBuffer()
{
// Clear the buffer
responseContent.reset();
}
public void reset()
{
// Clear the buffer
resetBuffer();
// And properties
getProperties().clear();
}
public boolean isCommitted()
{
return responseContent.isCommited();
}
public ResourceURL createResourceURL()
{
return ResourceURLImpl.createResourceURL(invocation, preq);
}
public void addProperty(String key, String value) throws IllegalArgumentException
{
if (MimeResponse.EXPIRATION_CACHE.equals(key))
{
if (value != null)
{
try
{
int expirationSecs = Integer.parseInt(value);
getCacheControl().setExpirationTime(expirationSecs);
}
catch (NumberFormatException e)
{
// todo: Make the portlet log instead
// log.warn("Portlet " + invocation.getTarget() +
// " set a non integer cache value override during render " + value, e);
}
}
}
else if (MimeResponse.ETAG.equals(key))
{
if (value != null)
{
getCacheControl().setETag(value);
}
}
else if (MimeResponse.USE_CACHED_CONTENT.equals(key))
{
if (value != null)
{
getCacheControl().setUseCachedContent(true);
}
}
else
{
super.addProperty(key, value);
}
}
public CacheControl getCacheControl()
{
if (cacheControl == null)
{
PortletInfo info = preq.container.getInfo();
CacheInfo cacheInfo = info.getCache();
cacheControl = new CacheControlImpl(cacheInfo.getExpirationSecs(), CacheScope.PRIVATE);
}
return cacheControl;
}
}