Implemented option to set POST and GET timeouts

1 files changed, 14 insertions, 2 deletions

diff --git a/src/main/java/com/pina/Holodex.java b/src/main/java/com/pina/Holodex.java
index 793e64f..6f8e8d2 100644
--- a/src/main/java/com/pina/Holodex.java
+++ b/src/main/java/com/pina/Holodex.java
@@ -30,6 +30,8 @@ import java.util.concurrent.TimeUnit;
  */
 public class Holodex {
     private HolodexService service;
+    private int readTimeout = 35;
+    private int writeTimeout = 35;
 
     /**
      * Instantiates a new Holodex with the default base url
@@ -52,8 +54,8 @@ public class Holodex {
 
     private void initializeHolodexService(String apiKey, String baseUrl){
         OkHttpClient.Builder httpClient = new OkHttpClient.Builder()
-                .writeTimeout(35, TimeUnit.SECONDS)
-                .readTimeout(35, TimeUnit.SECONDS);
+                .writeTimeout(writeTimeout, TimeUnit.SECONDS)
+                .readTimeout(readTimeout, TimeUnit.SECONDS);
         httpClient.addInterceptor(chain -> {
             Request original = chain.request();
             Request request = original.newBuilder()
@@ -70,6 +72,16 @@ public class Holodex {
         service = retrofit.create(HolodexService.class);
     }
 
+    private Holodex setReadTimeout(int readTimeout) {
+        this.readTimeout = readTimeout;
+        return this;
+    }
+
+    private Holodex setWriteTimeout(int writeTimeout) {
+        this.writeTimeout = writeTimeout;
+        return this;
+    }
+
     /**
      * Gets a list of upcoming and/or live SimpleVideos matching the VideoQueryBuilder attributes
      *