Merge pull request #102 from abrami/main

adding Timeouts

Author: abrami

src/main/java/org/texttechnologylab/DockerUnifiedUIMAInterface/driver/DUUIPipelineComponent.java

@@ -71,6 +71,7 @@ public class DUUIPipelineComponent {
 
     private static String sourceView = "sourceView";
     private static String targetView = "targetView";
+    private static String timeout = "timeout";
 
     private String getVersion() throws URISyntaxException, IOException {
         ClassLoader classLoader = DUUIPipelineComponent.class.getClassLoader();
@@ -505,11 +506,17 @@ public DUUIPipelineComponent withSourceView(String viewName) {
         return this;
     }
 
+
     public DUUIPipelineComponent withTargetView(String viewName) {
         _options.put(targetView, viewName);
         return this;
     }
 
+    public DUUIPipelineComponent withTimeout(long lLong) {
+        _options.put(timeout, String.valueOf(lLong));
+        return this;
+    }
+
     public static DUUIPipelineComponent fromJson(String json) throws URISyntaxException, IOException {
         JSONObject jobj = new JSONObject(json);
 
@@ -656,4 +663,8 @@ public DUUIPipelineComponent withSegmentationStrategy(DUUISegmentationStrategy s
     public DUUISegmentationStrategy getSegmentationStrategy() {
         return this.segmentationStrategy;
     }
+
+    public long getTimeout() {
+        return Long.valueOf(_options.getOrDefault(timeout, "60l"));
+    }
 }

src/main/java/org/texttechnologylab/DockerUnifiedUIMAInterface/driver/IDUUIInstantiatedPipelineComponent.java

@@ -143,11 +143,12 @@ public static void process(JCas jc, IDUUIInstantiatedPipelineComponent comp, DUU
         long annotatorStart = serializeEnd;
         int tries = 0;
         HttpResponse<byte[]> resp = null;
-        while(tries < 10) {
+        while (tries < 3) {
             tries++;
             try {
                 HttpRequest request = HttpRequest.newBuilder()
                         .uri(URI.create(queue.getValue0().generateURL() + DUUIComposer.V1_COMPONENT_ENDPOINT_PROCESS))
+                        .timeout(Duration.ofSeconds(comp.getPipelineComponent().getTimeout()))
                         .POST(HttpRequest.BodyPublishers.ofByteArray(ok))
                         .version(HttpClient.Version.HTTP_1_1)
                         .build();
@@ -160,7 +161,7 @@ public static void process(JCas jc, IDUUIInstantiatedPipelineComponent comp, DUU
             }
         }
         if(resp==null) {
-            throw new IOException("Could not reach endpoint after 10 tries!");
+            throw new IOException("Could not reach endpoint after 3 tries!");
         }