1
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
41 public class GZipResponse extends HttpServletResponseWrapper {
42
43 public GZipResponse(HttpServletResponse response) {
44 super(response);
45
46 _response = response;
47 }
48
49 public void finishResponse() {
50 try {
51 if (_writer != null) {
52 _writer.close();
53 }
54 else if (_stream != null) {
55 _stream.close();
56 }
57 }
58 catch (IOException e) {
59 }
60 }
61
62 public void flushBuffer() throws IOException {
63 if (_stream != null) {
64 _stream.flush();
65 }
66 }
67
68 public ServletOutputStream getOutputStream() throws IOException {
69 if (_writer != null) {
70 throw new IllegalStateException();
71 }
72
73 if (_stream == null) {
74 _stream = _createOutputStream();
75 }
76
77 return _stream;
78 }
79
80 public PrintWriter getWriter() throws IOException {
81 if (_writer != null) {
82 return _writer;
83 }
84
85 if (_stream != null) {
86 throw new IllegalStateException();
87 }
88
89 _stream = _createOutputStream();
90
91 _writer = new PrintWriter(new OutputStreamWriter(
92 _stream, StringPool.UTF8));
94
95 return _writer;
96 }
97
98 private ServletOutputStream _createOutputStream() throws IOException {
99 return new GZipStream(_response);
100 }
101
102 private HttpServletResponse _response = null;
103 private ServletOutputStream _stream = null;
104 private PrintWriter _writer = null;
105
106 }