fix init master key bug (#1094) (#1096)

Signed-off-by: Yaliang Wu <[email protected]>
(cherry picked from commit b782f29)

Co-authored-by: Yaliang Wu <[email protected]>

Author: opensearch-trigger-bot[bot]

ml-algorithms/src/main/java/org/opensearch/ml/engine/encryptor/EncryptorImpl.java

@@ -27,6 +27,7 @@
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.atomic.AtomicReference;
 
+import static java.util.concurrent.TimeUnit.SECONDS;
 import static org.opensearch.ml.common.CommonValue.MASTER_KEY;
 import static org.opensearch.ml.common.CommonValue.ML_CONFIG_INDEX;
 
@@ -111,7 +112,7 @@ private void initMasterKey() {
                 client.get(getRequest, new LatchedActionListener(ActionListener.<GetResponse>wrap(r -> {
                     if (r.isExists()) {
                         String masterKey = (String) r.getSourceAsMap().get(MASTER_KEY);
-                        setMasterKey(masterKey);
+                        this.masterKey = masterKey;
                     } else {
                         exceptionRef.set(new ResourceNotFoundException("ML encryption master key not initialized yet"));
                     }
@@ -122,6 +123,13 @@ private void initMasterKey() {
             }
         } else {
             exceptionRef.set(new ResourceNotFoundException("ML encryption master key not initialized yet"));
+            latch.countDown();
+        }
+
+        try {
+            latch.await(5, SECONDS);
+        } catch (InterruptedException e) {
+            throw new IllegalStateException(e);
         }
 
         if (exceptionRef.get() != null) {
@@ -132,5 +140,8 @@ private void initMasterKey() {
                 throw new MLException(exceptionRef.get());
             }
         }
+        if (masterKey == null) {
+            throw new ResourceNotFoundException("ML encryption master key not initialized yet");
+        }
     }
 }