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

optimize FlexEnumTypeHandler.java

Browse Source
This commit is contained in:
开源海哥 2023-04-19 11:45:35 +08:00
parent 930f796812
commit 7ac8a4f056
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mybatis-flex-core/src/main/java/com/mybatisflex/core/handler/FlexEnumTypeHandler.java
View File

@ -24,6 +24,7 @@ import org.apache.ibatis.type.JdbcType;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
@ -34,13 +35,14 @@ public class FlexEnumTypeHandler<E extends Enum<E>> extends BaseTypeHandler<E> {
private Class<?> enumPropertyType;
private E[] enums;
private Field property;
private Method getter;
public FlexEnumTypeHandler(Class<E> enumClass) {
List<Field> allFields = ClassUtil.getAllFields(enumClass, field -> field.getAnnotation(EnumValue.class) != null);
Field field = allFields.get(0);
String fieldGetterName = "get"+ StringUtil.firstCharToUpperCase(field.getName());
String fieldGetterName = "get" + StringUtil.firstCharToUpperCase(field.getName());
List<Method> allMethods = ClassUtil.getAllMethods(enumClass, method -> {
String methodName = method.getName();
return methodName.equals(fieldGetterName);
@ -49,11 +51,16 @@ public class FlexEnumTypeHandler<E extends Enum<E>> extends BaseTypeHandler<E> {
enumPropertyType = ClassUtil.wrap(field.getType());
enums = enumClass.getEnumConstants();
if (allMethods.isEmpty()){
throw new IllegalStateException("Can not find \"" + fieldGetterName+"()\" method in enum: " + enumClass.getName());
if (allMethods.isEmpty()) {
if (Modifier.isPublic(field.getModifiers())) {
property = field;
} else {
throw new IllegalStateException("Can not find \"" + fieldGetterName + "()\" method in enum: " + enumClass.getName());
}
} else {
getter = allMethods.get(0);
}
getter = allMethods.get(0);
}
@ -68,9 +75,11 @@ public class FlexEnumTypeHandler<E extends Enum<E>> extends BaseTypeHandler<E> {
}
private Object getValue(Object object) {
private Object getValue(E object) {
try {
return this.getter.invoke(object);
return getter != null
? getter.invoke(object)
: property.get(object);
} catch (Exception e) {
throw FlexExceptions.wrap(e);
}