diff --git a/src/main/java/cn/hezhaohui/thumb/constent/UserConstant.java b/src/main/java/cn/hezhaohui/thumb/constent/UserConstant.java new file mode 100644 index 0000000..d75c05d --- /dev/null +++ b/src/main/java/cn/hezhaohui/thumb/constent/UserConstant.java @@ -0,0 +1,6 @@ +package cn.hezhaohui.thumb.constent; + +public class UserConstant { + + public static final String LOGIN_USER = "login_user"; +} diff --git a/src/main/java/cn/hezhaohui/thumb/controller/UserController.java b/src/main/java/cn/hezhaohui/thumb/controller/UserController.java new file mode 100644 index 0000000..db1fdc9 --- /dev/null +++ b/src/main/java/cn/hezhaohui/thumb/controller/UserController.java @@ -0,0 +1,32 @@ +package cn.hezhaohui.thumb.controller; + +import cn.hezhaohui.thumb.common.BaseResponse; +import cn.hezhaohui.thumb.common.ResultUtils; +import cn.hezhaohui.thumb.constent.UserConstant; +import cn.hezhaohui.thumb.model.entity.User; +import cn.hezhaohui.thumb.service.UserService; +import jakarta.annotation.Resource; +import jakarta.servlet.http.HttpServletRequest; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +@RestController +@RequestMapping("user") +public class UserController { + @Resource + private UserService userService; + + @GetMapping("/login") + public BaseResponse login(long userId, HttpServletRequest request) { + User user = userService.getById(userId); + request.getSession().setAttribute(UserConstant.LOGIN_USER, user); + return ResultUtils.sucess(user); + } + + @GetMapping("/get/login") + public BaseResponse getLoginUser(HttpServletRequest request) { + User loginUser = (User) request.getSession().getAttribute(UserConstant.LOGIN_USER); + return ResultUtils.sucess(loginUser); + } +}