package cc.mrbird.febs.common.utils;

import cc.mrbird.febs.mall.entity.MallMember;
import cc.mrbird.febs.user.entity.ChatMember;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;

/**
 * @author xxx
 * @date 2020-08-24
 **/
public class LoginUserUtil {

    public static MallMember getLoginUser() {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
        return (MallMember) request.getSession().getAttribute("member");
    }

    public static ChatMember getLoginChatMember() {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
        return (ChatMember) request.getSession().getAttribute("chatMember");
    }
}