spring boot micrometer

Author: shalom938

jvm-common/src/main/kotlin/org/digma/intellij/plugin/idea/execution/flavor/SpringBootMicrometerInstrumentationFlavor.kt

@@ -4,6 +4,9 @@ import com.intellij.execution.configurations.RunConfiguration
 import com.intellij.execution.configurations.RunnerSettings
 import com.intellij.execution.configurations.SimpleProgramParameters
 import com.intellij.openapi.module.Module
+import org.digma.intellij.plugin.analytics.LOCAL_ENV
+import org.digma.intellij.plugin.analytics.LOCAL_TESTS_ENV
+import org.digma.intellij.plugin.analytics.isCentralized
 import org.digma.intellij.plugin.errorreporting.ErrorReporter
 import org.digma.intellij.plugin.execution.RunConfigurationInstrumentationService
 import org.digma.intellij.plugin.idea.execution.JavaToolOptionsBuilder
@@ -16,6 +19,10 @@ import org.digma.intellij.plugin.idea.execution.getModuleMetadata
 import org.digma.intellij.plugin.idea.execution.isMicrometerTracingInSettings
 
 
+private const val ENVIRONMENT_RESOURCE_ATTRIBUTE = "MANAGEMENT_OPENTELEMETRY_RESOURCE-ATTRIBUTES_digma_environment"
+private const val ENVIRONMENT_ID_RESOURCE_ATTRIBUTE = "MANAGEMENT_OPENTELEMETRY_RESOURCE-ATTRIBUTES_digma_environment_id"
+
+
 //SpringBootMicrometerInstrumentationFlavor supports the same gradle/maven tasks as DefaultInstrumentationFlavor.
 class SpringBootMicrometerInstrumentationFlavor : DefaultInstrumentationFlavor() {
 
@@ -109,8 +116,24 @@ class SpringBootMicrometerInstrumentationFlavor : DefaultInstrumentationFlavor()
         return try {
             val isTest = isTest(instrumentationService, configuration, params)
 
+            if (needToAddDigmaEnvironmentAttribute(parametersExtractor)) {
+                val envAttribute = if (isTest) {
+                    "$ENVIRONMENT_RESOURCE_ATTRIBUTE=$LOCAL_TESTS_ENV"
+                } else {
+                    "$ENVIRONMENT_RESOURCE_ATTRIBUTE=$LOCAL_ENV"
+                }
+
+                otelResourceAttributesBuilder.withOtelResourceAttribute(envAttribute)
+            }
+
+            if (!hasEnvironmentIdAttribute(parametersExtractor) &&
+                isCentralized(configuration.project)
+            ) {
+                otelResourceAttributesBuilder.withUserId()
+            }
+
             otelResourceAttributesBuilder
-                .withCommonResourceAttributes(isTest, parametersExtractor)
+                .withScmCommitId()
                 .build()
 
         } catch (e: Throwable) {
@@ -120,6 +143,19 @@ class SpringBootMicrometerInstrumentationFlavor : DefaultInstrumentationFlavor()
     }
 
 
+    private fun needToAddDigmaEnvironmentAttribute(parametersExtractor: ParametersExtractor): Boolean {
+        return !hasEnvironmentAttribute(parametersExtractor) && !hasEnvironmentIdAttribute(parametersExtractor)
+    }
+
+    private fun hasEnvironmentAttribute(parametersExtractor: ParametersExtractor): Boolean {
+        return parametersExtractor.extractEnvValue(ENVIRONMENT_RESOURCE_ATTRIBUTE) != null
+    }
+
+    private fun hasEnvironmentIdAttribute(parametersExtractor: ParametersExtractor): Boolean {
+        return parametersExtractor.extractEnvValue(ENVIRONMENT_ID_RESOURCE_ATTRIBUTE) != null
+    }
+
+
     //actually not relevant because this flavor never calls withOtelAgent
     override fun useOtelAgent(): Boolean {
         return false