ソースを参照

make timeout configurable

Evgeni Gordeev 7 年 前
コミット
ed9320c586

+ 7 - 1
src/main/java/com/github/jhonnymertz/wkhtmltopdf/wrapper/Pdf.java

@@ -42,6 +42,8 @@ public class Pdf {
 
     private boolean hasToc = false;
 
+    private int timeout = 10;
+
     public Pdf() {
         this(new WrapperConfig());
     }
@@ -97,6 +99,10 @@ public class Pdf {
         this.tocParams.add(param, params);
     }
 
+    public void setTimeout(int timeout) {
+        this.timeout = timeout;
+    }
+
     public File saveAs(String path) throws IOException, InterruptedException {
         File file = new File(path);
         FileUtils.writeByteArrayToFile(file, getPDF());
@@ -175,7 +181,7 @@ public class Pdf {
 
     private byte[] getFuture(Future<byte[]> future) {
         try {
-            return future.get(10, TimeUnit.SECONDS);
+            return future.get(this.timeout, TimeUnit.SECONDS);
         } catch (Exception e) {
             throw new RuntimeException(e);
         }