[backport 2.14] increase the wait timeout to fetch the master key (#3151

) (#3178) (#3213) * increase the wait timeout to fetch the master key (#3151) (#3178) * increase the wait timeout to fetch the master key Signed-off-by: Jing Zhang <[email protected]> * fix UTs Signed-off-by: Jing Zhang <[email protected]> --------- Signed-off-by: Jing Zhang <[email protected]> * spotless Signed-off-by: Jing Zhang <[email protected]> * fix UT Signed-off-by: Jing Zhang <[email protected]> --------- Signed-off-by: Jing Zhang <[email protected]> Co-authored-by: opensearch-trigger-bot[bot] <98922864+opensearch-trigger-bot[bot]@users.noreply.github.com>
opensearch-project · Nov 25, 2024 · 013d105 · 013d105
1 parent 00aea6a
commit 013d105
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/encryptor/EncryptorImpl.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/encryptor/EncryptorImpl.java
@@ -125,7 +125,10 @@ private void initMasterKey() {
         }
 
         try {
-            latch.await(5, SECONDS);
+            boolean completed = latch.await(3, SECONDS);
+            if (!completed) {
+                throw new MLException("Fetching master key timed out.");
+            }
         } catch (InterruptedException e) {
             throw new IllegalStateException(e);
         }