1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
|
12 | |
|
13 | |
|
14 | |
|
15 | |
|
16 | |
|
17 | |
|
18 | |
|
19 | |
|
20 | |
|
21 | |
|
22 | |
|
23 | |
|
24 | |
|
25 | |
|
26 | |
|
27 | |
|
28 | |
|
29 | |
|
30 | |
package com.jcabi.http.mock; |
31 | |
|
32 | |
import com.jcabi.aspects.Immutable; |
33 | |
import com.jcabi.http.ImmutableHeader; |
34 | |
import com.jcabi.immutable.ArrayMap; |
35 | |
import com.sun.grizzly.tcp.http11.GrizzlyRequest; |
36 | |
import java.io.ByteArrayOutputStream; |
37 | |
import java.io.IOException; |
38 | |
import java.io.InputStream; |
39 | |
import java.net.URI; |
40 | |
import java.nio.charset.Charset; |
41 | |
import java.util.Arrays; |
42 | |
import java.util.Collections; |
43 | |
import java.util.Enumeration; |
44 | |
import java.util.LinkedList; |
45 | |
import java.util.List; |
46 | |
import java.util.Map; |
47 | |
import java.util.concurrent.ConcurrentHashMap; |
48 | |
import java.util.concurrent.ConcurrentMap; |
49 | |
|
50 | |
|
51 | |
|
52 | |
|
53 | |
|
54 | |
|
55 | |
|
56 | |
|
57 | |
@Immutable |
58 | |
final class GrizzlyQuery implements MkQuery { |
59 | |
|
60 | |
|
61 | |
|
62 | |
|
63 | |
private static final String ENCODING = "UTF-8"; |
64 | |
|
65 | |
|
66 | |
|
67 | |
|
68 | 1 | private static final Charset CHARSET = Charset.forName(ENCODING); |
69 | |
|
70 | |
|
71 | |
|
72 | |
|
73 | |
private final transient String mtd; |
74 | |
|
75 | |
|
76 | |
|
77 | |
|
78 | |
@Immutable.Array |
79 | |
private final transient byte[] content; |
80 | |
|
81 | |
|
82 | |
|
83 | |
|
84 | |
private final transient String home; |
85 | |
|
86 | |
|
87 | |
|
88 | |
|
89 | |
private final transient ArrayMap<String, List<String>> hdrs; |
90 | |
|
91 | |
|
92 | |
|
93 | |
|
94 | |
|
95 | |
|
96 | 110 | GrizzlyQuery(final GrizzlyRequest request) throws IOException { |
97 | 110 | request.setCharacterEncoding(GrizzlyQuery.ENCODING); |
98 | 110 | this.home = GrizzlyQuery.uri(request); |
99 | 110 | this.mtd = request.getMethod(); |
100 | 108 | this.hdrs = GrizzlyQuery.headers(request); |
101 | |
|
102 | 110 | final byte[] buffer = new byte[8192]; |
103 | 112 | final InputStream input = request.getInputStream(); |
104 | 111 | final ByteArrayOutputStream output = new ByteArrayOutputStream(); |
105 | 111 | for (int bytes = input.read(buffer); bytes != -1; |
106 | 15 | bytes = input.read(buffer)) { |
107 | 15 | output.write(buffer, 0, bytes); |
108 | |
} |
109 | 110 | this.content = output.toByteArray(); |
110 | 110 | } |
111 | |
|
112 | |
@Override |
113 | |
public URI uri() { |
114 | 6 | return URI.create(this.home); |
115 | |
} |
116 | |
|
117 | |
@Override |
118 | |
public String method() { |
119 | 3 | return this.mtd; |
120 | |
} |
121 | |
|
122 | |
@Override |
123 | |
public Map<String, List<String>> headers() { |
124 | 19 | return Collections.unmodifiableMap(this.hdrs); |
125 | |
} |
126 | |
|
127 | |
@Override |
128 | |
public String body() { |
129 | 17 | return new String(this.content, GrizzlyQuery.CHARSET); |
130 | |
} |
131 | |
|
132 | |
@Override |
133 | |
public byte[] binary() { |
134 | 1 | return Arrays.copyOf(this.content, this.content.length); |
135 | |
} |
136 | |
|
137 | |
|
138 | |
|
139 | |
|
140 | |
|
141 | |
|
142 | |
private static String uri(final GrizzlyRequest request) { |
143 | 110 | final StringBuilder uri = new StringBuilder(request.getRequestURI()); |
144 | 108 | final String query = request.getQueryString(); |
145 | 108 | if (query != null && !query.isEmpty()) { |
146 | 4 | uri.append('?').append(query); |
147 | |
} |
148 | 108 | return uri.toString(); |
149 | |
} |
150 | |
|
151 | |
|
152 | |
|
153 | |
|
154 | |
|
155 | |
|
156 | |
private static ArrayMap<String, List<String>> headers( |
157 | |
final GrizzlyRequest request) { |
158 | 111 | final ConcurrentMap<String, List<String>> headers = |
159 | |
new ConcurrentHashMap<String, List<String>>(0); |
160 | 109 | final Enumeration<?> names = request.getHeaderNames(); |
161 | 812 | while (names.hasMoreElements()) { |
162 | 703 | final String name = names.nextElement().toString(); |
163 | 700 | headers.put( |
164 | |
ImmutableHeader.normalize(name), |
165 | |
GrizzlyQuery.headers(request, name) |
166 | |
); |
167 | 702 | } |
168 | 111 | return new ArrayMap<>(headers); |
169 | |
} |
170 | |
|
171 | |
|
172 | |
|
173 | |
|
174 | |
|
175 | |
|
176 | |
|
177 | |
private static List<String> headers( |
178 | |
final GrizzlyRequest request, final String name) { |
179 | 711 | final List<String> list = new LinkedList<String>(); |
180 | 712 | final Enumeration<?> values = request.getHeaders(name); |
181 | 1410 | while (values.hasMoreElements()) { |
182 | 708 | list.add(values.nextElement().toString()); |
183 | |
} |
184 | 702 | return list; |
185 | |
} |
186 | |
|
187 | |
} |