From 47a53d22a025c0ed095c9c4ed2e1ce35ce54184e Mon Sep 17 00:00:00 2001 From: k94314517 <8417338+k94314517@user.noreply.gitee.com> Date: 星期三, 30 十月 2024 14:16:16 +0800 Subject: [PATCH] 代码初始化 --- server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/HiddenDangerCloudController.java | 16 ++++++++++++++-- 1 files changed, 14 insertions(+), 2 deletions(-) diff --git a/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/HiddenDangerCloudController.java b/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/HiddenDangerCloudController.java index a4960d6..f6604e0 100644 --- a/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/HiddenDangerCloudController.java +++ b/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/HiddenDangerCloudController.java @@ -4,7 +4,10 @@ import com.doumee.config.annotation.CloudRequiredPermission; import com.doumee.core.annotation.excel.ExcelExporter; import com.doumee.core.annotation.pr.PreventRepeat; +import com.doumee.core.constants.ResponseStatus; +import com.doumee.core.exception.BusinessException; import com.doumee.core.model.ApiResponse; +import com.doumee.core.model.LoginUserInfo; import com.doumee.core.model.PageData; import com.doumee.core.model.PageWrap; import com.doumee.core.utils.Constants; @@ -16,6 +19,7 @@ import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletResponse; +import java.util.Objects; /** * @author 姹熻箘韫� @@ -34,7 +38,11 @@ @PostMapping("/create") @CloudRequiredPermission("business:hiddendanger:create") public ApiResponse create(@RequestBody HiddenDanger hiddenDanger,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ - hiddenDanger.setLoginUserInfo(this.getLoginUser(token)); + LoginUserInfo loginUserInfo = this.getLoginUser(token); + if(Objects.isNull(loginUserInfo.getMemberId())||Objects.isNull(loginUserInfo)){ + throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"褰撳墠璐︽埛鏃犳硶杩涜璇ユ搷浣�"); + } + hiddenDanger.setLoginUserInfo(loginUserInfo); return ApiResponse.success(hiddenDangerService.create(hiddenDanger)); } @@ -58,7 +66,11 @@ @PostMapping("/updateById") @CloudRequiredPermission("business:hiddendanger:update") public ApiResponse updateById(@RequestBody HiddenDanger hiddenDanger,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ - hiddenDanger.setLoginUserInfo(this.getLoginUser(token)); + LoginUserInfo loginUserInfo = this.getLoginUser(token); + if(Objects.isNull(loginUserInfo.getMemberId())||Objects.isNull(loginUserInfo)){ + throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"褰撳墠璐︽埛鏃犳硶杩涜璇ユ搷浣�"); + } + hiddenDanger.setLoginUserInfo(loginUserInfo); hiddenDangerService.updateById(hiddenDanger); return ApiResponse.success(null); } -- Gitblit v1.9.3