ported tools/stats.gradle (#3219)

* ported stats and removed redundant `project` receiver in release.gradle.kts

---------

Co-authored-by: Frank Liu <[email protected]>

build.gradle.kts

@@ -1,6 +1,5 @@
 plugins {
-    id ("com.github.spotbugs") apply false
+    id("com.github.spotbugs") apply false
     ai.djl.release
+    ai.djl.stats
 }
-
-//apply from: file("${rootProject.projectDir}/tools/gradle/stats.gradle")

buildSrc/src/main/kotlin/ai/djl/javaProject.gradle.kts

@@ -11,6 +11,7 @@ plugins {
     `java-library`
     id("ai.djl.javaFormatter")
     id("ai.djl.check")
+    id("ai.djl.stats")
 }
 
 tasks {

buildSrc/src/main/kotlin/ai/djl/release.gradle.kts

@@ -7,7 +7,7 @@ import text
 
 val libs = the<LibrariesForLibs>()
 
-project.tasks {
+tasks {
     register("increaseBuildVersion") {
         doLast {
             if (!project.hasProperty("targetVersion")) {

buildSrc/src/main/kotlin/ai/djl/stats.gradle.kts

@@ -0,0 +1,66 @@
+package ai.djl
+
+import org.gradle.api.plugins.ExtraPropertiesExtension
+import org.gradle.kotlin.dsl.support.serviceOf
+import org.gradle.tooling.events.FinishEvent
+import org.gradle.tooling.events.OperationCompletionListener
+import kotlin.time.Duration
+import kotlin.time.Duration.Companion.seconds
+import kotlin.time.ExperimentalTime
+import kotlin.time.TimeSource
+
+plugins {
+    `java-library`
+}
+
+@ExperimentalTime
+val timeSource = TimeSource.Monotonic
+val testsResults = mutableMapOf<Duration, String>()
+
+val demoListener = gradle.sharedServices.registerIfAbsent("demoListener ", StatisticsService::class) {
+    parameters.testsResults = testsResults
+}
+
+gradle.taskGraph.whenReady {
+    gradle.serviceOf<BuildEventsListenerRegistry>().onTaskCompletion(demoListener)
+}
+
+abstract class StatisticsService : BuildService<StatisticsService.Parameters>,
+    OperationCompletionListener, AutoCloseable {
+
+    interface Parameters : BuildServiceParameters {
+        var testsResults: MutableMap<Duration, String>
+    }
+
+    override fun onFinish(event: FinishEvent) {}
+
+    override fun close() {
+        if (parameters.testsResults.isNotEmpty()) {
+            println("========== Test duration ========== ")
+            for ((key, value) in parameters.testsResults.entries.sortedByDescending { it.key }.take(6)) {
+                // `.inWholeSeconds.seconds` truncate to integer units, without decimals
+                println("\t$value:\t${key.inWholeSeconds.seconds}")
+            }
+        }
+    }
+}
+
+tasks.test {
+    doFirst {
+        @OptIn(ExperimentalTime::class)
+        startTime = timeSource.markNow()
+    }
+    doLast {
+        @OptIn(ExperimentalTime::class)
+        if (state.didWork)
+            demoListener.get().parameters.testsResults[timeSource.markNow() - startTime] = project.name
+    }
+}
+
+@ExperimentalTime
+var Task.startTime: TimeSource.Monotonic.ValueTimeMark
+    get() = ext.get("startTime") as TimeSource.Monotonic.ValueTimeMark
+    set(value) = ext.set("startTime", value)
+
+val Task.ext: ExtraPropertiesExtension
+    get() = (this as ExtensionAware).extensions.getByName<ExtraPropertiesExtension>("ext")

settings.gradle.kts

@@ -1,7 +1,7 @@
 rootProject.name = "djl"
 
 plugins {
-    id("org.gradle.toolchains.foojay-resolver-convention") version("0.8.0")
+    id("org.gradle.toolchains.foojay-resolver-convention") version ("0.8.0")
 }
 
 include(":api")
@@ -45,6 +45,7 @@ if (JavaVersion.current() < JavaVersion.VERSION_21) {
 }
 
 dependencyResolutionManagement {
+    @Suppress("UnstableApiUsage")
     repositories {
         mavenCentral()
         maven("https://oss.sonatype.org/content/repositories/snapshots/")