manual rebase

apache · Oct 22, 2024 · 8dd32aa · 8dd32aa
1 parent 04dc09d
commit 8dd32aa
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/...ck-core/src/main/java/org/apache/bigtop/manager/stack/core/tarball/TarballDownloader.java b/...ck-core/src/main/java/org/apache/bigtop/manager/stack/core/tarball/TarballDownloader.java
@@ -38,7 +38,18 @@ public static Boolean downloadFile(String fileUrl, String saveDir) {
             int responseCode = httpConn.getResponseCode();
 
             if (responseCode == HttpURLConnection.HTTP_OK) {
-                String fileName = fileUrl.substring(fileUrl.lastIndexOf("/") + 1);
+                String fileName = "";
+                String disposition = httpConn.getHeaderField("Content-Disposition");
+
+                if (disposition != null) {
+                    int index = disposition.indexOf("filename=");
+                    if (index > 0) {
+                        fileName = disposition.substring(index + 10, disposition.length() - 1);
+                    }
+                } else {
+                    fileName = fileUrl.substring(fileUrl.lastIndexOf("/") + 1);
+                }
+
                 InputStream inputStream = httpConn.getInputStream();
                 String saveFilePath = saveDir + File.separator + fileName;