RequestUtils.java

package org.nafmii.common.utils;

import org.apache.commons.lang.StringUtils;
import org.nafmii.common.constant.MessageEnum;
import org.nafmii.common.constant.HttpClientHeaderEnum;
import org.nafmii.common.exception.BusinessException;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Map;


public class RequestUtils {

    public static HttpServletRequest getRequest() {
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        return (HttpServletRequest) requestAttributes.resolveReference(RequestAttributes.REFERENCE_REQUEST);
    }

    public static HttpServletResponse getResponse() {
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletResponse response = servletRequestAttributes.getResponse();
        return response;
    }

    public static Long getUserId(){
        //token 获取用户信息
        String token = getToken();
        if (StringUtils.isBlank(token)){
            throw  new BusinessException("令牌不能为空");
        }
        Map<String,Object> map  = (Map<String, Object>) JwtUtils.getUserInfo(token).get(HttpClientHeaderEnum.USER_VO.getCode());
        String userId = map.get(HttpClientHeaderEnum.USER_HEADER_USER_ID.getCode()).toString();
        if (StringUtils.isNotBlank(userId)){
            try {
                return Long.valueOf(userId);
            } catch (NumberFormatException e) {
                return null;
            }
        }
        throw  new BusinessException("用户主键不能为空");
    }

    public static String getToken(){
        String token = getRequest().getHeader(HttpClientHeaderEnum.USER_HEADER_USER_TOKEN.getCode());
        return StringUtils.isEmpty(token) ? null : token;
    }

    /**
     * 校验是否登录，登录返回用户ID
     * @return Long
     */
    public static Long checkLoginStatus(){
        Long userId = getUserId();
        String token = getToken();
        if (userId == null || StringUtils.isEmpty(token)){
            throw new BusinessException(MessageEnum.NO_LOGIN.getCode(), MessageEnum.NO_LOGIN.getRemarks());
        }
        return userId;
    }
}