From ddbbda532ce64746cc11894d9754443487e71b7d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=80=E6=BA=90=E6=B5=B7=E5=93=A5?= Date: Wed, 15 Mar 2023 13:51:17 +0800 Subject: [PATCH] fixed QueryEntityProcesser.java can not get annotation value --- .../processer/QueryEntityProcesser.java | 24 +++++++++++++++---- 1 file changed, 19 insertions(+), 5 deletions(-) diff --git a/mybatis-flex-annotation/src/main/java/com/mybatisflex/processer/QueryEntityProcesser.java b/mybatis-flex-annotation/src/main/java/com/mybatisflex/processer/QueryEntityProcesser.java index f51ef485..62b19236 100644 --- a/mybatis-flex-annotation/src/main/java/com/mybatisflex/processer/QueryEntityProcesser.java +++ b/mybatis-flex-annotation/src/main/java/com/mybatisflex/processer/QueryEntityProcesser.java @@ -25,9 +25,7 @@ import javax.annotation.processing.Filer; import javax.annotation.processing.ProcessingEnvironment; import javax.annotation.processing.RoundEnvironment; import javax.lang.model.SourceVersion; -import javax.lang.model.element.Element; -import javax.lang.model.element.ElementKind; -import javax.lang.model.element.TypeElement; +import javax.lang.model.element.*; import javax.lang.model.type.TypeMirror; import javax.lang.model.util.Elements; import javax.lang.model.util.Types; @@ -41,6 +39,7 @@ import java.sql.Timestamp; import java.time.*; import java.time.chrono.JapaneseDate; import java.util.*; +import java.util.function.BiConsumer; import java.util.function.Consumer; public class QueryEntityProcesser extends AbstractProcessor { @@ -159,8 +158,20 @@ public class QueryEntityProcesser extends AbstractProcessor { continue; } + //获取 typeHandlerClass 的名称，通过 column.typeHandler() 获取会抛出异常：MirroredTypeException: + //参考 https://stackoverflow.com/questions/7687829/java-6-annotation-processing-getting-a-class-from-an-annotation + final String[] typeHandlerClass = {""}; + List annotationMirrors = fieldElement.getAnnotationMirrors(); + for (AnnotationMirror annotationMirror : annotationMirrors) { + annotationMirror.getElementValues().forEach((BiConsumer) (executableElement, annotationValue) -> { + if (executableElement.getSimpleName().equals("typeHandler")){ + typeHandlerClass[0] = annotationValue.toString(); + } + }); + } + //未配置 typeHandler 的情况下，只支持基本数据类型，不支持比如 list set 或者自定义的类等 - if ((column == null || column.typeHandler() == UnknownTypeHandler.class) + if ((column == null || typeHandlerClass[0].equals(UnknownTypeHandler.class.getName())) && !defaultSupportColumnTypes.contains(typeMirror.toString())) { continue; } @@ -186,10 +197,13 @@ public class QueryEntityProcesser extends AbstractProcessor { } - return false; } + + + + private String buildClass(String entityClass, String tableName, Map propertyAndColumns, List defaultColumns) { // tableDefTemplate = " public static final @entityClassTableDef @tableField = new @entityClassTableDef(\"@tableName\");\n";