wip

Author: zeitlinger

instrumentation/spring/spring-boot-autoconfigure/build.gradle.kts

@@ -6,7 +6,8 @@ plugins {
 base.archivesName.set("opentelemetry-spring-boot-autoconfigure")
 group = "io.opentelemetry.instrumentation"
 
-val springBootVersion = "2.7.18" // AutoConfiguration is added in 2.7.0, but can be used with older versions
+val springBootVersion =
+  "2.7.18" // AutoConfiguration is added in 2.7.0, but can be used with older versions
 
 // r2dbc-proxy is shadowed to prevent org.springframework.boot.autoconfigure.r2dbc.R2dbcAutoConfiguration
 // from being loaded by Spring Boot (by the presence of META-INF/services/io.r2dbc.spi.ConnectionFactoryProvider) - even if the user doesn't want to use R2DBC.
@@ -41,7 +42,12 @@ dependencies {
   implementation(project(":instrumentation-annotations-support"))
   implementation(project(":instrumentation:kafka:kafka-clients:kafka-clients-2.6:library"))
   implementation(project(":instrumentation:mongo:mongo-3.1:library"))
-  compileOnly(project(path = ":instrumentation:r2dbc-1.0:library-instrumentation-shaded", configuration = "shadow"))
+  compileOnly(
+    project(
+      path = ":instrumentation:r2dbc-1.0:library-instrumentation-shaded",
+      configuration = "shadow"
+    )
+  )
   implementation(project(":instrumentation:spring:spring-kafka-2.7:library"))
   implementation(project(":instrumentation:spring:spring-web:spring-web-3.1:library"))
   implementation(project(":instrumentation:spring:spring-webmvc:spring-webmvc-5.3:library"))
@@ -107,8 +113,14 @@ dependencies {
   add("javaSpring3CompileOnly", files(sourceSets.main.get().output.classesDirs))
   add("javaSpring3CompileOnly", "org.springframework.boot:spring-boot-starter-web:3.2.4")
   add("javaSpring3CompileOnly", "io.opentelemetry:opentelemetry-sdk-extension-autoconfigure")
-  add("javaSpring3CompileOnly", project(":instrumentation:spring:spring-web:spring-web-3.1:library"))
-  add("javaSpring3CompileOnly", project(":instrumentation:spring:spring-webmvc:spring-webmvc-6.0:library"))
+  add(
+    "javaSpring3CompileOnly",
+    project(":instrumentation:spring:spring-web:spring-web-3.1:library")
+  )
+  add(
+    "javaSpring3CompileOnly",
+    project(":instrumentation:spring:spring-webmvc:spring-webmvc-6.0:library")
+  )
 
   // tests don't work with spring boot 4 yet
   latestDepTestLibrary("org.springframework.boot:spring-boot-starter-test:3.+") // documented limitation
@@ -130,8 +142,10 @@ if (latestDepTest) {
   }
 }
 
-val testJavaVersion = gradle.startParameter.projectProperties["testJavaVersion"]?.let(JavaVersion::toVersion)
-val testSpring3 = (testJavaVersion == null || testJavaVersion.compareTo(JavaVersion.VERSION_17) >= 0)
+val testJavaVersion =
+  gradle.startParameter.projectProperties["testJavaVersion"]?.let(JavaVersion::toVersion)
+val testSpring3 =
+  (testJavaVersion == null || testJavaVersion.compareTo(JavaVersion.VERSION_17) >= 0)
 
 testing {
   suites {
@@ -205,7 +219,80 @@ configurations.configureEach {
   }
 }
 
+val buildGraalVmReflectionJson = tasks.register("buildGraalVmReflectionJson") {
+  val targetFile = File(
+    projectDir,
+    "src/main/resources/META-INF/native-image/io.opentelemetry.instrumentation/opentelemetry-spring-boot/reflect-config.json"
+  )
+
+  onlyIf { !targetFile.exists() }
+
+  doLast {
+    val sourcePackage =
+      "io.opentelemetry.sdk.extension.incubator.fileconfig.internal.model"
+    val sourcePackagePath  = sourcePackage.replace(".", "/")
+
+    val cp = configurations.compileClasspath.get()
+
+    println("Scanning classpath for OpenTelemetry Incubator SDK extension jar...")
+    println(cp.asPath)
+
+    val incubatorJar = cp
+      .filter { it.name.contains("opentelemetry-sdk-extension-incubator") && it.name.endsWith(".jar") }
+      .singleFile
+
+    println("Found incubator jar: $incubatorJar")
+
+    val classes = mutableListOf<String>()
+
+    // list contents of incubatorJar in gradle cache
+
+    zipTree(incubatorJar).matching {
+      include("$sourcePackagePath/**")
+    }.forEach {
+      val path = it.path
+
+      println(path)
+
+      val className = path
+        .substringAfter(sourcePackagePath)
+        .removePrefix("/")
+        .removeSuffix(".class")
+        .replace("/", ".")
+      classes.add("$sourcePackage.$className")
+    }
+
+    // todo remove output to console
+    println("Discovered ${classes.size} classes for reflection:")
+    println(classes)
+
+    // write into targetFile in json format
+    targetFile.parentFile.mkdirs()
+    targetFile.bufferedWriter().use { writer ->
+      writer.write("[\n")
+      classes.forEachIndexed { index, className ->
+        writer.write("  {\n")
+        writer.write("    \"name\": \"$className\",\n")
+        writer.write("    \"allDeclaredMethods\": true,\n")
+        writer.write("    \"allDeclaredFields\": true,\n")
+        writer.write("    \"allDeclaredConstructors\": true\n")
+        writer.write("  }")
+        if (index < classes.size - 1) {
+          writer.write(",\n")
+        } else {
+          writer.write("\n")
+        }
+      }
+      writer.write("]\n")
+    }
+  }
+}
+
 tasks {
+  compileJava {
+    dependsOn(buildGraalVmReflectionJson)
+  }
+
   compileTestJava {
     options.compilerArgs.add("-parameters")
   }