Add client execution timeout

VertaAI · Jul 26, 2023 · d0ee082 · d0ee082 · github-actions · Jul 26, 2023
1 parent 3f8cfc2
commit d0ee082
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/...ommon/src/main/java/ai/verta/modeldb/common/artifactStore/storageservice/s3/S3Client.java b/...ommon/src/main/java/ai/verta/modeldb/common/artifactStore/storageservice/s3/S3Client.java
@@ -42,7 +42,8 @@ public S3Client(S3Config s3Config) throws IOException, ModelDBException {
     this.defaultClientConfig =
         new ClientConfiguration()
             .withMaxConnections(s3Config.getConnectionPoolSize())
-            .withRequestTimeout(s3Config.getRequestTimeoutMs());
+            .withRequestTimeout(s3Config.getRequestTimeoutMs())
+            .withClientExecutionTimeout(s3Config.getClientExecutionTimeout());
 
     // Start the counter with one because this class has a reference to it
     referenceCounter = new AtomicInteger(1);

diff --git a/backend/common/src/main/java/ai/verta/modeldb/common/config/S3Config.java b/backend/common/src/main/java/ai/verta/modeldb/common/config/S3Config.java
@@ -24,6 +24,7 @@ public class S3Config {
   @JsonProperty private String awsRegion = "us-east-1";
   @JsonProperty private Boolean s3presignedURLEnabled = true;
   @JsonProperty private Integer requestTimeoutMs = 30 * 1000;
+  @JsonProperty private Integer clientExecutionTimeout = 40 * 1000;
   @JsonProperty private Integer connectionPoolSize = 75;
 
   public void validate(String base) throws InvalidConfigException {