refactor retrieve annotation data

Author: kx412764776

plugin/main/src/kotlinx/benchmark/gradle/AndroidTasks.kt

@@ -37,13 +37,31 @@ fun Project.unpackAndProcessAar(target: KotlinJvmAndroidCompilation) {
             }
 
             val annotations = annotationProcessor.getClassAnnotations()
-            // Use the annotations data
             annotations.forEach { (className, classAnnotations) ->
-                println("Annotations for class: $className")
-                classAnnotations.forEach { (annotationName, annotationData) ->
-                    println("  - Annotation: $annotationName")
-                    annotationData.parameters.forEach { (paramName, paramValue) ->
-                        println("    - $paramName: $paramValue")
+                println("Annotation for class: $className")
+
+                classAnnotations.classAnnotations.forEach { (annotationDesc, annotationData) ->
+                    println("Class annotation: $annotationDesc")
+                    annotationData.parameters.forEach { (name, value) ->
+                        println("  - $name: $value")
+                    }
+                }
+
+                classAnnotations.methodAnnotations.forEach { (methodName, methodAnnotationMap) ->
+                    methodAnnotationMap.forEach { (annotationDesc, annotationData) ->
+                        println("Method annotation in $methodName: $annotationDesc")
+                        annotationData.parameters.forEach { (name, value) ->
+                            println("  - $name: $value")
+                        }
+                    }
+                }
+
+                classAnnotations.fieldAnnotations.forEach { (fieldName, fieldAnnotationMap) ->
+                    fieldAnnotationMap.forEach { (annotationDesc, annotationData) ->
+                        println("Field annotation in $fieldName: $annotationDesc")
+                        annotationData.parameters.forEach { (name, value) ->
+                            println("  - $name: $value")
+                        }
                     }
                 }
             }

plugin/main/src/kotlinx/benchmark/gradle/AnnotationProcessor.kt

@@ -8,29 +8,37 @@ data class AnnotationData(
     val parameters: Map<String, Any?>
 )
 
+data class ClassAnnotations(
+    val classAnnotations: Map<String, AnnotationData>,
+    val methodAnnotations: Map<String, Map<String, AnnotationData>>,
+    val fieldAnnotations: Map<String, Map<String, AnnotationData>>
+)
+
 class AnnotationProcessor {
 
-    private val classAnnotations = mutableMapOf<String, MutableMap<String, AnnotationData>>()
+    private val classAnnotationsMap = mutableMapOf<String, ClassAnnotations>()
 
     fun processClassFile(classFile: File) {
         val classReader = ClassReader(classFile.readBytes())
         val classNode = ClassNode()
         classReader.accept(classNode, 0)
 
-        val annotations = mutableMapOf<String, AnnotationData>()
+        val classAnnotations = mutableMapOf<String, AnnotationData>()
+        val methodAnnotations = mutableMapOf<String, MutableMap<String, AnnotationData>>()
+        val fieldAnnotations = mutableMapOf<String, MutableMap<String, AnnotationData>>()
 
         classNode.visibleAnnotations?.forEach { annotationNode ->
             if (annotationNode.desc != "Lkotlin/Metadata;") {
                 val annotationData = parseAnnotation(annotationNode)
-                annotations[annotationNode.desc] = annotationData
+                classAnnotations[annotationNode.desc] = annotationData
             }
         }
 
         classNode.methods?.forEach { methodNode ->
             methodNode.visibleAnnotations?.forEach { annotationNode ->
                 if (annotationNode.desc != "Lkotlin/Metadata;") {
                     val annotationData = parseAnnotation(annotationNode)
-                    annotations[annotationNode.desc] = annotationData
+                    methodAnnotations.getOrPut(methodNode.name) { mutableMapOf() }[annotationNode.desc] = annotationData
                 }
             }
         }
@@ -39,12 +47,12 @@ class AnnotationProcessor {
             fieldNode.visibleAnnotations?.forEach { annotationNode ->
                 if (annotationNode.desc != "Lkotlin/Metadata;") {
                     val annotationData = parseAnnotation(annotationNode)
-                    annotations[annotationNode.desc] = annotationData
+                    fieldAnnotations.getOrPut(fieldNode.name) { mutableMapOf() }[annotationNode.desc] = annotationData
                 }
             }
         }
 
-        classAnnotations[classNode.name] = annotations
+        classAnnotationsMap[classNode.name] = ClassAnnotations(classAnnotations, methodAnnotations, fieldAnnotations)
     }
 
     private fun parseAnnotation(annotationNode: AnnotationNode): AnnotationData {
@@ -95,7 +103,7 @@ class AnnotationProcessor {
         return sb.toString()
     }
 
-    fun getClassAnnotations(): Map<String, Map<String, AnnotationData>> {
-        return classAnnotations
+    fun getClassAnnotations(): Map<String, ClassAnnotations> {
+        return classAnnotationsMap
     }
 }