| GZipResponse.java |
1 /**
2 * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a copy
5 * of this software and associated documentation files (the "Software"), to deal
6 * in the Software without restriction, including without limitation the rights
7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8 * copies of the Software, and to permit persons to whom the Software is
9 * furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions 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.portal.servlet.filters.gzip;
24
25 import com.liferay.portal.kernel.util.StringPool;
26
27 import java.io.IOException;
28 import java.io.OutputStreamWriter;
29 import java.io.PrintWriter;
30
31 import javax.servlet.ServletOutputStream;
32 import javax.servlet.http.HttpServletResponse;
33 import javax.servlet.http.HttpServletResponseWrapper;
34
35 /**
36 * <a href="GZipResponse.java.html"><b><i>View Source</i></b></a>
37 *
38 * @author Jayson Falkner
39 * @author Brian Wing Shun Chan
40 *
41 */
42 public class GZipResponse extends HttpServletResponseWrapper {
43
44 public GZipResponse(HttpServletResponse response) {
45 super(response);
46
47 _response = response;
48 }
49
50 public void finishResponse() {
51 try {
52 if (_writer != null) {
53 _writer.close();
54 }
55 else if (_stream != null) {
56 _stream.close();
57 }
58 }
59 catch (IOException e) {
60 }
61 }
62
63 public void flushBuffer() throws IOException {
64 if (_stream != null) {
65 _stream.flush();
66 }
67 }
68
69 public ServletOutputStream getOutputStream() throws IOException {
70 if (_writer != null) {
71 throw new IllegalStateException();
72 }
73
74 if (_stream == null) {
75 _stream = _createOutputStream();
76 }
77
78 return _stream;
79 }
80
81 public PrintWriter getWriter() throws IOException {
82 if (_writer != null) {
83 return _writer;
84 }
85
86 if (_stream != null) {
87 throw new IllegalStateException();
88 }
89
90 _stream = _createOutputStream();
91
92 _writer = new PrintWriter(new OutputStreamWriter(
93 //_stream, _res.getCharacterEncoding()));
94 _stream, StringPool.UTF8));
95
96 return _writer;
97 }
98
99 private ServletOutputStream _createOutputStream() throws IOException {
100 return new GZipStream(_response);
101 }
102
103 private HttpServletResponse _response = null;
104 private ServletOutputStream _stream = null;
105 private PrintWriter _writer = null;
106
107 }