代码生成演示环境状态判断

pull/22/head
徐玉祥 2020-12-25 22:42:49 +08:00
parent 2247b75ee9
commit 36a753a604
1 changed files with 15 additions and 1 deletions

View File

@ -27,7 +27,9 @@ package com.cn.xiaonuo.generate.modular.controller;
import com.cn.xiaonuo.core.annotion.BusinessLog;
import com.cn.xiaonuo.core.annotion.Permission;
import com.cn.xiaonuo.core.context.constant.ConstantContextHolder;
import com.cn.xiaonuo.core.enums.LogAnnotionOpTypeEnum;
import com.cn.xiaonuo.core.exception.DemoException;
import com.cn.xiaonuo.core.pojo.page.PageResult;
import com.cn.xiaonuo.core.pojo.response.ResponseData;
import com.cn.xiaonuo.core.pojo.response.SuccessResponseData;
@ -35,7 +37,11 @@ import com.cn.xiaonuo.generate.modular.entity.CodeGenerate;
import com.cn.xiaonuo.generate.modular.param.CodeGenerateParam;
import com.cn.xiaonuo.generate.modular.service.CodeGenerateService;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.util.List;
@ -130,6 +136,10 @@ public class CodeGenerateController {
@PostMapping("/codeGenerate/runLocal")
@BusinessLog(title = "代码生成_本地项目", opType = LogAnnotionOpTypeEnum.OTHER)
public ResponseData runLocal(@RequestBody @Validated(CodeGenerateParam.detail.class) CodeGenerateParam codeGenerateParam) {
// 演示环境开启,则不允许操作
if (ConstantContextHolder.getDemoEnvFlag()) {
throw new DemoException();
}
this.codeGenerateService.runLocal(codeGenerateParam);
return new SuccessResponseData();
}
@ -144,6 +154,10 @@ public class CodeGenerateController {
@GetMapping("/codeGenerate/runDown")
@BusinessLog(title = "代码生成_下载方式", opType = LogAnnotionOpTypeEnum.OTHER)
public void runDown(@Validated(CodeGenerateParam.detail.class) CodeGenerateParam codeGenerateParam, HttpServletResponse response) {
// 演示环境开启,则不允许操作
if (ConstantContextHolder.getDemoEnvFlag()) {
throw new DemoException();
}
this.codeGenerateService.runDown(codeGenerateParam, response);
}
}