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