/*
* Copyright (c) 2003-2009, KNOPFLERFISH project
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following
* conditions are met:
*
* - Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* - Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following
* disclaimer in the documentation and/or other materials
* provided with the distribution.
*
* - Neither the name of the KNOPFLERFISH project nor the names of its
* contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
* OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.knopflerfish.bundle.http;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
public class ResponseWrapper
extends HttpServletResponseWrapper
implements Response
{
// private fields
private PrintWriter pw = null;
private ServletOutputStream sos = null;
// constructors
public ResponseWrapper(final HttpServletResponse response) {
super(response);
}
private HttpServletResponse _getHttpServletResponse() {
return (HttpServletResponse) super.getResponse();
}
// implements Response
public OutputStream getRawOutputStream() {
HttpServletResponse response = _getHttpServletResponse();
if (response instanceof Response) {
return ((Response) response).getRawOutputStream();
}
return null;
}
// override ServletResponse methods
public ServletOutputStream getOutputStream()
throws IOException
{
if (pw != null)
throw new IllegalStateException("getWriter() already called");
if (sos == null) {
final OutputStream os = getRawOutputStream();
if (os == null)
sos = _getHttpServletResponse().getOutputStream();
else
sos = new ServletOutputStreamImpl(os);
}
return sos;
}
public PrintWriter getWriter()
throws IOException
{
if (sos != null)
throw new IllegalStateException("getOutputStream() already called");
if (pw == null) {
final OutputStream os = getRawOutputStream();
if (os == null)
pw = _getHttpServletResponse().getWriter();
else
pw = new PrintWriter(new OutputStreamWriter(os,
getCharacterEncoding()));
}
return pw;
}
} // ResponseWrapper