diff --git a/src/main/java/cn/hezhaohui/thumb/controller/UserController.java b/src/main/java/cn/hezhaohui/thumb/controller/UserController.java index db1fdc9..e592c60 100644 --- a/src/main/java/cn/hezhaohui/thumb/controller/UserController.java +++ b/src/main/java/cn/hezhaohui/thumb/controller/UserController.java @@ -26,7 +26,6 @@ public class UserController { @GetMapping("/get/login") public BaseResponse getLoginUser(HttpServletRequest request) { - User loginUser = (User) request.getSession().getAttribute(UserConstant.LOGIN_USER); - return ResultUtils.sucess(loginUser); + return ResultUtils.sucess(userService.getLoginUser(request)); } } diff --git a/src/main/java/cn/hezhaohui/thumb/service/UserService.java b/src/main/java/cn/hezhaohui/thumb/service/UserService.java index 8574132..840d322 100644 --- a/src/main/java/cn/hezhaohui/thumb/service/UserService.java +++ b/src/main/java/cn/hezhaohui/thumb/service/UserService.java @@ -2,6 +2,7 @@ package cn.hezhaohui.thumb.service; import cn.hezhaohui.thumb.model.entity.User; import com.baomidou.mybatisplus.extension.service.IService; +import jakarta.servlet.http.HttpServletRequest; /** * @author 23117 @@ -10,4 +11,5 @@ import com.baomidou.mybatisplus.extension.service.IService; */ public interface UserService extends IService { + public User getLoginUser(HttpServletRequest request); } diff --git a/src/main/java/cn/hezhaohui/thumb/service/impl/UserServiceImpl.java b/src/main/java/cn/hezhaohui/thumb/service/impl/UserServiceImpl.java index caff796..e43506a 100644 --- a/src/main/java/cn/hezhaohui/thumb/service/impl/UserServiceImpl.java +++ b/src/main/java/cn/hezhaohui/thumb/service/impl/UserServiceImpl.java @@ -1,9 +1,11 @@ package cn.hezhaohui.thumb.service.impl; +import cn.hezhaohui.thumb.constent.UserConstant; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import cn.hezhaohui.thumb.model.entity.User; import cn.hezhaohui.thumb.service.UserService; import cn.hezhaohui.thumb.mapper.UserMapper; +import jakarta.servlet.http.HttpServletRequest; import org.springframework.stereotype.Service; /** @@ -15,6 +17,11 @@ import org.springframework.stereotype.Service; public class UserServiceImpl extends ServiceImpl implements UserService{ + + @Override + public User getLoginUser(HttpServletRequest request) { + return (User) request.getSession().getAttribute(UserConstant.LOGIN_USER); + } }