org.gatein.common.servlet.BufferingResponseWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-common
There is a newer version: 2.2.2.Final
/******************************************************************************
 * JBoss, a division of Red Hat                                               *
 * Copyright 2009, 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.gatein.common.servlet;

import org.gatein.common.io.UndeclaredIOException;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Locale;

/**
 * Redirection of the Writer
 *
 * @author Thomas Heute
 * @version $Revision: 7228 $
 */
public class BufferingResponseWrapper extends HttpServletResponseWrapper
{

   /** . */
   private ServletOutputStreamBuffer outputStream;

   /** . */
   private PrintWriter writer;

   /** . */
   private StringWriter chars;

   /** . */
   private String characterEncoding;

   /** Not really used but we need it to memorize what the client set optionally. */
   protected int bufferSize;

   public BufferingResponseWrapper(HttpServletResponse response)
   {
      super(response);

      // By default inherit the character encoding of the wrapped response
      this.characterEncoding = response.getCharacterEncoding();

      // 0 means no buffering - we say no buffering
      this.bufferSize = 0;
   }

   public String getContent()
   {
      if (outputStream != null)
      {
         try
         {
            outputStream.flush();
            return outputStream.getContent(characterEncoding);
         }
         catch (IOException e)
         {
            throw new UndeclaredIOException(e);
         }
      }
      else if (chars != null)
      {
         writer.flush();
         return chars.toString();
      }
      else
      {
         return null;
      }
   }

   public void addCookie(Cookie arg0)
   {
   }

   public void addDateHeader(String arg0, long arg1)
   {
   }

   public void addHeader(String arg0, String arg1)
   {
   }

   public void addIntHeader(String arg0, int arg1)
   {
   }

   public void sendError(int arg0) throws IOException
   {
   }

   public void sendError(int arg0, String arg1) throws IOException
   {
   }

   public void sendRedirect(String arg0) throws IOException
   {
   }

   public void setDateHeader(String arg0, long arg1)
   {
   }

   public void setHeader(String arg0, String arg1)
   {
   }

   public void setIntHeader(String arg0, int arg1)
   {
   }

   public void setStatus(int arg0)
   {
   }

   public void setStatus(int arg0, String arg1)
   {
   }

   public int getBufferSize()
   {
      return bufferSize;
   }

   public String getCharacterEncoding()
   {
      return characterEncoding;
   }

   public ServletOutputStream getOutputStream() throws IOException
   {
      if (writer != null)
      {
         throw new IllegalStateException("Already obtained a PrintWriter");
      }
      if (outputStream == null)
      {
         outputStream = new ServletOutputStreamBuffer(500);
      }
      return outputStream;
   }

   public PrintWriter getWriter() throws IOException
   {
      if (outputStream != null)
      {
         throw new IllegalStateException("Already obtained a ServletOutputStream");
      }
      if (writer == null)
      {
         chars = new StringWriter();
         writer = new PrintWriter(chars, false);
      }
      return writer;
   }

   public boolean isCommitted()
   {
      return false;
   }

   public void reset()
   {
      resetBuffer();
   }

   public void resetBuffer()
   {
      if (outputStream != null)
      {
         outputStream.reset();
      }
      else if (chars != null)
      {
         writer.flush();
         chars.getBuffer().setLength(0);
      }
   }

   public void setBufferSize(int bufferSize)
   {
      this.bufferSize = bufferSize;
   }

   public void setCharacterEncoding(String characterEncoding)
   {
      this.characterEncoding = characterEncoding;
   }

   public void setContentLength(int arg0)
   {
   }

   public void setContentType(String arg0)
   {
   }

   public void setLocale(Locale arg0)
   {
   }
}