升级代码生成

guns-admin/src/main/java/generator/GunsCodeGenerator.java

 package generator;
 
-import com.baomidou.mybatisplus.generator.AutoGenerator;
-import com.stylefeng.guns.core.template.engine.base.GunsTemplateEngine;
 import generator.config.GunsGeneratorConfig;
 
 /**
@@ -14,14 +12,18 @@ public class GunsCodeGenerator {
 
     public static void main(String[] args) {
 
-        //mp的生成器
+        /**
+         * Mybatis-Plus的代码生成器:
+         *      mp的代码生成器可以生成实体,mapper,mapper对应的xml,service
+         */
         GunsGeneratorConfig gunsGeneratorConfig = new GunsGeneratorConfig();
-        AutoGenerator generator = gunsGeneratorConfig.getGenerator();
-        generator.execute();
+        gunsGeneratorConfig.doMpGeneration();
 
-        //guns的生成器
-        GunsTemplateEngine gunsTemplateEngine = gunsGeneratorConfig.getGunsTemplateEngine();
-        gunsTemplateEngine.start();
+        /**
+         * guns的生成器:
+         *      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;
 import com.stylefeng.guns.core.template.config.ContextConfig;
 import com.stylefeng.guns.core.template.engine.SimpleTemplateEngine;
 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 {
         packageConfig();
         contextConfig();
 
-        //controller没用掉,生成之后会自动删掉
-        packageConfig.setController("TTT");
         packageConfig.setService("com.stylefeng.guns.modular." + contextConfig.getModuleName() + ".service");
         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() {
         init();
     }
 
-    public AutoGenerator getGenerator() {
+    public void doMpGeneration() {
         AutoGenerator autoGenerator = new AutoGenerator();
         autoGenerator.setGlobalConfig(globalConfig);
         autoGenerator.setDataSource(dataSourceConfig);
         autoGenerator.setStrategy(strategyConfig);
         autoGenerator.setPackageInfo(packageConfig);
-        return autoGenerator;
+        autoGenerator.execute();
+        destory();
     }
 
-    public GunsTemplateEngine getGunsTemplateEngine() {
+    public void doGunsGeneration() {
         GunsTemplateEngine gunsTemplateEngine = new SimpleTemplateEngine();
         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 {
         globalConfig.setEnableCache(false);
         globalConfig.setBaseResultMap(true);
         globalConfig.setBaseColumnList(true);
+        globalConfig.setOpen(false);
         globalConfig.setAuthor("stylefeng");
     }
 
@@ -33,7 +34,7 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig {
     @Override
     protected void strategyConfig() {
         //strategy.setTablePrefix(new String[]{"_"});// 此处可以修改为您的表前缀
-        strategyConfig.setInclude(new String[]{"bill"});
+        //strategyConfig.setInclude(new String[]{"bill"});
         strategyConfig.setNaming(NamingStrategy.underline_to_camel);
     }
 
@@ -57,13 +58,13 @@ public class GunsGeneratorConfig extends AbstractGeneratorConfig {
          * 生成器开关
          */
         contextConfig.setEntitySwitch(true);
-        contextConfig.setControllerSwitch(true);
+        contextConfig.setControllerSwitch(false);
         contextConfig.setDaoSwitch(true);
-        contextConfig.setServiceSwitch(true);
-        contextConfig.setIndexPageSwitch(true);
-        contextConfig.setAddPageSwitch(true);
-        contextConfig.setEditPageSwitch(true);
-        contextConfig.setJsSwitch(true);
-        contextConfig.setInfoJsSwitch(true);
+        contextConfig.setServiceSwitch(false);
+        contextConfig.setIndexPageSwitch(false);
+        contextConfig.setAddPageSwitch(false);
+        contextConfig.setEditPageSwitch(false);
+        contextConfig.setJsSwitch(false);
+        contextConfig.setInfoJsSwitch(false);
     }
 }