From 95c5a1da43d63a67e5ec35ed23b5eafabfc325f6 Mon Sep 17 00:00:00 2001 From: songhuangfeng123 <shf18767906695@163.com> Date: 星期一, 18 七月 2022 10:30:48 +0800 Subject: [PATCH] controller修改 --- emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/controller/EmergencyPlanController.java | 14 ++++++++------ 1 files changed, 8 insertions(+), 6 deletions(-) diff --git a/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/controller/EmergencyPlanController.java b/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/controller/EmergencyPlanController.java index d7b4ff1..287a79e 100644 --- a/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/controller/EmergencyPlanController.java +++ b/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/controller/EmergencyPlanController.java @@ -1,5 +1,6 @@ package com.gkhy.safePlatform.emergency.controller; +import com.gkhy.safePlatform.commons.co.ContextCacheUser; import com.gkhy.safePlatform.commons.query.PageQuery; import com.gkhy.safePlatform.commons.utils.PageUtils; import com.gkhy.safePlatform.commons.vo.ResultVO; @@ -9,6 +10,7 @@ import com.gkhy.safePlatform.emergency.query.EmergencyPlanQuery; import com.gkhy.safePlatform.emergency.service.EmergencyPlanService; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.security.core.Authentication; import org.springframework.web.bind.annotation.*; import java.security.Principal; @@ -34,9 +36,9 @@ * 应急预案新增 */ @RequestMapping(value = "/add",method = RequestMethod.POST) - public ResultVO addEmergencyPlan(Principal principal, @RequestBody EmergencyPlanReqDTO emergencyPlanReqDTO) { - String uid = principal.getName(); - return emergencyPlanService.addEmergencyPlan(Long.valueOf(uid), emergencyPlanReqDTO); + public ResultVO addEmergencyPlan(Authentication authentication, @RequestBody EmergencyPlanReqDTO emergencyPlanReqDTO) { + ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal(); + return emergencyPlanService.addEmergencyPlan(currentUser.getUid(), emergencyPlanReqDTO); } /** @@ -51,9 +53,9 @@ * 应急预案修改 */ @RequestMapping(value = "/update",method = RequestMethod.POST) - public ResultVO updateEmergencyPlan(Principal principal, @RequestBody EmergencyPlanReqDTO emergencyPlanReqDTO) { - String uid = principal.getName(); - return emergencyPlanService.updateEmergencyPlan(Long.valueOf(uid), emergencyPlanReqDTO); + public ResultVO updateEmergencyPlan(Authentication authentication, @RequestBody EmergencyPlanReqDTO emergencyPlanReqDTO) { + ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal(); + return emergencyPlanService.updateEmergencyPlan(currentUser.getUid(), emergencyPlanReqDTO); } /** -- Gitblit v1.9.2