Add JavaTypeVisitor#visitAnnotation()

openrewrite · Jan 23, 2025 · cf83f5e · cf83f5e
1 parent 474aae3
commit cf83f5e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/rewrite-java/src/main/java/org/openrewrite/java/JavaTypeVisitor.java b/rewrite-java/src/main/java/org/openrewrite/java/JavaTypeVisitor.java
@@ -69,6 +69,8 @@ public JavaType visitNonNull(JavaType javaType, P p) {
 
             if (javaType instanceof JavaType.Array) {
                 javaType = visitArray((JavaType.Array) javaType, p);
+            } else if (javaType instanceof JavaType.Annotation) {
+                javaType = visitAnnotation((JavaType.Annotation) javaType, p);
             } else if (javaType instanceof JavaType.Class) {
                 javaType = visitClass((JavaType.Class) javaType, p);
             } else if (javaType instanceof JavaType.GenericTypeVariable) {
@@ -105,6 +107,12 @@ public JavaType visitMultiCatch(JavaType.MultiCatch multiCatch, P p) {
         return multiCatch.withThrowableTypes(ListUtils.map(multiCatch.getThrowableTypes(), tt -> visit(tt, p)));
     }
 
+    public JavaType visitAnnotation(JavaType.Annotation annotation, P p) {
+        JavaType.Annotation a = annotation;
+        a = a.withType((JavaType.FullyQualified) visit(a.getType(), p));
+        return a;
+    }
+
     public JavaType visitArray(JavaType.Array array, P p) {
         JavaType.Array a = array;
         a = a.withElemType(visit(a.getElemType(), p));