Skip to content

G
guns-vip
Commit f33982ac by stylefeng
Options

升级代码生成

parent b9ebe93c
Showing with 54 additions and 22 deletions
guns-admin/src/main/java/generator/GunsCodeGenerator.java
package generator; package generator;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.stylefeng.guns.core.template.engine.base.GunsTemplateEngine;
import generator.config.GunsGeneratorConfig; import generator.config.GunsGeneratorConfig;
/** /**
...@@ -14,14 +12,18 @@ public class GunsCodeGenerator { ...@@ -14,14 +12,18 @@ public class GunsCodeGenerator {
public static void main(String[] args) { public static void main(String[] args) {
//mp的生成器 /**
* Mybatis-Plus的代码生成器:
* mp的代码生成器可以生成实体,mapper,mapper对应的xml,service
*/
GunsGeneratorConfig gunsGeneratorConfig = new GunsGeneratorConfig(); GunsGeneratorConfig gunsGeneratorConfig = new GunsGeneratorConfig();
AutoGenerator generator = gunsGeneratorConfig.getGenerator(); gunsGeneratorConfig.doMpGeneration();
generator.execute();
//guns的生成器 /**
GunsTemplateEngine gunsTemplateEngine = gunsGeneratorConfig.getGunsTemplateEngine(); * guns的生成器:
gunsTemplateEngine.start(); * guns的代码生成器可以生成controller,html页面,页面对应的js
*/
gunsGeneratorConfig.doGunsGeneration();
} }
} }
\ No newline at end of file
guns-admin/src/main/java/generator/config/AbstractGeneratorConfig.java
...@@ -8,6 +8,9 @@ import com.baomidou.mybatisplus.generator.config.StrategyConfig; ...@@ -8,6 +8,9 @@ import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.stylefeng.guns.core.template.config.ContextConfig; import com.stylefeng.guns.core.template.config.ContextConfig;
import com.stylefeng.guns.core.template.engine.SimpleTemplateEngine; import com.stylefeng.guns.core.template.engine.SimpleTemplateEngine;
import com.stylefeng.guns.core.template.engine.base.GunsTemplateEngine; import com.stylefeng.guns.core.template.engine.base.GunsTemplateEngine;
import com.stylefeng.guns.core.util.FileUtil;
import java.io.File;
/** /**
* 代码生成的抽象配置 * 代码生成的抽象配置
...@@ -51,28 +54,53 @@ public abstract class AbstractGeneratorConfig { ...@@ -51,28 +54,53 @@ public abstract class AbstractGeneratorConfig {
packageConfig(); packageConfig();
contextConfig(); contextConfig();
//controller没用掉,生成之后会自动删掉
packageConfig.setController("TTT");
packageConfig.setService("com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".service"); packageConfig.setService("com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".service");
packageConfig.setServiceImpl("com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".service.impl"); packageConfig.setServiceImpl("com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".service.impl");
//controller没用掉,生成之后会自动删掉
packageConfig.setController("TTT");
if(!contextConfig.getEntitySwitch()){
packageConfig.setEntity("TTT");
}
if(!contextConfig.getDaoSwitch()){
packageConfig.setMapper("TTT");
packageConfig.setXml("TTT");
}
if(!contextConfig.getServiceSwitch()){
packageConfig.setService("TTT");
packageConfig.setServiceImpl("TTT");
}
}
/**
* 删除不必要的代码
*/
public void destory(){
String outputDir = globalConfig.getOutputDir() + "/TTT";
FileUtil.deleteDir(new File(outputDir));
} }
public AbstractGeneratorConfig() { public AbstractGeneratorConfig() {
init(); init();
} }
public AutoGenerator getGenerator() { public void doMpGeneration() {
AutoGenerator autoGenerator = new AutoGenerator(); AutoGenerator autoGenerator = new AutoGenerator();
autoGenerator.setGlobalConfig(globalConfig); autoGenerator.setGlobalConfig(globalConfig);
autoGenerator.setDataSource(dataSourceConfig); autoGenerator.setDataSource(dataSourceConfig);
autoGenerator.setStrategy(strategyConfig); autoGenerator.setStrategy(strategyConfig);
autoGenerator.setPackageInfo(packageConfig); autoGenerator.setPackageInfo(packageConfig);
return autoGenerator; autoGenerator.execute();
destory();
} }
public GunsTemplateEngine getGunsTemplateEngine() { public void doGunsGeneration() {
GunsTemplateEngine gunsTemplateEngine = new SimpleTemplateEngine(); GunsTemplateEngine gunsTemplateEngine = new SimpleTemplateEngine();
gunsTemplateEngine.setContextConfig(contextConfig); gunsTemplateEngine.setContextConfig(contextConfig);
return gunsTemplateEngine; gunsTemplateEngine.start();
} }
} }
guns-admin/src/main/java/generator/config/GunsGeneratorConfig.java
...@@ -18,6 +18,7 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig { ...@@ -18,6 +18,7 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig {
globalConfig.setEnableCache(false); globalConfig.setEnableCache(false);
globalConfig.setBaseResultMap(true); globalConfig.setBaseResultMap(true);
globalConfig.setBaseColumnList(true); globalConfig.setBaseColumnList(true);
globalConfig.setOpen(false);
globalConfig.setAuthor("stylefeng"); globalConfig.setAuthor("stylefeng");
} }
...@@ -33,7 +34,7 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig { ...@@ -33,7 +34,7 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig {
@Override @Override
protected void strategyConfig() { protected void strategyConfig() {
//strategy.setTablePrefix(new String[]{"_"});// 此处可以修改为您的表前缀 //strategy.setTablePrefix(new String[]{"_"});// 此处可以修改为您的表前缀
strategyConfig.setInclude(new String[]{"bill"}); //strategyConfig.setInclude(new String[]{"bill"});
strategyConfig.setNaming(NamingStrategy.underline_to_camel); strategyConfig.setNaming(NamingStrategy.underline_to_camel);
} }
...@@ -57,13 +58,13 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig { ...@@ -57,13 +58,13 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig {
* 生成器开关 * 生成器开关
*/ */
contextConfig.setEntitySwitch(true); contextConfig.setEntitySwitch(true);
contextConfig.setControllerSwitch(true); contextConfig.setControllerSwitch(false);
contextConfig.setDaoSwitch(true); contextConfig.setDaoSwitch(true);
contextConfig.setServiceSwitch(true); contextConfig.setServiceSwitch(false);
contextConfig.setIndexPageSwitch(true); contextConfig.setIndexPageSwitch(false);
contextConfig.setAddPageSwitch(true); contextConfig.setAddPageSwitch(false);
contextConfig.setEditPageSwitch(true); contextConfig.setEditPageSwitch(false);
contextConfig.setJsSwitch(true); contextConfig.setJsSwitch(false);
contextConfig.setInfoJsSwitch(true); contextConfig.setInfoJsSwitch(false);
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment