From b59d55abf04533feeef029fa2182258b185e29bd Mon Sep 17 00:00:00 2001
From: Arpit Patawat <arpitptw@amazon.com>
Date: Fri, 11 Apr 2025 18:12:37 +0530
Subject: [PATCH] Add dependency changes for PA shard metrics

Signed-off-by: Arpit Patawat <arpitptw@amazon.com>
---
 .../commons/metrics/RTFMetrics.java           | 25 +++++++++++++++++++
 .../stats/metrics/StatExceptionCode.java      |  1 +
 .../commons/stats/metrics/StatMetrics.java    |  5 ++++
 3 files changed, 31 insertions(+)

diff --git a/src/main/java/org/opensearch/performanceanalyzer/commons/metrics/RTFMetrics.java b/src/main/java/org/opensearch/performanceanalyzer/commons/metrics/RTFMetrics.java
index f9d1f0a..af0b91c 100644
--- a/src/main/java/org/opensearch/performanceanalyzer/commons/metrics/RTFMetrics.java
+++ b/src/main/java/org/opensearch/performanceanalyzer/commons/metrics/RTFMetrics.java
@@ -108,6 +108,31 @@ public static class Constants {
         }
     }
 
+    public enum OperationsValue implements MetricValue {
+        INDEXING_RATE(Constants.INDEXING_RATE),
+        SEARCH_RATE(Constants.SEARCH_RATE),
+        INDEXING_LATENCY(Constants.INDEXING_LATENCY),
+        SEARCH_LATENCY(Constants.SEARCH_LATENCY);
+
+        private final String value;
+
+        OperationsValue(String value) {
+            this.value = value;
+        }
+
+        @Override
+        public String toString() {
+            return value;
+        }
+
+        public static class Constants {
+            public static final String INDEXING_RATE = "indexing_rate";
+            public static final String SEARCH_RATE = "search_rate";
+            public static final String INDEXING_LATENCY = "indexing_latency";
+            public static final String SEARCH_LATENCY = "search_latency";
+        }
+    }
+
     public enum ThreadPoolValue implements MetricValue {
         THREADPOOL_QUEUE_SIZE(Constants.QUEUE_SIZE_VALUE),
         THREADPOOL_REJECTED_REQS(Constants.REJECTED_VALUE),
diff --git a/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatExceptionCode.java b/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatExceptionCode.java
index dcb9bc2..63f467e 100644
--- a/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatExceptionCode.java
+++ b/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatExceptionCode.java
@@ -75,6 +75,7 @@ public enum StatExceptionCode {
     GC_INFO_COLLECTOR_ERROR("GCInfoCollectorError"),
     HEAP_METRICS_COLLECTOR_ERROR("HeapMetricsCollectorError"),
     RTF_HEAP_METRICS_COLLECTOR_ERROR("RTFHeapMetricsCollectorError"),
+    RTF_SHARD_OPERATION_RATE_COLLECTOR_ERROR("RTFShardOperationRateCollectorError"),
     MOUNTED_PARTITION_METRICS_COLLECTOR_ERROR("MountedPartitionMetricsCollectorError"),
     NETWORK_COLLECTION_ERROR("NetworkCollectionError"),
     OS_METRICS_COLLECTOR_ERROR("OSMetricsCollectorError"),
diff --git a/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatMetrics.java b/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatMetrics.java
index f0a5e00..b8e93a6 100644
--- a/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatMetrics.java
+++ b/src/main/java/org/opensearch/performanceanalyzer/commons/stats/metrics/StatMetrics.java
@@ -52,6 +52,11 @@ public enum StatMetrics implements MeasurementSet {
             "HeapMetricsCollectorExecutionTime", "millis", StatsType.LATENCIES, Statistics.SUM),
     RTF_HEAP_METRICS_COLLECTOR_EXECUTION_TIME(
             "RTFHeapMetricsCollectorExecutionTime", "millis", StatsType.LATENCIES, Statistics.SUM),
+    RTF_SHARD_OPERATION_RATE_COLLECTOR_EXECUTION_TIME(
+            "RTFShardOperationRateCollectorExecutionTime",
+            "millis",
+            StatsType.LATENCIES,
+            Statistics.SUM),
     MOUNTED_PARTITION_METRICS_COLLECTOR_EXECUTION_TIME(
             "MountedPartitionMetricsCollectorExecutionTime",
             "millis",