admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-06 16:48: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

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