From f63d3a3be83fea258e7920efccd2567b6a511bb3 Mon Sep 17 00:00:00 2001
From: jiangping <jp@doumee.com>
Date: 星期四, 16 一月 2025 16:52:46 +0800
Subject: [PATCH] 最新版本541200007

---
 server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/HiddenDangerCloudController.java |   27 +++++++++++++++++++++------
 1 files changed, 21 insertions(+), 6 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..4aabab3 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,9 +4,12 @@
 import com.doumee.config.annotation.CloudRequiredPermission;
 import com.doumee.core.annotation.excel.ExcelExporter;
 import com.doumee.core.annotation.pr.PreventRepeat;
-import com.doumee.core.model.ApiResponse;
-import com.doumee.core.model.PageData;
-import com.doumee.core.model.PageWrap;
+import com.doumee.core.constants.ResponseStatus;
+import com.doumee.core.exception.BusinessException;
+import com.doumee.service.business.third.model.ApiResponse;
+import com.doumee.service.business.third.model.LoginUserInfo;
+import com.doumee.service.business.third.model.PageData;
+import com.doumee.service.business.third.model.PageWrap;
 import com.doumee.core.utils.Constants;
 import com.doumee.dao.business.model.HiddenDanger;
 import com.doumee.service.business.HiddenDangerService;
@@ -16,6 +19,8 @@
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletResponse;
+import java.util.List;
+import java.util.Objects;
 
 /**
  * @author 姹熻箘韫�
@@ -34,7 +39,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 +67,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);
     }
@@ -76,7 +89,9 @@
     @CloudRequiredPermission("business:hiddendanger:exportExcel")
     public void exportExcel (@RequestBody PageWrap<HiddenDanger> pageWrap, HttpServletResponse response,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){
         pageWrap.getModel().setLoginUserInfo(this.getLoginUser(token));
-        ExcelExporter.build(HiddenDanger.class).export(hiddenDangerService.findPage(pageWrap).getRecords(), "闅愭偅淇℃伅琛�", response);
+        List<HiddenDanger> list = hiddenDangerService.findPage(pageWrap).getRecords();
+
+        ExcelExporter.build(HiddenDanger.class).export(list, "闅愭偅淇℃伅琛�", response);
     }
 
     @ApiOperation("鏍规嵁ID鏌ヨ")

--
Gitblit v1.9.3