From ed13fddc52995e8e6ec893a2a56f2b208e5b8db8 Mon Sep 17 00:00:00 2001
From: RuoYi <yzz_ivy@163.com>
Date: 星期三, 25 十二月 2024 15:55:45 +0800
Subject: [PATCH] 代码生成新增配置是否允许文件覆盖到本地
---
src/main/java/com/ruoyi/project/tool/gen/controller/GenController.java | 5 +++++
src/main/java/com/ruoyi/framework/config/GenConfig.java | 17 +++++++++++++++--
src/main/resources/application.yml | 4 +++-
3 files changed, 23 insertions(+), 3 deletions(-)
diff --git a/src/main/java/com/ruoyi/framework/config/GenConfig.java b/src/main/java/com/ruoyi/framework/config/GenConfig.java
index a74e2c3..6d432ea 100644
--- a/src/main/java/com/ruoyi/framework/config/GenConfig.java
+++ b/src/main/java/com/ruoyi/framework/config/GenConfig.java
@@ -18,11 +18,14 @@
/** 鐢熸垚鍖呰矾寰� */
public static String packageName;
- /** 鑷姩鍘婚櫎琛ㄥ墠缂�锛岄粯璁ゆ槸true */
+ /** 鑷姩鍘婚櫎琛ㄥ墠缂� */
public static boolean autoRemovePre;
- /** 琛ㄥ墠缂�(绫诲悕涓嶄細鍖呭惈琛ㄥ墠缂�) */
+ /** 琛ㄥ墠缂� */
public static String tablePrefix;
+
+ /** 鏄惁鍏佽鐢熸垚鏂囦欢瑕嗙洊鍒版湰鍦帮紙鑷畾涔夎矾寰勶級 */
+ public static boolean allowOverwrite;
public static String getAuthor()
{
@@ -63,4 +66,14 @@
{
GenConfig.tablePrefix = tablePrefix;
}
+
+ public static boolean isAllowOverwrite()
+ {
+ return allowOverwrite;
+ }
+
+ public void setAllowOverwrite(boolean allowOverwrite)
+ {
+ GenConfig.allowOverwrite = allowOverwrite;
+ }
}
diff --git a/src/main/java/com/ruoyi/project/tool/gen/controller/GenController.java b/src/main/java/com/ruoyi/project/tool/gen/controller/GenController.java
index 820f9c3..63562d4 100644
--- a/src/main/java/com/ruoyi/project/tool/gen/controller/GenController.java
+++ b/src/main/java/com/ruoyi/project/tool/gen/controller/GenController.java
@@ -27,6 +27,7 @@
import com.ruoyi.common.utils.sql.SqlUtil;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
+import com.ruoyi.framework.config.GenConfig;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.framework.web.page.TableDataInfo;
@@ -213,6 +214,10 @@
@GetMapping("/genCode/{tableName}")
public AjaxResult genCode(@PathVariable("tableName") String tableName)
{
+ if (!GenConfig.isAllowOverwrite())
+ {
+ return AjaxResult.error("銆愮郴缁熼璁俱�戜笉鍏佽鐢熸垚鏂囦欢瑕嗙洊鍒版湰鍦�");
+ }
genTableService.generatorCode(tableName);
return success();
}
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 549f47f..1148bf4 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -137,4 +137,6 @@
# 鑷姩鍘婚櫎琛ㄥ墠缂�锛岄粯璁ゆ槸true
autoRemovePre: false
# 琛ㄥ墠缂�锛堢敓鎴愮被鍚嶄笉浼氬寘鍚〃鍓嶇紑锛屽涓敤閫楀彿鍒嗛殧锛�
- tablePrefix: sys_
\ No newline at end of file
+ tablePrefix: sys_
+ # 鏄惁鍏佽鐢熸垚鏂囦欢瑕嗙洊鍒版湰鍦帮紙鑷畾涔夎矾寰勶級锛岄粯璁や笉鍏佽
+ allowOverwrite: false
\ No newline at end of file
--
Gitblit v1.9.3