| GZipResponse.java |
1 /**
2 * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3 *
4 * The contents of this file are subject to the terms of the Liferay Enterprise
5 * Subscription License ("License"). You may not use this file except in
6 * compliance with the License. You can obtain a copy of the License by
7 * contacting Liferay, Inc. See the License for the specific language governing
8 * permissions and limitations under the License, including but not limited to
9 * distribution rights of the Software.
10 *
11 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
12 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
13 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
14 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
15 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
17 * SOFTWARE.
18 */
19
20 package com.liferay.portal.servlet.filters.gzip;
21
22 import com.liferay.portal.kernel.util.StringPool;
23
24 import java.io.IOException;
25 import java.io.OutputStreamWriter;
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="GZipResponse.java.html"><b><i>View Source</i></b></a>
34 *
35 * @author Jayson Falkner
36 * @author Brian Wing Shun Chan
37 *
38 */
39 public class GZipResponse extends HttpServletResponseWrapper {
40
41 public GZipResponse(HttpServletResponse response) {
42 super(response);
43
44 _response = response;
45 }
46
47 public void finishResponse() {
48 try {
49 if (_writer != null) {
50 _writer.close();
51 }
52 else if (_stream != null) {
53 _stream.close();
54 }
55 }
56 catch (IOException e) {
57 }
58 }
59
60 public void flushBuffer() throws IOException {
61 if (_stream != null) {
62 _stream.flush();
63 }
64 }
65
66 public ServletOutputStream getOutputStream() throws IOException {
67 if (_writer != null) {
68 throw new IllegalStateException();
69 }
70
71 if (_stream == null) {
72 _stream = _createOutputStream();
73 }
74
75 return _stream;
76 }
77
78 public PrintWriter getWriter() throws IOException {
79 if (_writer != null) {
80 return _writer;
81 }
82
83 if (_stream != null) {
84 throw new IllegalStateException();
85 }
86
87 _stream = _createOutputStream();
88
89 _writer = new PrintWriter(new OutputStreamWriter(
90 //_stream, _res.getCharacterEncoding()));
91 _stream, StringPool.UTF8));
92
93 return _writer;
94 }
95
96 private ServletOutputStream _createOutputStream() throws IOException {
97 return new GZipStream(_response);
98 }
99
100 private HttpServletResponse _response = null;
101 private ServletOutputStream _stream = null;
102 private PrintWriter _writer = null;
103
104 }