Use updated method to lookup extension by unique id (#2741)

Signed-off-by: Craig Perkins <cwperx@amazon.com>
opensearch-project · May 5, 2023 · 9a271ec · 9a271ec
1 parent 26cafdc
commit 9a271ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/opensearch/security/transport/SecurityRequestHandler.java b/src/main/java/org/opensearch/security/transport/SecurityRequestHandler.java
@@ -331,7 +331,7 @@ protected void addAdditionalContextValues(final String action, final TransportRe
         String extensionUniqueId = getThreadContext().getHeader("extension_unique_id");
         if (extensionUniqueId != null) {
             ExtensionsManager extManager = OpenSearchSecurityPlugin.GuiceHolder.getExtensionsManager();
-            if (extManager.getExtensionIdMap().containsKey(extensionUniqueId)) {
+            if (extManager.lookupInitializedExtensionById(extensionUniqueId).isPresent()) {
                 getThreadContext().putTransient(ConfigConstants.OPENDISTRO_SECURITY_SSL_TRANSPORT_EXTENSION_REQUEST, Boolean.TRUE);
             }
         }