🌟 更新 Codec 注解及其处理器

Anvil-Dev · Sep 10, 2024 · 8ed146f · 8ed146f
1 parent fdf43bb
commit 8ed146f
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main/java/dev/anvilcraft/lib/codec/CodecAnnotationProcessor.java b/src/main/java/dev/anvilcraft/lib/codec/CodecAnnotationProcessor.java
@@ -11,6 +11,7 @@
 import javax.lang.model.SourceVersion;
 import javax.lang.model.element.Element;
 import javax.lang.model.element.TypeElement;
+import javax.lang.model.element.VariableElement;
 import javax.tools.Diagnostic;
 import java.util.Set;
 
@@ -25,10 +26,11 @@ public synchronized void init(ProcessingEnvironment processingEnv) {
     @Override
     public boolean process(Set<? extends TypeElement> annotations, @NotNull RoundEnvironment roundEnv) {
         for (Element element : roundEnv.getElementsAnnotatedWith(Codec.class)) {
-            TypeElement typeElem = (TypeElement) element;
-            String typeName = typeElem.getQualifiedName().toString();
-            Filer filer = processingEnv.getFiler();
-            log("Found @Codec annotation on " + typeName);
+            if (element instanceof VariableElement varElement) {
+                String typeName = varElement.getSimpleName().toString();
+                Filer filer = processingEnv.getFiler();
+                log("Found @Codec annotation on " + typeName);
+            }
         }
         return true;
     }