Skip to content

Commit 12af631

Browse files
committed
fix instrumentation flavor
1 parent 33477af commit 12af631

File tree

1 file changed

+14
-9
lines changed

1 file changed

+14
-9
lines changed

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

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import com.intellij.openapi.module.Module
66

77
const val INSTRUMENTATION_FLAVOR_ENV_NAME = "INSTRUMENTATION_FLAVOR"
88

9+
@Suppress("unused", "CanBeParameter", "MemberVisibilityCanBePrivate")
910
open class InstrumentationFlavor(
1011
private val configuration: RunConfigurationBase<*>,
1112
private val projectHeuristics: ProjectHeuristics,
@@ -23,28 +24,28 @@ open class InstrumentationFlavor(
2324

2425
private val resolvedModule: Module? = moduleResolver.resolveModule()
2526

26-
private val flavorInEnv: Flavors = try {
27+
private val flavorInEnv: Flavors? = try {
2728
val flavorValueInEnv = parametersExtractor.extractEnvValue(INSTRUMENTATION_FLAVOR_ENV_NAME)
2829
flavorValueInEnv?.let {
2930
Flavors.valueOf(it)
30-
} ?: Flavors.Default
31-
31+
}
3232
} catch (e: IllegalArgumentException) {
33-
Flavors.Default
33+
null
3434
}
3535

3636
open fun shouldUseOtelAgent(): Boolean {
37-
return listOf(isMicronautTracing(), isSpringBootMicrometerTracing()).none { it }
37+
return isUserConfiguredDefaultInEnv() ||
38+
listOf(isMicronautTracing(), isSpringBootMicrometerTracing()).none { it }
3839
}
3940

4041
open fun isSpringBootMicrometerTracing(): Boolean {
41-
return isUserConfiguredSpringBootMicrometerInEnv() ||
42-
isSpringBootMicrometerTracing(resolvedModule, projectHeuristics)
42+
if (isUserConfiguredDefaultInEnv()) return false
43+
return isUserConfiguredSpringBootMicrometerInEnv() || isSpringBootMicrometerTracing(resolvedModule, projectHeuristics)
4344
}
4445

4546
open fun isMicronautTracing(): Boolean {
46-
return isUserConfiguredMicronautInEnv() ||
47-
isMicronautTracing(resolvedModule, projectHeuristics)
47+
if (isUserConfiguredDefaultInEnv()) return false
48+
return isUserConfiguredMicronautInEnv() || isMicronautTracing(resolvedModule, projectHeuristics)
4849
}
4950

5051
private fun isSpringBootMicrometerTracing(module: Module?, projectHeuristics: ProjectHeuristics): Boolean {
@@ -82,4 +83,8 @@ open class InstrumentationFlavor(
8283
return flavorInEnv == Flavors.Quarkus
8384
}
8485

86+
fun isUserConfiguredDefaultInEnv(): Boolean {
87+
return flavorInEnv == Flavors.Default
88+
}
89+
8590
}

0 commit comments

Comments
 (0)