fix: replace listener

lingochamp · Nov 7, 2019 · 7a5159b · 7a5159b
1 parent 7fc3157
commit 7a5159b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/...wnload-filedownloader/src/main/java/com/liulishuo/filedownloader/DownloadTaskAdapter.java b/...wnload-filedownloader/src/main/java/com/liulishuo/filedownloader/DownloadTaskAdapter.java
@@ -567,6 +567,16 @@ public boolean isContainFinishListener() {
         return !finishListeners.isEmpty();
     }
 
+    public void replaceListener(FileDownloadListener fileDownloadListener) {
+        setListener(fileDownloadListener);
+        if (downloadTask == null) {
+            return;
+        }
+
+        compatListener = CompatListenerAdapter.create(listener);
+        downloadTask.replaceListener(compatListener);
+    }
+
     static final class Builder {
 
         private String url;

diff --git a/okdownload-filedownloader/src/main/java/com/liulishuo/filedownloader/FileDownloader.java b/okdownload-filedownloader/src/main/java/com/liulishuo/filedownloader/FileDownloader.java
@@ -472,8 +472,9 @@ public int replaceListener(int id, FileDownloadListener listener) {
             return 0;
         }
 
-        task.getOrigin().setListener(listener);
-        return task.getOrigin().getId();
+        final DownloadTaskAdapter downloadTask = (DownloadTaskAdapter) task.getOrigin();
+        downloadTask.replaceListener(listener);
+        return downloadTask.getId();
     }
 
     /**