package com.example.demo.common.util;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import com.example.demo.entity.user.UserDO;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
public class UserUtil {
public static final String LOGIN_USER = "loginUser";
public static void setUser(UserDO userDO, HttpSession session) {
session.setAttribute(LOGIN_USER, userDO);
}
public static UserDO getUser(HttpSession session) {
return (UserDO)session.getAttribute(LOGIN_USER);
}
public static UserDO getUser() {
return (UserDO)getsession().getAttribute(LOGIN_USER);
}
public static HttpSession getsession() {
return getRequest().getSession();
}
public static HttpServletRequest getRequest()
{
ServletRequestAttributes requestAttributes = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes();
HttpServletRequest request = requestAttributes.getRequest();
return request;
}
}