1
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
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, 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 }