Support configuring the Reporter class name

Author: Vinay

src/main/kotlin/org/simple/rmg/RoomMetadataGenerator.kt

@@ -18,7 +18,7 @@ import java.io.File
 import java.nio.file.Paths
 
 fun main(args: Array<String>) {
-	RoomMetadataGenerator().run(args[0], args[1])
+	RoomMetadataGenerator().run(args[0], args[1], args[2])
 }
 
 private val rxJavaTypes = setOf("Single", "Completable", "Observable", "Flowable")
@@ -44,7 +44,8 @@ class RoomMetadataGenerator {
 
 	fun run(
 		projectPath: String,
-		sourceSet: String
+		sourceSet: String,
+		reporterName: String
 	) {
 		val moduleGeneratedSourcesDirectory = Paths.get(projectPath, "build", "generated", "source", "kapt", sourceSet)
 
@@ -62,7 +63,7 @@ class RoomMetadataGenerator {
 
 		val transformedAsts = generatedRoomDaoAsts
 			.map { daoMetadata ->
-				val transformedAst = transformGeneratedDao(measureMethodCodeTemplate, daoMetadata.ast.clone())
+				val transformedAst = transformGeneratedDao(measureMethodCodeTemplate, daoMetadata.ast.clone(), reporterName)
 				daoMetadata.replaceAst(transformedAst)
 			}
 
@@ -73,11 +74,16 @@ class RoomMetadataGenerator {
 
 	fun transformGeneratedDao(
 		measureMethodCodeTemplate: String,
-		generatedDao: CompilationUnit
+		generatedDao: CompilationUnit,
+		reporterName: String
 	): CompilationUnit {
 		val classDeclaration = generatedDao.types.first() as ClassOrInterfaceDeclaration
 		val measureMethod = StaticJavaParser
-			.parse(measureMethodCodeTemplate.replace("\$CLASS_NAME\$", classDeclaration.nameAsString))
+			.parse(
+				measureMethodCodeTemplate
+					.replace("\$CLASS_NAME\$", classDeclaration.nameAsString)
+					.replace("\$REPORTER_NAME\$", reporterName)
+			)
 			.types
 			.first()
 			.methods

src/main/resources/MeasureMethod.java

@@ -11,7 +11,7 @@ private static <T> T measureAndReport(
         result = block.invoke();
         final java.time.Duration timeTaken = java.time.Duration.ofMillis(System.currentTimeMillis() - start);
 
-        org.simple.clinic.SqlPerformanceReporter.report("$CLASS_NAME$", methodName, timeTaken);
+        $REPORTER_NAME$.report("$CLASS_NAME$", methodName, timeTaken);
 
         return result;
     }

src/test/kotlin/org/simple/rmg/DaoTransformationTest.kt

@@ -3,7 +3,6 @@ package org.simple.rmg
 import com.github.javaparser.StaticJavaParser
 import com.github.javaparser.ast.CompilationUnit
 import com.google.common.truth.Truth.assertThat
-import org.junit.jupiter.api.Assertions
 import org.junit.jupiter.api.Assertions.assertThrows
 import org.junit.jupiter.api.Test
 
@@ -13,6 +12,8 @@ class DaoTransformationTest {
 
 	private val measureMethodCodeTemplate: String
 
+	private val reporterName = "org.simple.clinic.SqlPerformanceReporter"
+
 	init {
 		measureMethodCodeTemplate = readResourceAsString("MeasureMethod.java")
 	}
@@ -23,7 +24,7 @@ class DaoTransformationTest {
 		val daoImplementation = readResourceAsAst("test_data/transform/source/UserRoomDao_Impl.java")
 
 		// when
-		val transformedDao = app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation)
+		val transformedDao = app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation, reporterName)
 
 		// then
 		val expectedTransformedDao = readResourceAsAst("test_data/transform/modified/UserRoomDao_Impl.java")
@@ -37,7 +38,7 @@ class DaoTransformationTest {
 		val daoImplementation = readResourceAsAst("test_data/transform/source/BloodPressureMeasurementRoomDao_Impl.java")
 
 		// when
-		val transformedDao = app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation)
+		val transformedDao = app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation, reporterName)
 
 		// then
 		val expectedTransformedDao = readResourceAsAst("test_data/transform/modified/BloodPressureMeasurementRoomDao_Impl.java")
@@ -51,7 +52,7 @@ class DaoTransformationTest {
 		val daoImplementation = readResourceAsAst("test_data/transform/modified/UserRoomDao_Impl.java")
 
 		// when
-		val transformedDao = app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation)
+		val transformedDao = app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation, reporterName)
 
 		// then
 		val expectedTransformedDao = readResourceAsAst("test_data/transform/modified/UserRoomDao_Impl.java")
@@ -66,7 +67,7 @@ class DaoTransformationTest {
 
 		// then
 		assertThrows(RuntimeException::class.java) {
-			app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation)
+			app.transformGeneratedDao(measureMethodCodeTemplate, daoImplementation, reporterName)
 		}
 	}