| GenericServletResponse.java |
1 /**
2 * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3 *
4 *
5 *
6 *
7 * The contents of this file are subject to the terms of the Liferay Enterprise
8 * Subscription License ("License"). You may not use this file except in
9 * compliance with the License. You can obtain a copy of the License by
10 * contacting Liferay, Inc. See the License for the specific language governing
11 * permissions and limitations under the License, including but not limited to
12 * distribution rights of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20 * SOFTWARE.
21 */
22
23 package com.liferay.util.servlet;
24
25 import java.io.ByteArrayOutputStream;
26 import java.io.PrintWriter;
27
28 import javax.servlet.ServletOutputStream;
29 import javax.servlet.http.HttpServletResponse;
30 import javax.servlet.http.HttpServletResponseWrapper;
31
32 /**
33 * <a href="GenericServletResponse.java.html"><b><i>View Source</i></b></a>
34 *
35 * @author Brian Wing Shun Chan
36 */
37 public class GenericServletResponse extends HttpServletResponseWrapper {
38
39 public GenericServletResponse(HttpServletResponse response) {
40 super(response);
41
42 _baos = new ByteArrayOutputStream();
43 }
44
45 public byte[] getData() {
46 return _baos.toByteArray();
47 }
48
49 public int getContentLength() {
50 return _contentLength;
51 }
52
53 public void setContentLength(int length) {
54 super.setContentLength(length);
55
56 _contentLength = length;
57 }
58
59 public String getContentType() {
60 return _contentType;
61 }
62
63 public void setContentType(String type) {
64 super.setContentType(type);
65
66 _contentType = type;
67 }
68
69 public ServletOutputStream getOutputStream() {
70 return new GenericServletOutputStream(_baos);
71 }
72
73 public PrintWriter getWriter() {
74 return new PrintWriter(getOutputStream(), true);
75 }
76
77 private ByteArrayOutputStream _baos;
78 private int _contentLength;
79 private String _contentType;
80
81 }