lywsvip/likeadmin_java
1
0
Fork 0
mirror of https://gitee.com/likeadmin/likeadmin_java.git synced 2026-06-02 07:01:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加演示环境校验

Browse Source
This commit is contained in:
TinyAnts
2022-08-17 18:25:45 +08:00
parent 135058a1e3
commit d0478a8fc4
2 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,6 +3,9 @@ package com.hxkj.generator.controller;
import com.baomidou.mybatisplus.core.toolkit.Assert;
import com.hxkj.common.core.AjaxResult;
import com.hxkj.common.core.PageResult;
import com.hxkj.common.exception.OperateException;
import com.hxkj.common.utils.StringUtil;
import com.hxkj.common.utils.YmlUtil;
import com.hxkj.common.validator.annotation.IDMust;
import com.hxkj.generator.service.IGenerateService;
import com.hxkj.generator.validate.GenParam;
@@ -159,6 +162,11 @@ public class GenController {
*/
@GetMapping("/downloadCode")
public void downloadCode(HttpServletResponse response, String tables) throws IOException {
String production = YmlUtil.get("like.production");
if (StringUtil.isNull(production) || StringUtil.isEmpty(production) || production.equals("true")) {
throw new OperateException("抱歉,演示环境不允许操作！");
}
Assert.notNull(tables, "请选择要生成的表");
String[] tableNames = tables.split(",");
byte[] data = iGenerateService.downloadCode(tableNames);