fix: add is ready pool for mulipart upload (#254)

uploadcare · Apr 20, 2020 · fe7ca2a · fe7ca2a
1 parent e2330bb
commit fe7ca2a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/uploadFile/uploadMultipart.ts b/src/uploadFile/uploadMultipart.ts
@@ -6,6 +6,7 @@ import CancelController from '../tools/CancelController'
 import runWithConcurrency from '../tools/runWithConcurrency'
 import { UploadcareFile } from '../tools/UploadcareFile'
 import { getFileSize } from '../tools/isMultipart'
+import { isReadyPoll } from '../tools/isReadyPoll'
 import retrier from '../tools/retry'
 
 /* Types */
@@ -167,6 +168,22 @@ const uploadMultipart = (
         retryThrottledRequestMaxTimes
       })
     )
+    .then(fileInfo => {
+      if (fileInfo.isReady) {
+        return fileInfo
+      } else {
+        return isReadyPoll({
+          file: fileInfo.uuid,
+          publicKey,
+          baseURL,
+          source,
+          integration,
+          retryThrottledRequestMaxTimes,
+          onProgress,
+          cancel
+        })
+      }
+    })
     .then(fileInfo => new UploadcareFile(fileInfo, { baseCDN }))
 }