diff --git a/backend/src/main/java/com/designhub/controller/TaskProductController.java b/backend/src/main/java/com/designhub/controller/TaskProductController.java index 6e3d192..988ebc6 100644 --- a/backend/src/main/java/com/designhub/controller/TaskProductController.java +++ b/backend/src/main/java/com/designhub/controller/TaskProductController.java @@ -39,7 +39,8 @@ public class TaskProductController { @PathVariable Long taskId, @RequestBody Map request, @RequestHeader(value = "Authorization", required = false) String authHeader) { - Long userId = jwtUtil.getUserIdFromHeader(authHeader); + String token = jwtUtil.extractTokenFromHeader(authHeader); + Long userId = jwtUtil.getUserIdFromToken(token); String productId = request.get("productId"); String productName = request.get("productName"); String remark = request.get("remark"); @@ -54,7 +55,8 @@ public class TaskProductController { @PathVariable Long taskId, @RequestBody Map> request, @RequestHeader(value = "Authorization", required = false) String authHeader) { - Long userId = jwtUtil.getUserIdFromHeader(authHeader); + String token = jwtUtil.extractTokenFromHeader(authHeader); + Long userId = jwtUtil.getUserIdFromToken(token); List productIds = request.get("productIds"); int count = taskProductService.addProducts(taskId, productIds, userId); @@ -68,7 +70,8 @@ public class TaskProductController { @PathVariable Long id, @RequestBody Map request, @RequestHeader(value = "Authorization", required = false) String authHeader) { - Long userId = jwtUtil.getUserIdFromHeader(authHeader); + String token = jwtUtil.extractTokenFromHeader(authHeader); + Long userId = jwtUtil.getUserIdFromToken(token); String productId = request.get("productId"); String productName = request.get("productName"); String remark = request.get("remark"); @@ -83,7 +86,8 @@ public class TaskProductController { @PathVariable Long taskId, @PathVariable Long id, @RequestHeader(value = "Authorization", required = false) String authHeader) { - Long userId = jwtUtil.getUserIdFromHeader(authHeader); + String token = jwtUtil.extractTokenFromHeader(authHeader); + Long userId = jwtUtil.getUserIdFromToken(token); taskProductService.deleteProduct(id, userId); return Result.success(); }