feat: allow overriding default & no cookies request timeouts (#160)

lavalink-devs · Feb 1, 2025 · 4279237 · 4279237
1 parent 2f9dd05
commit 4279237
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/main/src/main/java/com/sedmelluq/discord/lavaplayer/tools/io/HttpClientTools.java b/main/src/main/java/com/sedmelluq/discord/lavaplayer/tools/io/HttpClientTools.java
@@ -37,20 +37,34 @@
 public class HttpClientTools {
     private static final Logger log = LoggerFactory.getLogger(HttpClientTools.class);
 
-    public static final RequestConfig DEFAULT_REQUEST_CONFIG = RequestConfig.custom()
+    public static RequestConfig DEFAULT_REQUEST_CONFIG = RequestConfig.custom()
         .setConnectTimeout(3000)
         .setConnectionRequestTimeout(3000)
         .setSocketTimeout(3000)
         .setCookieSpec(CookieSpecs.STANDARD)
         .build();
 
-    private static final RequestConfig NO_COOKIES_REQUEST_CONFIG = RequestConfig.custom()
+    private static RequestConfig NO_COOKIES_REQUEST_CONFIG = RequestConfig.custom()
         .setConnectTimeout(3000)
         .setConnectionRequestTimeout(3000)
         .setSocketTimeout(3000)
         .setCookieSpec(CookieSpecs.IGNORE_COOKIES)
         .build();
 
+    public static void setDefaultRequestTimeout(int timeout, int connectionRequestTimeout, int socketTimeout) {
+        DEFAULT_REQUEST_CONFIG = RequestConfig.copy(DEFAULT_REQUEST_CONFIG)
+            .setConnectTimeout(timeout)
+            .setConnectionRequestTimeout(connectionRequestTimeout)
+            .setSocketTimeout(socketTimeout)
+            .build();
+
+        NO_COOKIES_REQUEST_CONFIG = RequestConfig.copy(NO_COOKIES_REQUEST_CONFIG)
+            .setConnectTimeout(timeout)
+            .setConnectionRequestTimeout(connectionRequestTimeout)
+            .setSocketTimeout(socketTimeout)
+            .build();
+    }
+
     /**
      * @return An HttpClientBuilder which uses the same cookie store for all clients
      */