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