diff --git a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java index dad6de44..8d4e9d3e 100644 --- a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java +++ b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java @@ -188,7 +188,7 @@ public class QueryEntityProcessor extends AbstractProcessor { TypeElement classElement = (TypeElement) entityClassElement; do { - fillPropertyAndColumns(propertyAndColumns, defaultColumns, classElement); + fillPropertyAndColumns(propertyAndColumns, defaultColumns, classElement, (table == null || table.camelToUnderline())); classElement = (TypeElement) typeUtils.asElement(classElement.getSuperclass()); } while (classElement != null); @@ -233,7 +233,7 @@ public class QueryEntityProcessor extends AbstractProcessor { } - private void fillPropertyAndColumns(Map propertyAndColumns, List defaultColumns, TypeElement classElement) { + private void fillPropertyAndColumns(Map propertyAndColumns, List defaultColumns, TypeElement classElement, boolean camelToUnderline) { for (Element fieldElement : classElement.getEnclosedElements()) { //all fields @@ -286,7 +286,8 @@ public class QueryEntityProcessor extends AbstractProcessor { } - String columnName = column != null && column.value().trim().length() > 0 ? column.value() : camelToUnderline(fieldElement.toString()); + String columnName = column != null && column.value().trim().length() > 0 ? column.value() : + (camelToUnderline ? camelToUnderline(fieldElement.toString()) : fieldElement.toString()); propertyAndColumns.put(fieldElement.toString(), columnName); if (column == null || (!column.isLarge() && !column.isLogicDelete())) {