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

add config processer.mappersGenerateEnable

Browse Source
This commit is contained in:
开源海哥 2023-03-21 11:14:51 +08:00
parent 5342407c9a
commit 0d037f0c4b
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
mybatis-flex-annotation/src/main/java/com/mybatisflex/processer/QueryEntityProcesser.java
View File

@ -27,8 +27,6 @@ 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.Elements;
import javax.lang.model.util.Types;
import javax.tools.Diagnostic;
import javax.tools.JavaFileObject;
import java.io.*;
@ -101,15 +99,15 @@ public class QueryEntityProcesser extends AbstractProcessor {
private static final String allColumnsTemplate = " public QueryColumn[] ALL_COLUMNS = new QueryColumn[]{@allColumns};\n\n";
private Filer filer;
private Elements elementUtils;
private Types typeUtils;
// 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.typeUtils = processingEnvironment.getTypeUtils();
// this.elementUtils = processingEnvironment.getElementUtils();
// this.typeUtils = processingEnvironment.getTypeUtils();
}
@Override
@ -124,8 +122,9 @@ public class QueryEntityProcesser extends AbstractProcessor {
return true;
}
String genPath = props.getProperties().getProperty("processer.genPath", "");
final String genTablesPackage = props.getProperties().getProperty("processer.tablesPackage");
final String genMappersPackage = props.getProperties().getProperty("processer.mappersPackage");
String genTablesPackage = props.getProperties().getProperty("processer.tablesPackage");
String mappersGenerateEnable = props.getProperties().getProperty("processer.mappersGenerateEnable");
String genMappersPackage = props.getProperties().getProperty("processer.mappersPackage");
String className = props.getProperties().getProperty("processer.tablesClassName", "Tables");
StringBuilder guessPackage = new StringBuilder();
@ -200,9 +199,12 @@ public class QueryEntityProcesser extends AbstractProcessor {
tablesContent.append(buildTablesClass(entityClassName, tableName, propertyAndColumns, defaultColumns));
String realMapperPackage = genMappersPackage == null || genMappersPackage.trim().length() == 0
? guessPackage.substring(0, guessPackage.length() - 5) + "mapper" : genMappersPackage;
genMapperClass(genPath, realMapperPackage, entityClassElement.toString());
//是否开启 mapper 生成功能
if (!"false".equalsIgnoreCase(mappersGenerateEnable)) {
String realMapperPackage = genMappersPackage == null || genMappersPackage.trim().length() == 0
? guessPackage.substring(0, guessPackage.length() - 5) + "mapper" : genMappersPackage;
genMapperClass(genPath, realMapperPackage, entityClassElement.toString());
}
});
if (tablesContent.length() > 0) {