1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.servlets;
20
21 import java.io.IOException;
22 import java.io.OutputStream;
23 import java.io.OutputStreamWriter;
24 import java.io.PrintWriter;
25 import java.io.UnsupportedEncodingException;
26 import java.util.zip.Deflater;
27 import java.util.zip.DeflaterOutputStream;
28 import java.util.zip.GZIPOutputStream;
29
30 import javax.servlet.FilterConfig;
31 import javax.servlet.ServletException;
32 import javax.servlet.http.HttpServletRequest;
33 import javax.servlet.http.HttpServletResponse;
34
35 import org.eclipse.jetty.http.gzip.CompressedResponseWrapper;
36 import org.eclipse.jetty.http.gzip.AbstractCompressedStream;
37 import org.eclipse.jetty.io.UncheckedPrintWriter;
38
39
40
41
42
43
44
45
46
47
48
49
50
51 public class IncludableGzipFilter extends GzipFilter
52 {
53 boolean _uncheckedPrintWriter=false;
54
55 @Override
56 public void init(FilterConfig filterConfig) throws ServletException
57 {
58 super.init(filterConfig);
59
60 String tmp=filterConfig.getInitParameter("uncheckedPrintWriter");
61 if (tmp!=null)
62 _uncheckedPrintWriter=Boolean.valueOf(tmp).booleanValue();
63 }
64
65
66
67
68
69 @Override
70 protected CompressedResponseWrapper createWrappedResponse(HttpServletRequest request, HttpServletResponse response, final String compressionType)
71 {
72 CompressedResponseWrapper wrappedResponse = null;
73 if (compressionType.equals(GZIP))
74 {
75 wrappedResponse = new IncludableResponseWrapper(request,response)
76 {
77 @Override
78 protected AbstractCompressedStream newCompressedStream(HttpServletRequest request,HttpServletResponse response,long contentLength,int bufferSize, int minCompressSize) throws IOException
79 {
80 return new AbstractCompressedStream(compressionType,request,response,contentLength,bufferSize,minCompressSize)
81 {
82 @Override
83 protected DeflaterOutputStream createStream() throws IOException
84 {
85 return new GZIPOutputStream(_response.getOutputStream(),_bufferSize);
86 }
87
88 @Override
89 protected void setHeader(String name, String value)
90 {
91 super.setHeader(name, value);
92 HttpServletResponse response = (HttpServletResponse)getResponse();
93 if (!response.containsHeader(name))
94 response.setHeader("org.eclipse.jetty.server.include." + name, value);
95 }
96 };
97 }
98 };
99 }
100 else if (compressionType.equals(DEFLATE))
101 {
102 wrappedResponse = new IncludableResponseWrapper(request,response)
103 {
104 @Override
105 protected AbstractCompressedStream newCompressedStream(HttpServletRequest request,HttpServletResponse response,long contentLength,int bufferSize, int minCompressSize) throws IOException
106 {
107 return new AbstractCompressedStream(compressionType,request,response,contentLength,bufferSize,minCompressSize)
108 {
109 @Override
110 protected DeflaterOutputStream createStream() throws IOException
111 {
112 return new DeflaterOutputStream(_response.getOutputStream(),new Deflater(_deflateCompressionLevel, _deflateNoWrap));
113 }
114
115 @Override
116 protected void setHeader(String name, String value)
117 {
118 super.setHeader(name, value);
119 HttpServletResponse response = (HttpServletResponse)getResponse();
120 if (!response.containsHeader(name))
121 response.setHeader("org.eclipse.jetty.server.include." + name, value);
122 }
123 };
124 }
125 };
126 }
127 else
128 {
129 throw new IllegalStateException(compressionType + " not supported");
130 }
131 configureWrappedResponse(wrappedResponse);
132 return wrappedResponse;
133 }
134
135
136
137 private abstract class IncludableResponseWrapper extends CompressedResponseWrapper
138 {
139 public IncludableResponseWrapper(HttpServletRequest request, HttpServletResponse response)
140 {
141 super(request,response);
142 }
143
144 @Override
145 public void setHeader(String name,String value)
146 {
147 super.setHeader(name,value);
148 HttpServletResponse response = (HttpServletResponse)getResponse();
149 if (!response.containsHeader(name))
150 response.setHeader("org.eclipse.jetty.server.include."+name,value);
151 }
152 @Override
153 protected PrintWriter newWriter(OutputStream out, String encoding) throws UnsupportedEncodingException
154 {
155 if (_uncheckedPrintWriter)
156 return encoding == null?new UncheckedPrintWriter(out):new UncheckedPrintWriter(new OutputStreamWriter(out,encoding));
157 return super.newWriter(out,encoding);
158 }
159 }
160
161 }