Add cancel mechanism to Android #39 #37

Author: wkh237

src/android/src/main/java/com/RNFetchBlob/RNFetchBlob.java

@@ -150,6 +150,11 @@ public void readStream(String path, String encoding, int bufferSize) {
         fs.readStream(path, encoding, bufferSize);
     }
 
+    @ReactMethod
+    public void cancel(String taskId) {
+        RNFetchBlobReq.cancelTask(taskId);
+    }
+
     @ReactMethod
     public void fetchBlob(ReadableMap options, String taskId, String method, String url, ReadableMap headers, String body, final Callback callback) {
         new RNFetchBlobReq(options, taskId, method, url, headers, body, null, callback).run();

src/android/src/main/java/com/RNFetchBlob/RNFetchBlobReq.java

@@ -2,7 +2,6 @@
 
 import android.app.DownloadManager;
 import android.content.BroadcastReceiver;
-import android.content.ContentResolver;
 import android.content.Context;
 import android.content.Intent;
 import android.content.IntentFilter;
@@ -21,17 +20,15 @@
 import java.io.ByteArrayInputStream;
 import java.io.File;
 import java.io.FileInputStream;
-import java.io.FileNotFoundException;
-import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.MalformedURLException;
 import java.net.URL;
+import java.util.HashMap;
 
 import okhttp3.Call;
 import okhttp3.Interceptor;
 import okhttp3.MediaType;
-import okhttp3.MultipartBody;
 import okhttp3.OkHttpClient;
 import okhttp3.Request;
 import okhttp3.RequestBody;
@@ -55,6 +52,8 @@ enum ResponseType {
         FileStorage
     };
 
+    static HashMap<String, Call> taskTable = new HashMap<>();
+
     MediaType contentType = RNFetchBlobConst.MIME_OCTET;
     ReactApplicationContext ctx;
     RNFetchBlobConfig options;
@@ -94,6 +93,14 @@ else if (arrayBody != null)
             requestType = RequestType.WithoutBody;
     }
 
+    public static void cancelTask(String taskId) {
+        if(taskTable.containsKey(taskId)) {
+            Call call = taskTable.get(taskId);
+            call.cancel();
+            taskTable.remove(taskId);
+        }
+    }
+
     @Override
     public void run() {
 
@@ -146,14 +153,14 @@ public void run() {
         else if(this.options.fileCache == true)
             this.destPath = RNFetchBlobFS.getTmpPath(RNFetchBlob.RCTContext, cacheKey);
 
-        OkHttpClient.Builder client;
+        OkHttpClient.Builder clientBuilder;
 
         try {
             // use trusty SSL socket
             if (this.options.trusty) {
-                client = RNFetchBlobUtils.getUnsafeOkHttpClient();
+                clientBuilder = RNFetchBlobUtils.getUnsafeOkHttpClient();
             } else {
-                client = new OkHttpClient.Builder();
+                clientBuilder = new OkHttpClient.Builder();
             }
 
             final Request.Builder builder = new Request.Builder();
@@ -203,7 +210,7 @@ else if(this.options.fileCache == true)
             final Request req = builder.build();
 
 //          create response handler
-            client.addInterceptor(new Interceptor() {
+            clientBuilder.addInterceptor(new Interceptor() {
                 @Override
                 public Response intercept(Chain chain) throws IOException {
                 Response originalResponse = chain.proceed(req);
@@ -233,7 +240,10 @@ public Response intercept(Chain chain) throws IOException {
                 }
             });
 
-            client.build().newCall(req).enqueue(new okhttp3.Callback() {
+            OkHttpClient client = clientBuilder.build();
+            Call call = client.newCall(req);
+            taskTable.put(taskId, call);
+            call.enqueue(new okhttp3.Callback() {
                 @Override
                 public void onFailure(Call call, IOException e) {
                     callback.invoke(e.getLocalizedMessage(), null);
@@ -266,6 +276,7 @@ public void onResponse(Call call, Response response) throws IOException {
 
         } catch (Exception error) {
             error.printStackTrace();
+            taskTable.remove(taskId);
             callback.invoke("RNFetchBlob request error: " + error.getMessage() + error.getCause());
         }
     }
@@ -300,6 +311,8 @@ private void done(Response resp) {
                 }
                 break;
         }
+        if(taskTable.containsKey(taskId))
+            taskTable.remove(taskId);
     }
 
     /**