prevent null pointer exceptions

novoda · Aug 20, 2019 · 01cff2d · 01cff2d
1 parent 646c010
commit 01cff2d
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/demo-simple/src/main/java/com/novoda/downloadmanager/demo/CustomHttpResponse.java b/demo-simple/src/main/java/com/novoda/downloadmanager/demo/CustomHttpResponse.java
@@ -6,6 +6,7 @@
 import java.io.InputStream;
 
 import okhttp3.Response;
+import okhttp3.ResponseBody;
 
 class CustomHttpResponse implements NetworkResponse {
 
@@ -32,16 +33,31 @@ public String header(String name, String defaultValue) {
 
     @Override
     public InputStream openByteStream() throws IOException {
-        return response.body().byteStream();
+        ResponseBody body = response.body();
+        if (body == null) {
+            throw new IOException("Response body is null");
+        } else {
+            return body.byteStream();
+        }
     }
 
     @Override
     public void closeByteStream() throws IOException {
-        response.body().close();
+        ResponseBody body = response.body();
+        if (body == null) {
+            throw new IOException("Response body is null");
+        } else {
+            body.close();
+        }
     }
 
     @Override
     public long bodyContentLength() {
-        return response.body().contentLength();
+        ResponseBody body = response.body();
+        if (body == null) {
+            return -1;
+        } else {
+            return body.contentLength();
+        }
     }
 }