1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
|
12 | |
|
13 | |
|
14 | |
|
15 | |
|
16 | |
|
17 | |
|
18 | |
|
19 | |
|
20 | |
|
21 | |
|
22 | |
package groovyx.net.http; |
23 | |
|
24 | |
import java.io.IOException; |
25 | |
|
26 | |
import org.apache.http.Header; |
27 | |
import org.apache.http.HeaderElement; |
28 | |
import org.apache.http.HttpEntity; |
29 | |
import org.apache.http.HttpException; |
30 | |
import org.apache.http.HttpRequest; |
31 | |
import org.apache.http.HttpRequestInterceptor; |
32 | |
import org.apache.http.HttpResponse; |
33 | |
import org.apache.http.HttpResponseInterceptor; |
34 | |
import org.apache.http.protocol.HttpContext; |
35 | |
|
36 | |
|
37 | |
|
38 | |
|
39 | |
|
40 | 72 | public abstract class ContentEncoding { |
41 | |
|
42 | |
public static final String ACCEPT_ENC_HDR = "Accept-Encoding"; |
43 | |
public static final String CONTENT_ENC_HDR = "Content-Encoding"; |
44 | |
|
45 | |
protected abstract String getContentEncoding(); |
46 | |
protected abstract HttpEntity wrapResponseEntity( HttpEntity raw ); |
47 | |
|
48 | |
public HttpRequestInterceptor getRequestInterceptor() { |
49 | 72 | return new RequestInterceptor(); |
50 | |
} |
51 | |
|
52 | |
public HttpResponseInterceptor getResponseInterceptor() { |
53 | 72 | return new ResponseInterceptor(); |
54 | |
} |
55 | |
|
56 | |
|
57 | |
|
58 | |
|
59 | 16 | public static enum Type { |
60 | 4 | GZIP, |
61 | 4 | COMPRESS, |
62 | 4 | DEFLATE; |
63 | |
|
64 | |
|
65 | |
@Override public String toString() { |
66 | 400 | return this.name().toLowerCase(); |
67 | |
} |
68 | |
} |
69 | |
|
70 | |
|
71 | |
|
72 | |
|
73 | |
|
74 | |
|
75 | 72 | protected class RequestInterceptor implements HttpRequestInterceptor { |
76 | |
public void process( final HttpRequest req, |
77 | |
final HttpContext context ) throws HttpException, IOException { |
78 | |
|
79 | |
|
80 | 128 | String encoding = getContentEncoding(); |
81 | 128 | if ( !req.containsHeader( ACCEPT_ENC_HDR ) ) |
82 | 64 | req.addHeader( ACCEPT_ENC_HDR, encoding ); |
83 | |
|
84 | |
else { |
85 | 64 | StringBuilder values = new StringBuilder(); |
86 | 128 | for ( Header h : req.getHeaders( ACCEPT_ENC_HDR ) ) |
87 | 64 | values.append( h.getValue() ).append( "," ); |
88 | |
|
89 | 64 | String encList = (!values.toString().contains( encoding )) ? values |
90 | |
.append( encoding ).toString() |
91 | |
: values.toString().substring( 0, values.lastIndexOf( "," ) ); |
92 | |
|
93 | 64 | req.setHeader( ACCEPT_ENC_HDR, encList ); |
94 | |
} |
95 | |
|
96 | |
|
97 | 128 | } |
98 | |
} |
99 | |
|
100 | |
|
101 | |
|
102 | |
|
103 | |
|
104 | |
|
105 | 72 | protected class ResponseInterceptor implements HttpResponseInterceptor { |
106 | |
public void process( final HttpResponse response, final HttpContext context ) |
107 | |
throws HttpException, IOException { |
108 | |
|
109 | 128 | if ( hasEncoding( response, getContentEncoding() ) ) |
110 | 16 | response.setEntity( wrapResponseEntity( response.getEntity() ) ); |
111 | 128 | } |
112 | |
|
113 | |
protected boolean hasEncoding( final HttpResponse response, final String encoding ) { |
114 | 128 | HttpEntity entity = response.getEntity(); |
115 | 128 | if ( entity == null ) return false; |
116 | 112 | Header ceHeader = entity.getContentEncoding(); |
117 | 112 | if ( ceHeader == null ) return false; |
118 | |
|
119 | 32 | HeaderElement[] codecs = ceHeader.getElements(); |
120 | 48 | for ( int i = 0; i < codecs.length; i++ ) |
121 | 32 | if ( encoding.equalsIgnoreCase( codecs[i].getName() ) ) |
122 | 16 | return true; |
123 | |
|
124 | 16 | return false; |
125 | |
} |
126 | |
} |
127 | |
} |