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 #I6VKDF

Browse Source
This commit is contained in:
开源海哥 2023-04-14 14:06:28 +08:00
parent 9067480138
commit 3536e0ac74
3 changed files with 67 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,6 +27,7 @@ import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.*;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.util.Types;
import javax.tools.Diagnostic;
import javax.tools.JavaFileObject;
import java.io.*;
@ -100,14 +101,14 @@ public class QueryEntityProcessor extends AbstractProcessor {
private Filer filer;
// private Elements elementUtils;
// private Types typeUtils;
private Types typeUtils;
@Override
public synchronized void init(ProcessingEnvironment processingEnvironment) {
super.init(processingEnvironment);
this.filer = processingEnvironment.getFiler();
// this.elementUtils = processingEnvironment.getElementUtils();
// this.typeUtils = processingEnvironment.getTypeUtils();
this.typeUtils = processingEnvironment.getTypeUtils();
}
@Override
@ -155,45 +156,10 @@ public class QueryEntityProcessor extends AbstractProcessor {
List<String> defaultColumns = new ArrayList<>();
TypeElement classElement = (TypeElement) entityClassElement;
for (Element fieldElement : classElement.getEnclosedElements()) {
//all fields
if (ElementKind.FIELD == fieldElement.getKind()) {
TypeMirror typeMirror = fieldElement.asType();
Column column = fieldElement.getAnnotation(Column.class);
if (column != null && column.ignore()) {
continue;
}
//获取 typeHandlerClass 的名称，通过 column.typeHandler() 获取会抛出异常：MirroredTypeException:
//参考 https://stackoverflow.com/questions/7687829/java-6-annotation-processing-getting-a-class-from-an-annotation
final String[] typeHandlerClass = {""};
List<? extends AnnotationMirror> annotationMirrors = fieldElement.getAnnotationMirrors();
for (AnnotationMirror annotationMirror : annotationMirrors) {
annotationMirror.getElementValues().forEach((BiConsumer<ExecutableElement, AnnotationValue>) (executableElement, annotationValue) -> {
if (executableElement.getSimpleName().equals("typeHandler")) {
typeHandlerClass[0] = annotationValue.toString();
}
});
}
//未配置 typeHandler 的情况下，只支持基本数据类型，不支持比如 list set 或者自定义的类等
if ((column == null || typeHandlerClass[0].equals(UnknownTypeHandler.class.getName()))
&& !defaultSupportColumnTypes.contains(typeMirror.toString())) {
continue;
}
String columnName = column != null && column.value().trim().length() > 0 ? column.value() : camelToUnderline(fieldElement.toString());
propertyAndColumns.put(fieldElement.toString(), columnName);
if (column == null || (!column.isLarge() && !column.isLogicDelete())) {
defaultColumns.add(columnName);
}
}
}
do {
fillPropertyAndColumns(propertyAndColumns, defaultColumns, classElement);
classElement = (TypeElement) typeUtils.asElement(classElement.getSuperclass());
}while (classElement!= null && "java.lang.Object".equals(classElement.toString()));
String entityClassName = entityClassElement.getSimpleName().toString();
tablesContent.append(buildTablesClass(entityClassName, tableName, propertyAndColumns, defaultColumns));
@ -218,6 +184,49 @@ public class QueryEntityProcessor extends AbstractProcessor {
}
private void fillPropertyAndColumns(Map<String, String> propertyAndColumns, List<String> defaultColumns, TypeElement classElement) {
for (Element fieldElement : classElement.getEnclosedElements()) {
//all fields
if (ElementKind.FIELD == fieldElement.getKind()) {
TypeMirror typeMirror = fieldElement.asType();
Column column = fieldElement.getAnnotation(Column.class);
if (column != null && column.ignore()) {
continue;
}
//获取 typeHandlerClass 的名称，通过 column.typeHandler() 获取会抛出异常：MirroredTypeException:
//参考 https://stackoverflow.com/questions/7687829/java-6-annotation-processing-getting-a-class-from-an-annotation
final String[] typeHandlerClass = {""};
List<? extends AnnotationMirror> annotationMirrors = fieldElement.getAnnotationMirrors();
for (AnnotationMirror annotationMirror : annotationMirrors) {
annotationMirror.getElementValues().forEach((BiConsumer<ExecutableElement, AnnotationValue>) (executableElement, annotationValue) -> {
if (executableElement.getSimpleName().equals("typeHandler")) {
typeHandlerClass[0] = annotationValue.toString();
}
});
}
//未配置 typeHandler 的情况下，只支持基本数据类型，不支持比如 list set 或者自定义的类等
if ((column == null || typeHandlerClass[0].equals(UnknownTypeHandler.class.getName()))
&& !defaultSupportColumnTypes.contains(typeMirror.toString())) {
continue;
}
String columnName = column != null && column.value().trim().length() > 0 ? column.value() : camelToUnderline(fieldElement.toString());
propertyAndColumns.put(fieldElement.toString(), columnName);
if (column == null || (!column.isLarge() && !column.isLogicDelete())) {
defaultColumns.add(columnName);
}
}
}
}
private static String guessMapperPackage(String entityClassName) {
if (!entityClassName.contains(".")) {
return "mapper";

2
mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/Account.java
View File

@ -9,7 +9,7 @@ import java.util.HashMap;
import java.util.Map;
@Table(value = "tb_account",dataSource = "ds2",onSet = AccountOnSetListener.class)
public class Account {
public class Account extends BaseAccount{
@Id(keyType = KeyType.Auto)
private Long id;

16
mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/BaseAccount.java Normal file
View File

@ -0,0 +1,16 @@
package com.mybatisflex.test;
import java.util.Date;
public class BaseAccount {
private Date createTime;
public Date getCreateTime() {
return createTime;
}
public void setCreateTime(Date createTime) {
this.createTime = createTime;
}
}