Emit Metrics for Broker Adaptive Server Selector type (#12482)

apache · Feb 23, 2024 · 59551e4 · 59551e4
1 parent efd7786
commit 59551e4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pinot-broker/src/main/java/org/apache/pinot/broker/broker/helix/BaseBrokerStarter.java b/pinot-broker/src/main/java/org/apache/pinot/broker/broker/helix/BaseBrokerStarter.java
@@ -260,6 +260,9 @@ public void start()
         _brokerConf.getProperty(Broker.CONFIG_OF_ALLOWED_TABLES_FOR_EMITTING_METRICS, Collections.emptyList()));
     _brokerMetrics.initializeGlobalMeters();
     _brokerMetrics.setValueOfGlobalGauge(BrokerGauge.VERSION, PinotVersion.VERSION_METRIC_NAME, 1);
+    _brokerMetrics.setValueOfGlobalGauge(BrokerGauge.ADAPTIVE_SERVER_SELECTOR_TYPE,
+        _brokerConf.getProperty(Broker.AdaptiveServerSelector.CONFIG_OF_TYPE,
+            Broker.AdaptiveServerSelector.DEFAULT_TYPE), 1);
     BrokerMetrics.register(_brokerMetrics);
     // Set up request handling classes
     _serverRoutingStatsManager = new ServerRoutingStatsManager(_brokerConf, _brokerMetrics);

diff --git a/pinot-common/src/main/java/org/apache/pinot/common/metrics/BrokerGauge.java b/pinot-common/src/main/java/org/apache/pinot/common/metrics/BrokerGauge.java
@@ -58,6 +58,12 @@ public enum BrokerGauge implements AbstractMetrics.Gauge {
   NETTY_POOLED_THREADLOCALCACHE("bytes", true),
   NETTY_POOLED_CHUNK_SIZE("bytes", true),
 
+  /**
+   * The type of Adaptive Server Selector.
+   * See {@link org.apache.pinot.spi.utils.CommonConstants.Broker.AdaptiveServerSelector.Type}
+   */
+  ADAPTIVE_SERVER_SELECTOR_TYPE("adaptiveServerSelectorType", true),
+
   /**
    * The queue size of ServerRoutingStatsManager main executor service.
    */