张延森 4 years ago
parent
commit
988e0e5b65

+ 1
- 1
generator_service/src/main/java/com/example/demo/controller/DbController.java View File

@@ -171,7 +171,7 @@ public class DbController {
171 171
         //strategy.setSuperEntityClass("com.baomidou.ant.common.BaseEntity");
172 172
         strategy.setEntityLombokModel(true);
173 173
         strategy.setRestControllerStyle(true);
174
-        strategy.setSuperControllerClass("com.example.demo.common.BaseController");
174
+//        strategy.setSuperControllerClass(dbConfig.getPackPath() + ".common.BaseController");
175 175
         //表名
176 176
         strategy.setInclude(tableName);
177 177
         strategy.setSuperEntityColumns("id");

+ 33
- 64
generator_service/src/main/resources/templatesPlus/controller.java.ftl View File

@@ -3,7 +3,8 @@ package ${package.Controller};
3 3
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
4 4
 import com.baomidou.mybatisplus.core.metadata.IPage;
5 5
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
6
-import com.community.huiju.common.model.ResponseBean;
6
+import ${package.Parent}.common.ResponseBean;
7
+import ${package.Parent}.common.BaseController;
7 8
 import org.slf4j.Logger;
8 9
 import org.slf4j.LoggerFactory;
9 10
 import org.springframework.beans.factory.annotation.Autowired;
@@ -32,6 +33,8 @@ import ${superControllerClassPackage};
32 33
  * @author ${author}
33 34
  * @since ${date}
34 35
  */
36
+
37
+@Api(tags = "${table.comment}")
35 38
 <#if restControllerStyle>
36 39
 @RestController
37 40
 <#else>
@@ -60,23 +63,16 @@ public class ${table.controllerName} {
60 63
      * @return
61 64
      */
62 65
     @RequestMapping(value="/${table.entityPath}",method= RequestMethod.GET)
63
-    public ResponseBean ${table.entityPath}List(@RequestParam(value ="pageNum",defaultValue = "1") Integer pageNum,
64
-									 @RequestParam(value ="pageSize",defaultValue = "10") Integer pageSize){
65
-        ResponseBean responseBean = new ResponseBean();
66
-        try {
67
-            //使用分页插件
66
+    @ApiOperation(value="列表", notes = "列表", httpMethod = "GET", response = ResponseBean.class)
67
+    public ResponseBean ${table.entityPath}List(@ApiParam("页码") @RequestParam(value ="pageNum",defaultValue = "1") Integer pageNum,
68
+									 @ApiParam("单页数据量") @RequestParam(value ="pageSize",defaultValue = "10") Integer pageSize) throws Exception{
69
+
68 70
 		    IPage<${entity}> pg = new Page<>(pageNum, pageSize);
69 71
             QueryWrapper<${entity}> queryWrapper = new QueryWrapper<>();
70 72
             queryWrapper.orderByDesc("create_date");
71 73
 
72 74
             IPage<${entity}> result = i${entity}Service.page(pg, queryWrapper);
73
-            responseBean.addSuccess(result);
74
-        }catch (Exception e){
75
-            e.printStackTrace();
76
-            logger.error("${table.entityPath}List -=- {}",e.toString());
77
-            responseBean.addError(e.getMessage());
78
-        }
79
-        return responseBean;
75
+            return ResponseBean.success(result);
80 76
     }
81 77
 
82 78
     /**
@@ -85,42 +81,28 @@ public class ${table.controllerName} {
85 81
      * @return
86 82
      */
87 83
     @RequestMapping(value="/${table.entityPath}",method= RequestMethod.POST)
88
-    public ResponseBean ${table.entityPath}Add(@RequestBody ${entity} ${table.entityPath}){
89
-        ResponseBean responseBean = new ResponseBean();
90
-        try {
91
-            if (i${entity}Service.save(${table.entityPath})){
92
-                responseBean.addSuccess(${table.entityPath});
93
-            }else {
94
-                responseBean.addError("fail");
95
-            }
96
-        }catch (Exception e){
97
-            e.printStackTrace();
98
-            logger.error("${table.entityPath}Add -=- {}",e.toString());
99
-            responseBean.addError(e.getMessage());
84
+    @ApiOperation(value="保存", notes = "保存", httpMethod = "POST", response = ResponseBean.class)
85
+    public ResponseBean ${table.entityPath}Add(@ApiParam("保存内容") @RequestBody ${entity} ${table.entityPath}) throws Exception{
86
+
87
+        if (i${entity}Service.save(${table.entityPath})){
88
+            return ResponseBean.success(${table.entityPath});
89
+        }else {
90
+            return ResponseBean.error("保存失败, 请重试", ResponseBean.ERROR_UNAVAILABLE);
100 91
         }
101
-        return responseBean;
102 92
     }
103 93
 
104 94
     /**
105 95
      * 根据id删除对象
106 96
      * @param id  实体ID
107 97
      */
108
-    @ResponseBody
109 98
     @RequestMapping(value="/${table.entityPath}/{id}", method= RequestMethod.DELETE)
110
-    public ResponseBean ${table.entityPath}Delete(@PathVariable Integer id){
111
-        ResponseBean responseBean = new ResponseBean();
112
-        try {
113
-            if(i${entity}Service.removeById(id)){
114
-                responseBean.addSuccess("success");
115
-            }else {
116
-                responseBean.addError("fail");
117
-            }
118
-        }catch (Exception e){
119
-            e.printStackTrace();
120
-            logger.error("${table.entityPath}Delete -=- {}",e.toString());
121
-            responseBean.addError(e.getMessage());
99
+    @ApiOperation(value="删除", notes = "删除", httpMethod = "DELETE", response = ResponseBean.class)
100
+    public ResponseBean ${table.entityPath}Delete(@ApiParam("对象ID") @PathVariable Integer id) throws Exception{
101
+        if(i${entity}Service.removeById(id)){
102
+            return ResponseBean.success("success");
103
+        }else {
104
+            return ResponseBean.error("删除失败, 请重试", ResponseBean.ERROR_UNAVAILABLE);
122 105
         }
123
-        return responseBean;
124 106
     }
125 107
 
126 108
     /**
@@ -130,21 +112,15 @@ public class ${table.controllerName} {
130 112
      * @return
131 113
      */
132 114
     @RequestMapping(value="/${table.entityPath}/{id}",method= RequestMethod.PUT)
133
-    public ResponseBean ${table.entityPath}Update(@PathVariable Integer id,
134
-                                        @RequestBody ${entity} ${table.entityPath}){
135
-        ResponseBean responseBean = new ResponseBean();
136
-        try {
137
-            if (i${entity}Service.updateById(${table.entityPath})){
138
-                responseBean.addSuccess(${table.entityPath});
139
-            }else {
140
-                responseBean.addError("fail");
141
-            }
142
-        }catch (Exception e){
143
-            e.printStackTrace();
144
-            logger.error("${table.entityPath}Update -=- {}",e.toString());
145
-            responseBean.addError(e.getMessage());
115
+    @ApiOperation(value="更新", notes = "更新", httpMethod = "PUT", response = ResponseBean.class)
116
+    public ResponseBean ${table.entityPath}Update(@ApiParam("对象ID") @PathVariable Integer id,
117
+                                        @ApiParam("更新内容") @RequestBody ${entity} ${table.entityPath}) throws Exception{
118
+
119
+        if (i${entity}Service.updateById(${table.entityPath})){
120
+            return ResponseBean.success(i${entity}Service.getById(id));
121
+        }else {
122
+            return ResponseBean.error("修改失败, 请重试", ResponseBean.ERROR_UNAVAILABLE);
146 123
         }
147
-        return responseBean;
148 124
     }
149 125
 
150 126
     /**
@@ -152,16 +128,9 @@ public class ${table.controllerName} {
152 128
      * @param id  实体ID
153 129
      */
154 130
     @RequestMapping(value="/${table.entityPath}/{id}",method= RequestMethod.GET)
155
-    public ResponseBean ${table.entityPath}Get(@PathVariable Integer id){
156
-        ResponseBean responseBean = new ResponseBean();
157
-        try {
158
-            responseBean.addSuccess(i${entity}Service.getById(id));
159
-        }catch (Exception e){
160
-            e.printStackTrace();
161
-            logger.error("${table.entityPath}Delete -=- {}",e.toString());
162
-            responseBean.addError(e.getMessage());
163
-        }
164
-        return responseBean;
131
+    @ApiOperation(value="详情", notes = "详情", httpMethod = "GET", response = ResponseBean.class)
132
+    public ResponseBean ${table.entityPath}Get(@ApiParam("对象ID") @PathVariable Integer id) throws Exception{
133
+        return ResponseBean.success(i${entity}Service.getById(id));
165 134
     }
166 135
 }
167 136
 </#if>