test changes for PA

Signed-off-by: Gagan Juneja <[email protected]>
Gaganjuneja · May 6, 2024 · d7250c5 · d7250c5
1 parent b4692c8
commit d7250c5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/server/src/main/java/org/opensearch/cluster/service/ClusterService.java b/server/src/main/java/org/opensearch/cluster/service/ClusterService.java
@@ -69,6 +69,8 @@ public class ClusterService extends AbstractLifecycleComponent {
 
     private final ClusterApplierService clusterApplierService;
 
+    private MetricsRegistry metricsRegistry;
+
     public static final org.opensearch.common.settings.Setting.AffixSetting<String> USER_DEFINED_METADATA = Setting.prefixKeySetting(
         "cluster.metadata.",
         (key) -> Setting.simpleString(key, Property.Dynamic, Property.NodeScope)
@@ -117,6 +119,10 @@ public ClusterService(
         this.clusterApplierService = clusterApplierService;
     }
 
+    public MetricsRegistry setMetricsRegistry(MetricsRegistry metricsRegistry) {
+        return this.metricsRegistry = metricsRegistry;
+    }
+
     public synchronized void setNodeConnectionsService(NodeConnectionsService nodeConnectionsService) {
         clusterApplierService.setNodeConnectionsService(nodeConnectionsService);
     }

diff --git a/server/src/main/java/org/opensearch/node/Node.java b/server/src/main/java/org/opensearch/node/Node.java
@@ -637,9 +637,9 @@ protected Node(
                 tracerFactory = new NoopTracerFactory();
                 metricsRegistryFactory = new NoopMetricsRegistryFactory();
             }
-
             tracer = tracerFactory.getTracer();
             metricsRegistry = metricsRegistryFactory.getMetricsRegistry();
+            clusterService.setMetricsRegistry(metricsRegistry);
             resourcesToClose.add(tracer::close);
             resourcesToClose.add(metricsRegistry::close);