refactor: 项目配置增加是否为生产环境配置项

continew-org · Oct 25, 2023 · 38deb95 · 38deb95
1 parent f25de2d
commit 38deb95
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/...ommon/src/main/java/top/charles7c/cnadmin/common/config/properties/ProjectProperties.java b/...ommon/src/main/java/top/charles7c/cnadmin/common/config/properties/ProjectProperties.java
@@ -81,6 +81,11 @@ public class ProjectProperties {
  @NestedConfigurationProperty
  private License license;
 
+ /**
+ * 是否为生产环境
+ */
+ private boolean production = false;
+
  /**
  * 是否本地解析 IP 归属地
  */

diff --git a/...ebapi/src/main/java/top/charles7c/cnadmin/webapi/controller/tool/GeneratorController.java b/...ebapi/src/main/java/top/charles7c/cnadmin/webapi/controller/tool/GeneratorController.java
@@ -30,8 +30,8 @@
 import org.springframework.web.bind.annotation.*;
 
 import cn.dev33.satoken.annotation.SaCheckPermission;
-import cn.hutool.extra.spring.SpringUtil;
 
+import top.charles7c.cnadmin.common.config.properties.ProjectProperties;
 import top.charles7c.cnadmin.common.model.query.PageQuery;
 import top.charles7c.cnadmin.common.model.vo.PageDataVO;
 import top.charles7c.cnadmin.common.model.vo.R;
@@ -57,6 +57,7 @@
 public class GeneratorController {
 
  private final GeneratorService generatorService;
+ private final ProjectProperties projectProperties;
 
  @Operation(summary = "分页查询数据表", description = "分页查询数据表")
  @SaCheckPermission("tool:generator:list")
@@ -97,7 +98,7 @@ public R saveConfig(@Validated @RequestBody GenConfigRequest request, @PathVaria
  @SaCheckPermission("tool:generator:list")
  @PostMapping("/{tableName}")
  public R generate(@PathVariable String tableName) {
- ValidationUtils.throwIf("prod".equals(SpringUtil.getActiveProfile()), "仅支持在开发环境生成代码");
+ ValidationUtils.throwIf(projectProperties.isProduction(), "仅支持在开发环境生成代码");
  generatorService.generate(tableName);
  return R.ok("生成成功，请查看生成代码是否正确");
  }

diff --git a/continew-admin-webapi/src/main/resources/config/application-prod.yml b/continew-admin-webapi/src/main/resources/config/application-prod.yml
@@ -2,6 +2,8 @@
 project:
  # URL
  url: https://cnadmin.charles7c.top
+ # 是否为生产环境
+ production: true
 
 --- ### 服务器配置
 server: