From aeb0b4113762102e78dc967d4129db7a5d3ae9ee Mon Sep 17 00:00:00 2001
From: Athithyaa Selvam <aselvam@cloudera.com>
Date: Sun, 8 Oct 2023 20:45:03 -0700
Subject: [PATCH] chunksize and max number of concurrent connections config
 support

(cherry picked from commit 2cf8b9817293637ae6307228a2cd70800fd60ac0)
---
 apps/filebrowser/src/filebrowser/conf.py      | 12 +++++++
 .../templates/listdir_components.mako         | 31 ++++++++-----------
 2 files changed, 25 insertions(+), 18 deletions(-)

diff --git a/apps/filebrowser/src/filebrowser/conf.py b/apps/filebrowser/src/filebrowser/conf.py
index 7681e984071..ba338c572c7 100644
--- a/apps/filebrowser/src/filebrowser/conf.py
+++ b/apps/filebrowser/src/filebrowser/conf.py
@@ -40,6 +40,18 @@
   default=None,
   type=str)
 
+FILE_UPLOAD_CHUNK_SIZE = Config(
+  key="file_upload_chunk_size",
+  default=5000000,
+  type=int,
+  help=_('flag to change the chunk size of file upload'))
+
+CONCURRENT_MAX_CONNECTIONS = Config(
+  key="concurrent_max_connections",
+  default=5,
+  type=int,
+  help=_('flag to set the maximum number of conncurrent connections to upload file'))
+
 def get_desktop_enable_download():
   """Get desktop enable_download default"""
   return ENABLE_DOWNLOAD.get()
diff --git a/apps/filebrowser/src/filebrowser/templates/listdir_components.mako b/apps/filebrowser/src/filebrowser/templates/listdir_components.mako
index 44d0a80e65e..f7c61139b46 100644
--- a/apps/filebrowser/src/filebrowser/templates/listdir_components.mako
+++ b/apps/filebrowser/src/filebrowser/templates/listdir_components.mako
@@ -22,7 +22,7 @@ from django.template.defaultfilters import urlencode, stringformat, filesizeform
 from desktop.lib.django_util import reverse_with_get, extract_field_data
 from django.utils.encoding import smart_str
 
-from filebrowser.conf import ENABLE_EXTRACT_UPLOADED_ARCHIVE
+from filebrowser.conf import ENABLE_EXTRACT_UPLOADED_ARCHIVE, FILE_UPLOAD_CHUNK_SIZE, CONCURRENT_MAX_CONNECTIONS
 
 if sys.version_info[0] > 2:
   from django.utils.translation import gettext as _
@@ -2059,11 +2059,13 @@ else:
                   inputName: "hdfs_file"
               }
           },
+          maxConnections: window.CONCURRENT_MAX_CONNECTIONS || 5,
           chunking: {
               enabled: true,
               concurrent: {
                   enabled: true
               },
+              partSize: window.FILE_UPLOAD_CHUNK_SIZE || 5000000,
               success: {
                   endpoint: "/filebrowser/uploaddone/"
               },
@@ -2077,24 +2079,15 @@ else:
           },
 
           template: 'qq-template',
-          // params: {
-          //   dest: self.currentPath(),
-          //   fileFieldLabel: "hdfs_file",
-          //   partIndex: "qqpartindex",
-          //   partByteOffset: "qqpartbyteoffset",
-          //   chunkSize: "qqchunksize",
-          //   totalFileSize: "qqtotalfilesize",
-          //   totalParts: "qqtotalparts"
-          // },
           callbacks: {
-                onProgress: function (id, fileName, loaded, total) {
-                console.log(loaded);
-                $('.qq-upload-files').find('li').each(function(){
-                  var listItem = $(this);
-                  if (listItem.find('.qq-upload-file-selector').text() == fileName){
-                    listItem.find('.progress-row-bar').css('width', (loaded/total)*100 + '%');
-                  }
-                });
+              onProgress: function (id, fileName, loaded, total) {
+              console.log(loaded);
+              $('.qq-upload-files').find('li').each(function(){
+                var listItem = $(this);
+                if (listItem.find('.qq-upload-file-selector').text() == fileName){
+                  listItem.find('.progress-row-bar').css('width', (loaded/total)*100 + '%');
+                }
+              });
               },
               onComplete: function (id, fileName, response) {
                 self.pendingUploads(self.pendingUploads() - 1);
@@ -2115,6 +2108,8 @@ else:
                 $('#uploadFileModal').modal('hide');
               },
               onSubmit: function (id, fileName, responseJSON) {
+                var newPath = "/filebrowser/uploadchunks/file?dest=" + encodeURIComponent(self.currentPath());
+                this.setEndpoint(newPath);
                 self.pendingUploads(self.pendingUploads() + 1);
               },
               onCancel: function (id, fileName) {