Add timeout when fetching response from downloader.

This commit is contained in:
Kavin 2023-12-03 19:15:22 +00:00
parent 665306b39f
commit 03da4db386
No known key found for this signature in database
GPG Key ID: 6E4598CA5C92C41F
1 changed files with 4 additions and 2 deletions

View File

@ -12,6 +12,8 @@ import java.net.HttpCookie;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
public class DownloaderImpl extends Downloader {
@ -110,8 +112,8 @@ public class DownloaderImpl extends Downloader {
}, Multithreading.getCachedExecutor());
try {
return responseFuture.get();
} catch (InterruptedException | ExecutionException e) {
return responseFuture.get(10, TimeUnit.SECONDS);
} catch (InterruptedException | ExecutionException | TimeoutException e) {
throw new IOException(e);
}
}