admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-08 01:28:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

close #I6ZPD1

Browse Source
This commit is contained in:
开源海哥 2023-04-28 19:49:57 +08:00
parent a86551633c
commit 61eb072a14
1 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java
View File

@ -29,6 +29,7 @@ import javax.lang.model.element.*;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.util.Elements;
import javax.lang.model.util.Types;
import javax.tools.Diagnostic;
import javax.tools.JavaFileObject;
@ -102,14 +103,14 @@ public class QueryEntityProcessor extends AbstractProcessor {
private static final String allColumnsTemplate = " public QueryColumn[] ALL_COLUMNS = new QueryColumn[]{@allColumns};\n\n";
private Filer filer;
// private Elements elementUtils;
private Elements elementUtils;
private Types typeUtils;
@Override
public synchronized void init(ProcessingEnvironment processingEnvironment) {
super.init(processingEnvironment);
this.filer = processingEnvironment.getFiler();
// this.elementUtils = processingEnvironment.getElementUtils();
this.elementUtils = processingEnvironment.getElementUtils();
this.typeUtils = processingEnvironment.getTypeUtils();
}
@ -216,8 +217,6 @@ public class QueryEntityProcessor extends AbstractProcessor {
continue;
}
TypeMirror typeMirror = fieldElement.asType();
Column column = fieldElement.getAnnotation(Column.class);
if (column != null && column.ignore()) {
continue;
@ -236,18 +235,17 @@ public class QueryEntityProcessor extends AbstractProcessor {
});
}
TypeElement typeElement = null;
if (typeMirror.getKind() == TypeKind.DECLARED) {
typeElement = (TypeElement) ((DeclaredType) typeMirror).asElement();
TypeMirror typeMirror = fieldElement.asType();
Element element = typeUtils.asElement(typeMirror);
if (element != null) {
typeMirror = element.asType();
}
String typeString = typeMirror.toString().trim();
if (typeString.startsWith("(") && typeString.endsWith(")")) {
typeString = typeString.substring(1, typeString.length() - 1);
}
int lastIndexOf = typeString.lastIndexOf(":");
if (lastIndexOf > 0) {
typeString = typeString.substring(lastIndexOf + 1).trim();
TypeElement typeElement = null;
if (typeMirror.getKind() == TypeKind.DECLARED) {
typeElement = (TypeElement) ((DeclaredType) typeMirror).asElement();
}
//未配置 typeHandler 的情况下，只支持基本数据类型，不支持比如 list set 或者自定义的类等