diff --git a/warehousing-system/project/wh-manage/src/main/java/com/wh/controller/UserController.java b/warehousing-system/project/wh-manage/src/main/java/com/wh/controller/UserController.java index 83aac3ff..6b8f8ec8 100644 --- a/warehousing-system/project/wh-manage/src/main/java/com/wh/controller/UserController.java +++ b/warehousing-system/project/wh-manage/src/main/java/com/wh/controller/UserController.java @@ -7,6 +7,8 @@ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.util.StringUtils; import org.springframework.web.bind.annotation.*; +import javax.servlet.http.HttpSession; + @RestController @CrossOrigin @RequestMapping("/user/") @@ -14,13 +16,21 @@ public class UserController { @Autowired private UserService userService; + @Autowired + private HttpSession session; @PostMapping("login") public SysResult login(@RequestBody User user){ - String token = userService.findUserByUP(user); - if (StringUtils.hasLength(token)){ - return SysResult.success(token); + String fixationToken = (String)session.getAttribute("token"); + if (StringUtils.hasLength(fixationToken)){ + return SysResult.success(fixationToken); }else { - return SysResult.fail(); + String token = userService.findUserByUP(user); + if (StringUtils.hasLength(token)){ + session.setAttribute("token",token); + return SysResult.success(token); + }else { + return SysResult.fail(); + } } } }