Skip to content

Commit dce65ea

Browse files
committed
Use try-with-resources rather than manage streams ourselves
2011 called, and it's asking if you'd like to use some new language features in Java.
1 parent 4fc3ac7 commit dce65ea

File tree

1 file changed

+13
-29
lines changed

1 file changed

+13
-29
lines changed

java/server/src/org/openqa/selenium/remote/server/DriverServlet.java

Lines changed: 13 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919

2020
import static com.google.common.base.Preconditions.checkArgument;
2121
import static com.google.common.base.Strings.nullToEmpty;
22-
import static java.util.Collections.list;
2322

2423
import com.google.common.annotations.VisibleForTesting;
2524
import com.google.common.base.Strings;
@@ -234,25 +233,17 @@ private static HttpRequest createInternalRequest(HttpServletRequest servletReque
234233
path);
235234

236235
Enumeration<String> headerNames = servletRequest.getHeaderNames();
237-
for (String name : list(headerNames)) {
236+
while (headerNames.hasMoreElements()) {
237+
String name = headerNames.nextElement();
238238
Enumeration<String> headerValues = servletRequest.getHeaders(name);
239-
for (String value : list(headerValues)) {
239+
while (headerValues.hasMoreElements()) {
240+
String value = headerValues.nextElement();
240241
request.setHeader(name, value);
241242
}
242243
}
243244

244-
InputStream stream = null;
245-
try {
246-
stream = servletRequest.getInputStream();
245+
try (InputStream stream = servletRequest.getInputStream()) {
247246
request.setContent(ByteStreams.toByteArray(stream));
248-
} finally {
249-
if (stream != null) {
250-
try {
251-
stream.close();
252-
} catch (IOException ignored) {
253-
// Do nothing.
254-
}
255-
}
256247
}
257248

258249
return request;
@@ -266,10 +257,10 @@ private void sendResponse(HttpResponse response, HttpServletResponse servletResp
266257
servletResponse.addHeader(name, value);
267258
}
268259
}
269-
OutputStream output = servletResponse.getOutputStream();
270-
output.write(response.getContent());
271-
output.flush();
272-
output.close();
260+
261+
try (OutputStream output = servletResponse.getOutputStream()) {
262+
output.write(response.getContent());
263+
}
273264
}
274265

275266
private class DriverSessionsSupplier implements Supplier<DriverSessions> {
@@ -326,21 +317,14 @@ public void service(HttpServletRequest request, HttpServletResponse response)
326317
byte[] data = getResourceData(url);
327318
response.setContentLength(data.length);
328319

329-
OutputStream output = response.getOutputStream();
330-
output.write(data);
331-
output.flush();
332-
output.close();
320+
try (OutputStream output = response.getOutputStream()) {
321+
output.write(data);
322+
}
333323
}
334324

335325
private byte[] getResourceData(URL url) throws IOException {
336-
InputStream stream = null;
337-
try {
338-
stream = url.openStream();
326+
try (InputStream stream = url.openStream()) {
339327
return ByteStreams.toByteArray(stream);
340-
} finally {
341-
if (stream != null) {
342-
stream.close();
343-
}
344328
}
345329
}
346330
}

0 commit comments

Comments
 (0)