/*
|
WebBox - Dein Server.
|
Copyright (C) 2017 Ulrich Hilger, http://uhilger.de
|
|
This program is free software: you can redistribute it and/or modify
|
it under the terms of the GNU Affero General Public License as
|
published by the Free Software Foundation, either version 3 of the
|
License, or (at your option) any later version.
|
|
This program 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 Affero General Public License for more details.
|
|
You should have received a copy of the GNU Affero General Public License
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
*/
|
package de.uhilger.wbx.web;
|
|
import java.io.ByteArrayOutputStream;
|
import java.io.PrintWriter;
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletResponse;
|
import javax.servlet.http.HttpServletResponseWrapper;
|
|
/**
|
*
|
* @author ulrich
|
*/
|
public class GenericResponseWrapper extends HttpServletResponseWrapper {
|
private ByteArrayOutputStream output;
|
private int contentLength;
|
private String contentType;
|
|
public GenericResponseWrapper(HttpServletResponse response) {
|
super(response);
|
output=new ByteArrayOutputStream();
|
}
|
|
public byte[] getData() {
|
return output.toByteArray();
|
}
|
|
public ServletOutputStream getOutputStream() {
|
FilterServletOutputStream f = new FilterServletOutputStream(output);
|
return f;
|
}
|
|
public PrintWriter getWriter() {
|
PrintWriter pw = new PrintWriter(getOutputStream(),true);
|
return pw;
|
}
|
|
public void setContentLength(int length) {
|
this.contentLength = length;
|
super.setContentLength(length);
|
}
|
|
public int getContentLength() {
|
return contentLength;
|
}
|
|
public void setContentType(String type) {
|
this.contentType = type;
|
super.setContentType(type);
|
}
|
|
public String getContentType() {
|
return contentType;
|
}
|
}
|
