mirage
/
edu
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改生产环境拦截问题

This commit is contained in:
TinyAnts 2022-08-17 18:52:31 +08:00
parent 24ce27cf6c
commit 3a9aa46118
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java
View File

@ -145,6 +145,11 @@ public class GenController {
*/ */
@GetMapping("/genCode") @GetMapping("/genCode")
public Object genCode(String tables) { public Object genCode(String tables) {
String production = YmlUtil.get("like.production");
if (StringUtil.isNotEmpty(production) && production.equals("true")) {
throw new OperateException("抱歉,演示环境不允许操作！");
}
Assert.notNull(tables, "请选择要生成的表"); Assert.notNull(tables, "请选择要生成的表");
String[] tableNames = tables.split(","); String[] tableNames = tables.split(",");
for (String tableName : tableNames) { for (String tableName : tableNames) {
@ -162,11 +167,6 @@ public class GenController {
*/ */
@GetMapping("/downloadCode") @GetMapping("/downloadCode")
public void downloadCode(HttpServletResponse response, String tables) throws IOException { public void downloadCode(HttpServletResponse response, String tables) throws IOException {
String production = YmlUtil.get("like.production");
if (StringUtil.isNotEmpty(production) && production.equals("true")) {
throw new OperateException("抱歉,演示环境不允许操作！");
}
Assert.notNull(tables, "请选择要生成的表"); Assert.notNull(tables, "请选择要生成的表");
String[] tableNames = tables.split(","); String[] tableNames = tables.split(",");
byte[] data = iGenerateService.downloadCode(tableNames); byte[] data = iGenerateService.downloadCode(tableNames);