fix bug

imyzt · Aug 17, 2019 · dae1725 · dae1725
1 parent 5fb6486
commit dae1725
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/main/java/com/github/mars05/crud/intellij/plugin/util/PsiFileUtils.java b/src/main/java/com/github/mars05/crud/intellij/plugin/util/PsiFileUtils.java
@@ -230,12 +230,12 @@ public String outputFile(TableInfo tableInfo) {
         strategy.setNaming(NamingStrategy.underline_to_camel);
         strategy.setColumnNaming(NamingStrategy.underline_to_camel);
         strategy.setEntityLombokModel(true);
-        strategy.setSuperControllerClass("cn.ideamake.template.web.controller.AbstractController");
+//        strategy.setSuperControllerClass("cn.ideamake.template.web.controller.AbstractController");
         strategy.setInclude(getTableNameList(tables));
         strategy.setControllerMappingHyphenStyle(true);
         strategy.setRestControllerStyle(true);
         //表前缀
-        strategy.setTablePrefix("im_");
+        strategy.setTablePrefix(SelectionContext.getTablePrefix());
         mpg.setStrategy(strategy);
         mpg.setTemplateEngine(new FreemarkerTemplateEngine());
         mpg.execute();

diff --git a/src/main/resources/templates/Swagger2Config.ftl b/src/main/resources/templates/Swagger2Config.ftl
@@ -3,6 +3,7 @@ package ${package+'.config'};
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import springfox.documentation.builders.ApiInfoBuilder;
+import springfox.documentation.service.Contact;
 import springfox.documentation.builders.PathSelectors;
 import springfox.documentation.builders.RequestHandlerSelectors;
 import springfox.documentation.service.ApiInfo;