feats

generator_service/src/main/java/com/example/demo/controller/DbController.java

@@ -20,9 +20,7 @@ import javax.servlet.http.HttpServletResponse;
 import java.io.*;
 import java.net.URLEncoder;
 import java.sql.*;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.UUID;
+import java.util.*;
 
 @RestController
 @RequestMapping("/mydb")
@@ -129,7 +127,11 @@ public class DbController {
         InjectionConfig cfg = new InjectionConfig() {
             @Override
             public void initMap() {
-                // to do nothing
+                Map<String, Object> map = new HashMap<>();
+                map.put("BasePackage", dbConfig.getPackPath());
+                map.put("ResponseClassPackage", dbConfig.getPackPath() + ".common.ResponseBean");
+                map.put("ResponseClass", "ResponseBean");
+                this.setMap(map);
             }
         };
 
@@ -160,6 +162,7 @@ public class DbController {
         // templateConfig.setEntity();
         // templateConfig.setService();
         templateConfig.setController("templatesPlus/controller.java");
+        templateConfig.setMapper("templatesPlus/mapper.java");
 
         templateConfig.setXml(null);
         mpg.setTemplate(templateConfig);
@@ -171,7 +174,7 @@ public class DbController {
         //strategy.setSuperEntityClass("com.baomidou.ant.common.BaseEntity");
         strategy.setEntityLombokModel(true);
         strategy.setRestControllerStyle(true);
-//        strategy.setSuperControllerClass(dbConfig.getPackPath() + ".common.BaseController");
+        strategy.setSuperControllerClass(dbConfig.getPackPath() + ".common.BaseController");
         //表名
         strategy.setInclude(tableName);
         strategy.setSuperEntityColumns("id");

generator_service/src/main/resources/application.yml

@@ -1,2 +1,2 @@
 server:
-  port: 8087
+  port: 8080

generator_service/src/main/resources/templatesPlus/controller.java.ftl

@@ -3,8 +3,11 @@ package ${package.Controller};
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import ${package.Parent}.common.ResponseBean;
-import ${package.Parent}.common.BaseController;
+import ${superControllerClassPackage};
+import ${cfg.ResponseClassPackage};
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import io.swagger.annotations.ApiParam;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -13,7 +16,6 @@ import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.ResponseBody;
 import ${package.Service}.${table.serviceName};
 import ${package.Entity}.${table.entityName};
 <#if restControllerStyle>
@@ -21,9 +23,6 @@ import org.springframework.web.bind.annotation.RestController;
 <#else>
 import org.springframework.stereotype.Controller;
 </#if>
-<#if superControllerClassPackage??>
-import ${superControllerClassPackage};
-</#if>
 
 /**
  * <p>
@@ -63,8 +62,8 @@ public class ${table.controllerName} {
      * @return
      */
     @RequestMapping(value="/${table.entityPath}",method= RequestMethod.GET)
-    @ApiOperation(value="列表", notes = "列表", httpMethod = "GET", response = ResponseBean.class)
-    public ResponseBean ${table.entityPath}List(@ApiParam("页码") @RequestParam(value ="pageNum",defaultValue = "1") Integer pageNum,
+    @ApiOperation(value="列表", notes = "列表", httpMethod = "GET", response = ${cfg.ResponseClass}.class)
+    public ${cfg.ResponseClass} ${table.entityPath}List(@ApiParam("页码") @RequestParam(value ="pageNum",defaultValue = "1") Integer pageNum,
 									 @ApiParam("单页数据量") @RequestParam(value ="pageSize",defaultValue = "10") Integer pageSize) throws Exception{
 
 		    IPage<${entity}> pg = new Page<>(pageNum, pageSize);
@@ -81,13 +80,13 @@ public class ${table.controllerName} {
      * @return
      */
     @RequestMapping(value="/${table.entityPath}",method= RequestMethod.POST)
-    @ApiOperation(value="保存", notes = "保存", httpMethod = "POST", response = ResponseBean.class)
-    public ResponseBean ${table.entityPath}Add(@ApiParam("保存内容") @RequestBody ${entity} ${table.entityPath}) throws Exception{
+    @ApiOperation(value="保存", notes = "保存", httpMethod = "POST", response = ${cfg.ResponseClass}.class)
+    public ${cfg.ResponseClass} ${table.entityPath}Add(@ApiParam("保存内容") @RequestBody ${entity} ${table.entityPath}) throws Exception{
 
         if (i${entity}Service.save(${table.entityPath})){
-            return ResponseBean.success(${table.entityPath});
+            return ${cfg.ResponseClass}.success(${table.entityPath});
         }else {
-            return ResponseBean.error("保存失败, 请重试", ResponseBean.ERROR_UNAVAILABLE);
+            return ${cfg.ResponseClass}.error("保存失败, 请重试", ${cfg.ResponseClass}.ERROR_UNAVAILABLE);
         }
     }
 
@@ -96,12 +95,12 @@ public class ${table.controllerName} {
      * @param id  实体ID
      */
     @RequestMapping(value="/${table.entityPath}/{id}", method= RequestMethod.DELETE)
-    @ApiOperation(value="删除", notes = "删除", httpMethod = "DELETE", response = ResponseBean.class)
-    public ResponseBean ${table.entityPath}Delete(@ApiParam("对象ID") @PathVariable Integer id) throws Exception{
+    @ApiOperation(value="删除", notes = "删除", httpMethod = "DELETE", response = ${cfg.ResponseClass}.class)
+    public ${cfg.ResponseClass} ${table.entityPath}Delete(@ApiParam("对象ID") @PathVariable Integer id) throws Exception{
         if(i${entity}Service.removeById(id)){
-            return ResponseBean.success("success");
+            return ${cfg.ResponseClass}.success("success");
         }else {
-            return ResponseBean.error("删除失败, 请重试", ResponseBean.ERROR_UNAVAILABLE);
+            return ${cfg.ResponseClass}.error("删除失败, 请重试", ${cfg.ResponseClass}.ERROR_UNAVAILABLE);
         }
     }
 
@@ -112,14 +111,14 @@ public class ${table.controllerName} {
      * @return
      */
     @RequestMapping(value="/${table.entityPath}/{id}",method= RequestMethod.PUT)
-    @ApiOperation(value="更新", notes = "更新", httpMethod = "PUT", response = ResponseBean.class)
-    public ResponseBean ${table.entityPath}Update(@ApiParam("对象ID") @PathVariable Integer id,
+    @ApiOperation(value="更新", notes = "更新", httpMethod = "PUT", response = ${cfg.ResponseClass}.class)
+    public ${cfg.ResponseClass} ${table.entityPath}Update(@ApiParam("对象ID") @PathVariable Integer id,
                                         @ApiParam("更新内容") @RequestBody ${entity} ${table.entityPath}) throws Exception{
 
         if (i${entity}Service.updateById(${table.entityPath})){
-            return ResponseBean.success(i${entity}Service.getById(id));
+            return ${cfg.ResponseClass}.success(i${entity}Service.getById(id));
         }else {
-            return ResponseBean.error("修改失败, 请重试", ResponseBean.ERROR_UNAVAILABLE);
+            return ${cfg.ResponseClass}.error("修改失败, 请重试", ${cfg.ResponseClass}.ERROR_UNAVAILABLE);
         }
     }
 
@@ -128,9 +127,9 @@ public class ${table.controllerName} {
      * @param id  实体ID
      */
     @RequestMapping(value="/${table.entityPath}/{id}",method= RequestMethod.GET)
-    @ApiOperation(value="详情", notes = "详情", httpMethod = "GET", response = ResponseBean.class)
-    public ResponseBean ${table.entityPath}Get(@ApiParam("对象ID") @PathVariable Integer id) throws Exception{
-        return ResponseBean.success(i${entity}Service.getById(id));
+    @ApiOperation(value="详情", notes = "详情", httpMethod = "GET", response = ${cfg.ResponseClass}.class)
+    public ${cfg.ResponseClass} ${table.entityPath}Get(@ApiParam("对象ID") @PathVariable Integer id) throws Exception{
+        return ${cfg.ResponseClass}.success(i${entity}Service.getById(id));
     }
 }
 </#if>

generator_service/src/main/resources/templatesPlus/mapper.java.ftl

@@ -0,0 +1,22 @@
+package ${package.Mapper};
+
+import ${package.Entity}.${entity};
+import ${superMapperClassPackage};
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * <p>
+ * ${table.comment!} Mapper 接口
+ * </p>
+ *
+ * @author ${author}
+ * @since ${date}
+ */
+<#if kotlin>
+interface ${table.mapperName} : ${superMapperClass}<${entity}>
+<#else>
+@Mapper
+public interface ${table.mapperName} extends ${superMapperClass}<${entity}> {
+
+}
+</#if>