[OpAMP] Enable/disable snapshot profiling with remote config

opamp/src/main/java/com/splunk/opentelemetry/opamp/OpampActivator.java

@@ -33,6 +33,7 @@
 import com.splunk.opentelemetry.profiler.ProfilingDataType;
 import com.splunk.opentelemetry.profiler.ProfilingSupervisor;
 import com.splunk.opentelemetry.profiler.exporter.PprofLogDataExporter;
+import com.splunk.opentelemetry.profiler.snapshot.SnapshotProfilingSupervisor;
 import io.opentelemetry.javaagent.extension.AgentListener;
 import io.opentelemetry.opamp.client.OpampClient;
 import io.opentelemetry.opamp.client.OpampClientBuilder;
@@ -152,7 +153,9 @@ private static RemoteConfigProcessor buildRemoteConfigProcessor(
       OpampClientConfiguration opampClientConfiguration) {
     if (opampClientConfiguration.isRemoteConfigurationEnabled()) {
       return new RemoteConfigProcessorImpl(
-          ProfilingSupervisor.SUPPLIER.get(), effectiveConfigReporter);
+          ProfilingSupervisor.SUPPLIER.get(),
+          SnapshotProfilingSupervisor.SUPPLIER.get(),
+          effectiveConfigReporter);
     }
     return RemoteConfigProcessor.NOOP;
   }

opamp/src/main/java/com/splunk/opentelemetry/opamp/RemoteConfigProcessorImpl.java

@@ -23,6 +23,9 @@
 import com.splunk.opentelemetry.profiler.ProfilerConfiguration;
 import com.splunk.opentelemetry.profiler.ProfilerDeclarativeConfigurationFactory;
 import com.splunk.opentelemetry.profiler.ProfilingSupervisor;
+import com.splunk.opentelemetry.profiler.snapshot.SnapshotProfilingConfiguration;
+import com.splunk.opentelemetry.profiler.snapshot.SnapshotProfilingDeclarativeConfigurationFactory;
+import com.splunk.opentelemetry.profiler.snapshot.SnapshotProfilingSupervisor;
 import io.opentelemetry.api.incubator.config.DeclarativeConfigProperties;
 import io.opentelemetry.opamp.client.OpampClient;
 import io.opentelemetry.sdk.autoconfigure.declarativeconfig.DeclarativeConfiguration;
@@ -44,15 +47,18 @@ public class RemoteConfigProcessorImpl implements RemoteConfigProcessor {
   private static final String PROFILING_NODE_NAME = "profiling";
 
   private final ProfilingSupervisor profilingSupervisor;
+  private final SnapshotProfilingSupervisor snapshotProfilingSupervisor;
   private final EffectiveConfigReporter effectiveConfigReporter;
 
   public RemoteConfigProcessorImpl(
-      ProfilingSupervisor profilingSupervisor, EffectiveConfigReporter effectiveConfigReporter) {
+      ProfilingSupervisor profilingSupervisor,
+      SnapshotProfilingSupervisor snapshotProfilingSupervisor,
+      EffectiveConfigReporter effectiveConfigReporter) {
     this.profilingSupervisor = Objects.requireNonNull(profilingSupervisor);
+    this.snapshotProfilingSupervisor = Objects.requireNonNull(snapshotProfilingSupervisor);
     this.effectiveConfigReporter = Objects.requireNonNull(effectiveConfigReporter);
   }
 
-  @Override
   public void applyConfig(AgentRemoteConfig remoteConfig, OpampClient opampClient) {
     Objects.requireNonNull(opampClient);
 
@@ -77,22 +83,8 @@ public void applyConfig(AgentRemoteConfig remoteConfig, OpampClient opampClient)
 
       // Update profiler configuration only when profiling node exists
       if (distributionRemoteConfigProperties.getPropertyKeys().contains(PROFILING_NODE_NAME)) {
-        ProfilerConfiguration receivedProfilerConfig =
-            ProfilerDeclarativeConfigurationFactory.create(
-                distributionRemoteConfigProperties.getStructured(PROFILING_NODE_NAME, empty()));
-
-        ProfilerConfiguration currentProfilerConfiguration = ProfilerConfiguration.SUPPLIER.get();
-        ProfilerConfiguration updatedProfilerConfig =
-            currentProfilerConfiguration.toBuilder()
-                .setEnabled(receivedProfilerConfig.isEnabled())
-                .setCallStackInterval(receivedProfilerConfig.getCallStackInterval())
-                .setMemoryEnabled(receivedProfilerConfig.getMemoryEnabled())
-                .build();
-
-        if (!currentProfilerConfiguration.equals(updatedProfilerConfig)) {
-          ProfilerConfiguration.SUPPLIER.configure(updatedProfilerConfig);
-          profilingSupervisor.requestReinitializeProfiling();
-        }
+        applyAlwaysOnProfilingConfiguration(distributionRemoteConfigProperties);
+        applySnapshotProfilingConfiguration(distributionRemoteConfigProperties);
       }
 
       // Confirm to the OpAMP Server that remote config has been applied.
@@ -115,6 +107,43 @@ public void applyConfig(AgentRemoteConfig remoteConfig, OpampClient opampClient)
     effectiveConfigReporter.reportEffectiveConfigIfChanged();
   }
 
+  private void applyAlwaysOnProfilingConfiguration(
+      DeclarativeConfigProperties distributionRemoteConfigProperties) {
+    ProfilerConfiguration receivedConfiguration =
+        ProfilerDeclarativeConfigurationFactory.create(
+            distributionRemoteConfigProperties.getStructured(PROFILING_NODE_NAME, empty()));
+
+    ProfilerConfiguration currentConfiguration = ProfilerConfiguration.SUPPLIER.get();
+    ProfilerConfiguration updatedConfiguration =
+        currentConfiguration.toBuilder()
+            .setEnabled(receivedConfiguration.isEnabled())
+            .setCallStackInterval(receivedConfiguration.getCallStackInterval())
+            .setMemoryEnabled(receivedConfiguration.getMemoryEnabled())
+            .build();
+
+    if (!currentConfiguration.equals(updatedConfiguration)) {
+      ProfilerConfiguration.SUPPLIER.configure(updatedConfiguration);
+      profilingSupervisor.requestReinitializeProfiling();
+    }
+  }
+
+  private void applySnapshotProfilingConfiguration(
+      DeclarativeConfigProperties distributionRemoteConfigProperties) {
+    SnapshotProfilingConfiguration receivedConfiguration =
+        SnapshotProfilingDeclarativeConfigurationFactory.create(
+            distributionRemoteConfigProperties.getStructured(PROFILING_NODE_NAME, empty()));
+
+    SnapshotProfilingConfiguration currentConfiguration =
+        SnapshotProfilingConfiguration.SUPPLIER.get();
+    SnapshotProfilingConfiguration updatedConfiguration =
+        currentConfiguration.toBuilder().setEnabled(receivedConfiguration.isEnabled()).build();
+
+    if (!currentConfiguration.equals(updatedConfiguration)) {
+      SnapshotProfilingConfiguration.SUPPLIER.configure(updatedConfiguration);
+      snapshotProfilingSupervisor.requestReinitializeProfiling();
+    }
+  }
+
   @VisibleForTesting
   static DeclarativeConfigProperties toDeclarativeConfigProperties(AgentConfigFile configFile) {
     return DeclarativeConfiguration.toConfigProperties(