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

This commit is contained in:
Charles7c 2023-10-25 22:00:41 +08:00
parent f25de2d7f8
commit 38deb950ac
3 changed files with 10 additions and 2 deletions

View File

@ -81,6 +81,11 @@ public class ProjectProperties {
@NestedConfigurationProperty @NestedConfigurationProperty
private License license; private License license;
/**
* 是否为生产环境
*/
private boolean production = false;
/** /**
* 是否本地解析 IP 归属地 * 是否本地解析 IP 归属地
*/ */

View File

@ -30,8 +30,8 @@ import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import cn.dev33.satoken.annotation.SaCheckPermission; 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.query.PageQuery;
import top.charles7c.cnadmin.common.model.vo.PageDataVO; import top.charles7c.cnadmin.common.model.vo.PageDataVO;
import top.charles7c.cnadmin.common.model.vo.R; import top.charles7c.cnadmin.common.model.vo.R;
@ -57,6 +57,7 @@ import top.charles7c.cnadmin.tool.service.GeneratorService;
public class GeneratorController { public class GeneratorController {
private final GeneratorService generatorService; private final GeneratorService generatorService;
private final ProjectProperties projectProperties;
@Operation(summary = "分页查询数据表", description = "分页查询数据表") @Operation(summary = "分页查询数据表", description = "分页查询数据表")
@SaCheckPermission("tool:generator:list") @SaCheckPermission("tool:generator:list")
@ -97,7 +98,7 @@ public class GeneratorController {
@SaCheckPermission("tool:generator:list") @SaCheckPermission("tool:generator:list")
@PostMapping("/{tableName}") @PostMapping("/{tableName}")
public R generate(@PathVariable String tableName) { public R generate(@PathVariable String tableName) {
ValidationUtils.throwIf("prod".equals(SpringUtil.getActiveProfile()), "仅支持在开发环境生成代码"); ValidationUtils.throwIf(projectProperties.isProduction(), "仅支持在开发环境生成代码");
generatorService.generate(tableName); generatorService.generate(tableName);
return R.ok("生成成功,请查看生成代码是否正确"); return R.ok("生成成功,请查看生成代码是否正确");
} }

View File

@ -2,6 +2,8 @@
project: project:
# URL # URL
url: https://cnadmin.charles7c.top url: https://cnadmin.charles7c.top
# 是否为生产环境
production: true
--- ### 服务器配置 --- ### 服务器配置
server: server: