fix: Fixed that callback support was gone for polling in download man…

…ager
pct-org · Oct 2, 2020 · 1e753e2 · 1e753e2
1 parent 89a945e
commit 1e753e2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/app/modules/DownloadManager/DownloadManager.js b/app/modules/DownloadManager/DownloadManager.js
@@ -137,7 +137,7 @@ export class DownloadManager extends React.Component {
    *
    * @returns {Promise<void>}
    */
-  handlePollDownload = (download) => {
+  handlePollDownload = (download, callback = null) => {
     const { apollo } = this.props
 
     if (this.pollingDownloads[download._id]) {
@@ -152,7 +152,12 @@ export class DownloadManager extends React.Component {
       },
     }).subscribe(({ data }) => {
       if (data?.download) {
-        this.handleUpdateDownload(data?.download)
+        if (callback) {
+          callback(data?.download)
+
+        } else {
+          this.handleUpdateDownload(data?.download)
+        }
       }
     })
   }