package com.xcong.farmer.cms.modules.system.util;

import com.xcong.farmer.cms.modules.system.entity.UserEntity;
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 UserEntity getLoginUser() {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();
        return (UserEntity) request.getSession().getAttribute("user");
    }

    public static Long getCompanyId() {
        return getLoginUser().getCompanyId() == null ? 0L : getLoginUser().getCompanyId();
    }
}