Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
package datadog.gradle.plugin.config

import org.gradle.api.DefaultTask
import org.gradle.kotlin.dsl.property
import org.gradle.api.model.ObjectFactory
import org.gradle.api.tasks.Input
import org.gradle.api.tasks.InputFile
Expand All @@ -12,6 +11,7 @@ import com.fasterxml.jackson.databind.ObjectMapper
import org.gradle.api.tasks.CacheableTask
import org.gradle.api.tasks.PathSensitive
import org.gradle.api.tasks.PathSensitivity
import org.gradle.kotlin.dsl.property
import java.io.File
import java.io.FileInputStream
import java.io.PrintWriter
Expand Down Expand Up @@ -131,11 +131,29 @@ abstract class ParseV2SupportedConfigurationsTask @Inject constructor(
out.println(" public static final Map<String, String> REVERSE_PROPERTY_KEYS_MAP;")
out.println()
out.println(" static {")
out.println(" SUPPORTED = initSupported();")
out.println(" ALIASES = initAliases();")
out.println(" ALIAS_MAPPING = initAliasMapping();")
out.println(" DEPRECATED = initDeprecated();")
out.println(" REVERSE_PROPERTY_KEYS_MAP = initReversePropertyKeysMap();")
out.println(" }")
out.println()

// SUPPORTED
// initSupported() - split into two helper functions to avoid "code too large" error
out.println(" private static Map<String, List<SupportedConfiguration>> initSupported() {")
out.println(" Map<String, List<SupportedConfiguration>> supportedMap = new HashMap<>();")
for ((key, configList) in supported.toSortedMap()) {
out.println(" initSupported1(supportedMap);")
out.println(" initSupported2(supportedMap);")
out.println(" return Collections.unmodifiableMap(supportedMap);")
out.println(" }")
out.println()

val sortedSupported = supported.toSortedMap().entries.toList()
val midpoint = sortedSupported.size / 2

// initSupported1() - first half
out.println(" private static void initSupported1(Map<String, List<SupportedConfiguration>> supportedMap) {")
for ((key, configList) in sortedSupported.take(midpoint)) {
out.print(" supportedMap.put(\"${esc(key)}\", Collections.unmodifiableList(Arrays.asList(")
val configIter = configList.iterator()
while (configIter.hasNext()) {
Expand All @@ -151,11 +169,33 @@ abstract class ParseV2SupportedConfigurationsTask @Inject constructor(
}
out.println(")));")
}
out.println(" SUPPORTED = Collections.unmodifiableMap(supportedMap);")
out.println(" }")
out.println()

// ALIASES
out.println(" // Note: This top-level alias mapping will be deprecated once Config Registry is mature enough to understand which version of a config a customer is using")
// initSupported2() - second half
out.println(" private static void initSupported2(Map<String, List<SupportedConfiguration>> supportedMap) {")
for ((key, configList) in sortedSupported.drop(midpoint)) {
out.print(" supportedMap.put(\"${esc(key)}\", Collections.unmodifiableList(Arrays.asList(")
val configIter = configList.iterator()
while (configIter.hasNext()) {
val config = configIter.next()
out.print("new SupportedConfiguration(")
out.print("${escNullableString(config.version)}, ")
out.print("${escNullableString(config.type)}, ")
out.print("${escNullableString(config.default)}, ")
out.print("Arrays.asList(${quoteList(config.aliases)}), ")
out.print("Arrays.asList(${quoteList(config.propertyKeys)})")
out.print(")")
if (configIter.hasNext()) out.print(", ")
}
out.println(")));")
}
out.println(" }")
out.println()

// initAliases()
out.println(" // Note: This top-level alias mapping will be deprecated once Config Registry is mature enough to understand which version of a config a customer is using")
out.println(" private static Map<String, List<String>> initAliases() {")
out.println(" Map<String, List<String>> aliasesMap = new HashMap<>();")
for ((canonical, list) in aliases.toSortedMap()) {
out.printf(
Expand All @@ -164,33 +204,37 @@ abstract class ParseV2SupportedConfigurationsTask @Inject constructor(
quoteList(list)
)
}
out.println(" ALIASES = Collections.unmodifiableMap(aliasesMap);")
out.println(" return Collections.unmodifiableMap(aliasesMap);")
out.println(" }")
out.println()

// ALIAS_MAPPING
// initAliasMapping()
out.println(" private static Map<String, String> initAliasMapping() {")
out.println(" Map<String, String> aliasMappingMap = new HashMap<>();")
for ((alias, target) in aliasMapping.toSortedMap()) {
out.printf(" aliasMappingMap.put(\"%s\", \"%s\");\n", esc(alias), esc(target))
}
out.println(" ALIAS_MAPPING = Collections.unmodifiableMap(aliasMappingMap);")
out.println(" return Collections.unmodifiableMap(aliasMappingMap);")
out.println(" }")
out.println()

// DEPRECATED
// initDeprecated()
out.println(" private static Map<String, String> initDeprecated() {")
out.println(" Map<String, String> deprecatedMap = new HashMap<>();")
for ((oldKey, note) in deprecated.toSortedMap()) {
out.printf(" deprecatedMap.put(\"%s\", \"%s\");\n", esc(oldKey), esc(note))
}
out.println(" DEPRECATED = Collections.unmodifiableMap(deprecatedMap);")
out.println(" return Collections.unmodifiableMap(deprecatedMap);")
out.println(" }")
out.println()

// REVERSE_PROPERTY_KEYS_MAP
// initReversePropertyKeysMap()
out.println(" private static Map<String, String> initReversePropertyKeysMap() {")
out.println(" Map<String, String> reversePropertyKeysMapping = new HashMap<>();")
for ((propertyKey, config) in reversePropertyKeysMap.toSortedMap()) {
out.printf(" reversePropertyKeysMapping.put(\"%s\", \"%s\");\n", esc(propertyKey), esc(config))
}
out.println(" REVERSE_PROPERTY_KEYS_MAP = Collections.unmodifiableMap(reversePropertyKeysMapping);")
out.println()

out.println(" return Collections.unmodifiableMap(reversePropertyKeysMapping);")
out.println(" }")
out.println("}")
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ class SupportedConfigPlugin : Plugin<Project> {

private fun generateSupportedConfigurations(targetProject: Project, extension: SupportedTracerConfigurations) {
val generateTask =
targetProject.tasks.register("generateSupportedConfigurations", ParseSupportedConfigurationsTask::class.java) {
targetProject.tasks.register("generateSupportedConfigurations", ParseV2SupportedConfigurationsTask::class.java) {
jsonFile.set(extension.jsonFile)
destinationDirectory.set(extension.destinationDirectory)
className.set(extension.className)
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,199 @@
package datadog.gradle.plugin.config

import org.gradle.testkit.runner.BuildResult
import org.gradle.testkit.runner.GradleRunner
import org.gradle.testkit.runner.TaskOutcome
import org.junit.jupiter.api.Assertions.assertEquals
import org.junit.jupiter.api.Assertions.assertTrue
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.io.TempDir
import java.io.File
import java.nio.file.Paths

class ParseV2SupportedConfigurationsTest {
@Test
fun `should generate Java file from JSON configuration`(@TempDir projectDir: File) {
val (buildResult, generatedFile) = runGradleTask(projectDir)

assertEquals(TaskOutcome.SUCCESS, buildResult.task(":generateSupportedConfigurations")?.outcome)

assertTrue(generatedFile.exists(), "Generated Java file should exist")

val content = generatedFile.readText()
assertTrue(content.contains("package datadog.test;"))
assertTrue(content.contains("public final class TestGeneratedSupportedConfigurations {"))

assertTrue(content.contains("public static final Map<String, List<SupportedConfiguration>> SUPPORTED;"))
assertTrue(content.contains("public static final Map<String, List<String>> ALIASES;"))
assertTrue(content.contains("public static final Map<String, String> ALIAS_MAPPING;"))
assertTrue(content.contains("public static final Map<String, String> DEPRECATED;"))
assertTrue(content.contains("public static final Map<String, String> REVERSE_PROPERTY_KEYS_MAP;"))

assertTrue(content.contains("private static Map<String, List<SupportedConfiguration>> initSupported()"))
assertTrue(content.contains("private static void initSupported1(Map<String, List<SupportedConfiguration>> supportedMap)"))
assertTrue(content.contains("private static void initSupported2(Map<String, List<SupportedConfiguration>> supportedMap)"))
assertTrue(content.contains("private static Map<String, List<String>> initAliases()"))
assertTrue(content.contains("private static Map<String, String> initAliasMapping()"))
assertTrue(content.contains("private static Map<String, String> initDeprecated()"))
assertTrue(content.contains("private static Map<String, String> initReversePropertyKeysMap()"))

assertContainsSupportedConfig(
content,
key = "DD_ACTION_EXECUTION_ID",
version = "A",
type = "string",
default = "null",
aliases = emptyList(),
propertyKeys = listOf("property.key")
)

assertContainsSupportedConfig(
content,
key = "DD_AGENTLESS_LOG_SUBMISSION_ENABLED",
version = "A",
type = "boolean",
default = "false",
aliases = emptyList()
)

assertContainsSupportedConfig(
content,
key = "DD_AGENTLESS_LOG_SUBMISSION_ENABLED",
version = "B",
type = "boolean",
default = "true",
aliases = listOf("DD_ALIAS")
)

assertTrue(content.contains("""aliasesMap.put("DD_ACTION_EXECUTION_ID", Collections.unmodifiableList(Arrays.asList()))"""))
assertTrue(content.contains("""aliasesMap.put("DD_AGENTLESS_LOG_SUBMISSION_ENABLED", Collections.unmodifiableList(Arrays.asList("DD_ALIAS")))"""))

assertTrue(content.contains("""aliasMappingMap.put("DD_ALIAS", "DD_AGENTLESS_LOG_SUBMISSION_ENABLED")"""))

assertTrue(content.contains("""deprecatedMap.put("old.config", "Use test.config instead")"""))
assertTrue(content.contains("""deprecatedMap.put("legacy.setting", "No longer supported")"""))

assertTrue(content.contains("""reversePropertyKeysMapping.put("property.key", "DD_ACTION_EXECUTION_ID")"""))
}

private fun runGradleTask(projectDir: File): Pair<BuildResult, File> {
val jsonFile = file(projectDir, "test-supported-configurations.json")
jsonFile.writeText(
"""
{
"supportedConfigurations": {
"DD_ACTION_EXECUTION_ID": [
{
"version": "A",
"type": "string",
"default": null,
"aliases": [],
"propertyKeys": ["property.key"]
}
],
"DD_AGENTLESS_LOG_SUBMISSION_ENABLED": [
{
"version": "A",
"type": "boolean",
"default": "false",
"aliases": []
},
{
"version": "B",
"type": "boolean",
"default": "true",
"aliases": ["DD_ALIAS"]
}
]
},
"deprecations": {
"old.config": "Use test.config instead",
"legacy.setting": "No longer supported"
}
}
""".trimIndent()
)

setupGradleProject(projectDir)

val buildResult = GradleRunner.create()
.forwardOutput()
.withPluginClasspath()
.withArguments("generateSupportedConfigurations")
.withProjectDir(projectDir)
.build()

val generatedFile = file(projectDir, "build", "generated", "supportedConfigurations", "datadog", "test", "TestGeneratedSupportedConfigurations.java")
return Pair(buildResult, generatedFile)
}

private fun setupGradleProject(projectDir: File) {
file(projectDir, "settings.gradle.kts").writeText(
"""
rootProject.name = "test-config-project"
""".trimIndent()
)

file(projectDir, "build.gradle.kts").writeText(
"""
plugins {
id("java")
id("dd-trace-java.supported-config-generator")
}

group = "datadog.config.test"

supportedTracerConfigurations {
jsonFile.set(file("test-supported-configurations.json"))
destinationDirectory.set(file("build/generated/supportedConfigurations"))
className.set("datadog.test.TestGeneratedSupportedConfigurations")
}
""".trimIndent()
)
}

private fun file(projectDir: File, vararg parts: String, makeDirectory: Boolean = false): File {
val f = Paths.get(projectDir.absolutePath, *parts).toFile()

if (makeDirectory) {
f.parentFile.mkdirs()
}

return f
}

private fun assertContainsSupportedConfig(
content: String,
key: String,
version: String,
type: String,
default: String,
aliases: List<String>,
propertyKeys: List<String> = emptyList()
) {
val aliasesArray = aliases.joinToString(", ") { "\"$it\"" }
val propertyKeysArray = propertyKeys.joinToString(", ") { "\"$it\"" }

assertTrue(
content.contains("""supportedMap.put("$key""""),
"Should contain supportedMap.put for key: $key"
)

val expectedPattern = buildString {
append("new SupportedConfiguration(")
append("\"$version\", ")
append("\"$type\", ")
append(if (default == "null") "null" else "\"$default\"")
append(", ")
append("Arrays.asList($aliasesArray)")
append(", ")
append("Arrays.asList($propertyKeysArray)")
append(")")
}

assertTrue(
content.contains(expectedPattern),
"Should contain SupportedConfiguration: $expectedPattern"
)
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,8 @@ public static void onEnter(@Advice.Argument(value = 0, readOnly = false) String[
+ "datadog.trace.bootstrap.WeakMapContextStore:build_time,"
+ "datadog.trace.config.inversion.ConfigHelper:rerun,"
+ "datadog.trace.config.inversion.ConfigHelper$StrictnessPolicy:rerun,"
+ "datadog.trace.config.inversion.GeneratedSupportedConfigurations:rerun,"
+ "datadog.trace.config.inversion.GeneratedSupportedConfigurations:build_time,"
+ "datadog.trace.config.inversion.SupportedConfiguration:build_time,"
+ "datadog.trace.instrumentation.guava10.GuavaAsyncResultExtension:build_time,"
+ "datadog.trace.instrumentation.reactivestreams.ReactiveStreamsAsyncResultExtension:build_time,"
+ "datadog.trace.instrumentation.reactor.core.ReactorAsyncResultExtension:build_time,"
Expand Down
Loading