Changes related to PA-RTF merge

src/main/java/org/opensearch/performanceanalyzer/commons/collectors/ScheduledMetricCollectorsExecutor.java

@@ -16,14 +16,21 @@
 import org.apache.logging.log4j.Logger;
 import org.opensearch.performanceanalyzer.commons.stats.ServiceMetrics;
 import org.opensearch.performanceanalyzer.commons.stats.metrics.StatMetrics;
+import org.opensearch.performanceanalyzer.commons.util.Util;
 
+/*
+ * This class is responsible for scheduling the metrics collection.
+ * It creates a thread pool of size collectorThreadCount and runs the metrics collection
+ * in the thread pool.
+ */
 public class ScheduledMetricCollectorsExecutor extends Thread {
     private static final Logger LOG = LogManager.getLogger(ScheduledMetricCollectorsExecutor.class);
     private final int collectorThreadCount;
     private static final int DEFAULT_COLLECTOR_THREAD_COUNT = 5;
     private static final int COLLECTOR_THREAD_KEEPALIVE_SECS = 1000;
     private final boolean checkFeatureDisabledFlag;
     private boolean paEnabled = false;
+    private int collectorsSetting = Util.CollectorMode.RCA.getValue();
     private boolean threadContentionMonitoringEnabled = false;
     private int minTimeIntervalToSleep = Integer.MAX_VALUE;
     private Map<PerformanceAnalyzerMetricsCollector, Long> metricsCollectors;
@@ -48,6 +55,14 @@ public synchronized void setEnabled(final boolean enabled) {
         paEnabled = enabled;
     }
 
+    public synchronized void setCollectorsSetting(final int value) {
+        collectorsSetting = value;
+    }
+
+    public synchronized int getCollectorsSetting() {
+        return collectorsSetting;
+    }
+
     public synchronized boolean getEnabled() {
         return paEnabled;
     }
@@ -117,6 +132,13 @@ public void run() {
                                     StatMetrics.COLLECTORS_MUTED, collector.getCollectorName(), 1);
                             continue;
                         }
+                        if (!canSchedule(collector)) {
+                            LOG.debug(
+                                    "Skipping {} Collector execution since PA is running in CollectorMode: {}",
+                                    collector.getCollectorName(),
+                                    collectorsSetting);
+                            continue;
+                        }
                         metricsCollectors.put(
                                 collector, entry.getValue() + collector.getTimeInterval());
                         if (!collector.inProgress()) {
@@ -159,4 +181,13 @@ public void run() {
             }
         }
     }
+
+    private boolean canSchedule(PerformanceAnalyzerMetricsCollector collector) {
+        if (collector instanceof TelemetryCollector) {
+            return (collectorsSetting == Util.CollectorMode.DUAL.getValue())
+                    || (collectorsSetting == Util.CollectorMode.TELEMETRY.getValue());
+        }
+        return (collectorsSetting == Util.CollectorMode.DUAL.getValue())
+                || (collectorsSetting == Util.CollectorMode.RCA.getValue());
+    }
 }

src/main/java/org/opensearch/performanceanalyzer/commons/collectors/TelemetryCollector.java

@@ -0,0 +1,9 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package org.opensearch.performanceanalyzer.commons.collectors;
+
+/** Marker interface for telemetry collectors. */
+public interface TelemetryCollector {}