Enable connection keep alive if there is more than one file to upload

gotev · Jan 28, 2015 · 9a7b35b · 9a7b35b
1 parent 6c517d2
commit 9a7b35b
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/com/alexbbb/uploadservice/UploadService.java b/src/com/alexbbb/uploadservice/UploadService.java
@@ -159,7 +159,7 @@ private void handleFileUpload(final String uploadId, final String url, final Str
         InputStream responseStream = null;
 
         try {
-            conn = getMultipartHttpURLConnection(url, method, boundary);
+            conn = getMultipartHttpURLConnection(url, method, boundary, filesToUpload.size());
 
             setRequestHeaders(conn, requestHeaders);
 
@@ -233,18 +233,20 @@ private byte[] getTrailerBytes(final String boundary) throws UnsupportedEncoding
         return builder.toString().getBytes("US-ASCII");
     }
 
-    private
-            HttpURLConnection
-            getMultipartHttpURLConnection(final String url, final String method, final String boundary)
-                                                                                                       throws IOException {
+    private HttpURLConnection getMultipartHttpURLConnection(final String url, final String method,
+                                                            final String boundary, int totalFiles) throws IOException {
         final HttpURLConnection conn = (HttpURLConnection) new URL(url).openConnection();
 
         conn.setDoInput(true);
         conn.setDoOutput(true);
         conn.setUseCaches(false);
         conn.setChunkedStreamingMode(0);
         conn.setRequestMethod(method);
-        conn.setRequestProperty("Connection", "close");
+        if (totalFiles <= 1) {
+            conn.setRequestProperty("Connection", "close");
+        } else {
+            conn.setRequestProperty("Connection", "Keep-Alive");
+        }
         conn.setRequestProperty("ENCTYPE", "multipart/form-data");
         conn.setRequestProperty("Content-Type", "multipart/form-data; boundary=" + boundary);