From 108ed1c548b7dc17ce47dcadea80e4c42a2f29e7 Mon Sep 17 00:00:00 2001
From: lyfO_o <764716047@qq.com>
Date: 星期六, 16 七月 2022 17:33:17 +0800
Subject: [PATCH] 改造

---
 safePlatfrom-out-web/src/main/java/com/gkhy/safePlatform/accountController/DepartmentController.java |   20 +++++++++++---------
 1 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/safePlatfrom-out-web/src/main/java/com/gkhy/safePlatform/accountController/DepartmentController.java b/safePlatfrom-out-web/src/main/java/com/gkhy/safePlatform/accountController/DepartmentController.java
index 3bc2502..b44713c 100644
--- a/safePlatfrom-out-web/src/main/java/com/gkhy/safePlatform/accountController/DepartmentController.java
+++ b/safePlatfrom-out-web/src/main/java/com/gkhy/safePlatform/accountController/DepartmentController.java
@@ -6,9 +6,11 @@
 import com.gkhy.safePlatform.account.rpc.apimodel.model.req.DepAddRPCReqDTO;
 import com.gkhy.safePlatform.account.rpc.apimodel.model.req.DepModRPCReqDTO;
 import com.gkhy.safePlatform.account.rpc.apimodel.model.resp.DepRPCRespDTO;
+import com.gkhy.safePlatform.commons.co.ContextCacheUser;
 import com.gkhy.safePlatform.commons.enums.ResultCodes;
 import com.gkhy.safePlatform.commons.vo.ResultVO;
 import org.apache.dubbo.config.annotation.DubboReference;
+import org.springframework.security.core.Authentication;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
@@ -33,24 +35,24 @@
 
 
     @RequestMapping(value = "/add", method = RequestMethod.POST)
-    public ResultVO<String> addDepartment(Principal principal, @RequestBody DepAddRPCReqDTO depAddRPCReqDTO) {
-        String userId = principal.getName();
-        return accountDepartmentService.addDep(Long.valueOf(userId) , depAddRPCReqDTO);
+    public ResultVO<String> addDepartment(Authentication authentication, @RequestBody DepAddRPCReqDTO depAddRPCReqDTO) {
+        ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal();
+        return accountDepartmentService.addDep(currentUser.getUid(), depAddRPCReqDTO);
     }
 
 
     @RequestMapping(value = "/mod", method = RequestMethod.POST)
-    public ResultVO<String> addDepartment(Principal principal, @RequestBody DepModRPCReqDTO depModRPCReqDTO) {
-        String userId = principal.getName();
-        return accountDepartmentService.modDep(Long.valueOf(userId) , depModRPCReqDTO);
+    public ResultVO<String> addDepartment(Authentication authentication, @RequestBody DepModRPCReqDTO depModRPCReqDTO) {
+        ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal();
+        return accountDepartmentService.modDep(currentUser.getUid(), depModRPCReqDTO);
     }
 
 
     @RequestMapping(value = "/del", method = RequestMethod.POST)
-    public ResultVO<String> addDepartment(Principal principal, @RequestBody JSONObject json) {
-        String userId = principal.getName();
+    public ResultVO<String> addDepartment(Authentication authentication, @RequestBody JSONObject json) {
+        ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal();
         Long depId = json.getLong("depId");
-        return accountDepartmentService.delDep(Long.valueOf(userId) , depId);
+        return accountDepartmentService.delDep(currentUser.getUid(), depId);
     }
 
 

--
Gitblit v1.9.2