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

apt add "processor.entity.ignoreSuffixes" config

Browse Source
This commit is contained in:
开源海哥 2023-04-24 12:17:49 +08:00
parent b6fe442da9
commit 6c740d0e36
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -129,9 +129,11 @@ public class QueryEntityProcessor extends AbstractProcessor {
String mappersGenerateEnable = props.getProperties().getProperty("processor.mappersGenerateEnable", "false");
String genMappersPackage = props.getProperties().getProperty("processor.mappersPackage");
String className = props.getProperties().getProperty("processor.tablesClassName", "Tables");
String entityNameReplaceEnabled = props.getProperties().getProperty("processor.entityName.replace.enabled", "false");
String entityNameReplaceRegex = props.getProperties().getProperty("processor.entityName.replace.regex", "");
String entityNameReplaceReplacement = props.getProperties().getProperty("processor.entityName.replace.replacement", "");
String[] entityIgnoreSuffixes = props.getProperties().getProperty("processor.entity.ignoreSuffixes", "").split(",");
// String entityNameReplaceRegex = props.getProperties().getProperty("processor.entityName.replace.regex", "");
// String entityNameReplaceReplacement = props.getProperties().getProperty("processor.entityName.replace.replacement", "");
StringBuilder guessPackage = new StringBuilder();
@ -166,21 +168,24 @@ public class QueryEntityProcessor extends AbstractProcessor {
classElement = (TypeElement) typeUtils.asElement(classElement.getSuperclass());
} while (classElement != null);
String entityClassName = entityClassElement.getSimpleName().toString();
String entityName = entityClassName;
if("true".equals(entityNameReplaceEnabled)){
entityName = entityName.replace(entityNameReplaceRegex, entityNameReplaceReplacement);
String entitySimpleName = entityClassElement.getSimpleName().toString();
if (entityIgnoreSuffixes.length > 0) {
for (String entityIgnoreSuffix : entityIgnoreSuffixes) {
if (entitySimpleName.endsWith(entityIgnoreSuffix.trim())) {
entitySimpleName = entitySimpleName.substring(0, entitySimpleName.length() - entityIgnoreSuffix.length());
break;
}
}
}
tablesContent.append(buildTablesClass(entityName, tableName, propertyAndColumns, defaultColumns));
tablesContent.append(buildTablesClass(entitySimpleName, tableName, propertyAndColumns, defaultColumns));
//是否开启 mapper 生成功能
if ("true".equalsIgnoreCase(mappersGenerateEnable) && table.mapperGenerateEnable()) {
String realMapperPackage = genMappersPackage == null || genMappersPackage.trim().length() == 0
? guessMapperPackage(entityClassElement.toString()) : genMappersPackage;
genMapperClass(genPath, realMapperPackage, entityClassElement.toString(),baseMapperClass, entityName);
genMapperClass(genPath, realMapperPackage, entityClassElement.toString(), baseMapperClass, entitySimpleName);
}
});
@ -382,7 +387,6 @@ public class QueryEntityProcessor extends AbstractProcessor {
}
/**
*
* @param genBasePath 生成路径
* @param genPackageName 包名
* @param entityClass 实体类名