diff --git a/server/like-admin/src/main/resources/META-INF/additional-spring-configuration-metadata.json b/server/like-admin/src/main/resources/META-INF/additional-spring-configuration-metadata.json index 6a59a7b2..379bbd72 100644 --- a/server/like-admin/src/main/resources/META-INF/additional-spring-configuration-metadata.json +++ b/server/like-admin/src/main/resources/META-INF/additional-spring-configuration-metadata.json @@ -4,6 +4,11 @@ "name": "like.upload-directory", "type": "java.lang.String", "description": "Description for like.upload-directory." + }, + { + "name": "like.production", + "type": "java.lang.String", + "description": "Description for like.production." } ] } \ No newline at end of file diff --git a/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java b/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java index 4fbfc4de..ef4bdae7 100644 --- a/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java +++ b/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java @@ -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.isNotEmpty(production) && production.equals("true")) { + throw new OperateException("抱歉,演示环境不允许操作!"); + } + Assert.notNull(tables, "请选择要生成的表"); String[] tableNames = tables.split(","); byte[] data = iGenerateService.downloadCode(tableNames);