spring boot ApplicationContextAware 的使用

柔情只为你懂 2023-07-09 02:17 42阅读 0赞

package com.xxx.utils;
    
    import com.xxx.auth.util.JwtTokenUtil;
    import com.xxx.config.properties.JwtProperties;
    import com.xxx.model.User;
    import org.apache.commons.lang3.StringUtils;
    import org.springframework.beans.BeansException;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.context.ApplicationContext;
    import org.springframework.context.ApplicationContextAware;
    import org.springframework.context.ConfigurableApplicationContext;
    import org.springframework.stereotype.Component;
    import org.springframework.web.context.request.RequestContextHolder;
    import org.springframework.web.context.request.ServletRequestAttributes;
    
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpSession;
    import java.io.UnsupportedEncodingException;
    import java.net.URLEncoder;
    import java.util.Map;
    
    @Component
    public class SpringContextUtils implements ApplicationContextAware {
        private static ConfigurableApplicationContext applicationContext;
    
        @Autowired
        private JwtTokenUtil jwtTokenUtil;
    
        @Autowired
        private JwtProperties jwtProperties;
    
        @Override
        public void setApplicationContext(ApplicationContext context) throws BeansException {
            applicationContext = (ConfigurableApplicationContext) context;
        }
    
        /**
         * 获取ApplicationContext对象
         */
        public static ApplicationContext getApplicationContext() {
            return applicationContext;
        }
    
        /**
         * 停止应用程序
         */
        public static void close() {
            if (applicationContext != null) {
                applicationContext.close();
            }
        }
    
        /**
         * 根据bean的名称获取bean
         */
        public static Object getBeanByName(String name) {
            return applicationContext.getBean(name);
        }
    
        /**
         * 根据bean的class来查找对象
         */
        public static <T> T getBeanByClass(Class<T> clazz) {
            return applicationContext.getBean(clazz);
        }
    
        /**
         * 根据bean的class来查找所有的对象（包括子类）
         */
        public static <T> Map<String, T> getBeansByClass(Class<T> c) {
            return applicationContext.getBeansOfType(c);
        }
    
        /**
         * 获取HttpServletRequest
         */
        public static HttpServletRequest getRequest() {
            ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.currentRequestAttributes();
            return attributes.getRequest();
        }
    
        /**
         * 获取HttpSession
         */
        public static HttpSession getSession() {
            ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.currentRequestAttributes();
            return attributes.getRequest().getSession();
        }
    
        /**
         * 获取完整的请求URL
         */
        public static String getRequestUrl() {
            return getRequestUrl(getRequest());
        }
    
        /**
         * 获取完整的请求URL
         */
        public static String getRequestUrl(HttpServletRequest request) {
            //当前请求路径
            String currentUrl = request.getRequestURL().toString();
            //请求参数
            String queryString = request.getQueryString();
            if (!StringUtils.isEmpty(queryString)) {
                currentUrl = currentUrl + "?" + queryString;
            }
    
            String result = "";
            try {
                result = URLEncoder.encode(currentUrl, "UTF-8");
            } catch (UnsupportedEncodingException e) {
                //ignore
            }
    
            return result;
        }
    
        public static String getRequestSsoUrl(HttpServletRequest request) {
            //当前请求路径
            if (!request.getRequestURL().toString().contains("/sso/token")) {
                return getRequestUrl(request);
            }
    
            String currentUrl = request.getContextPath().toString() + "/sso/token";
    
            //请求参数
            String queryString = request.getQueryString();
            if (!StringUtils.isEmpty(queryString)) {
                currentUrl = currentUrl + "?" + queryString;
            }
    
            String result = "";
            try {
                result = URLEncoder.encode(currentUrl, "UTF-8");
            } catch (UnsupportedEncodingException e) {
                //ignore
            }
    
            return result;
        }
    
        /**
         * 获取请求的客户端IP
         */
        public static String getRequestIp(HttpServletRequest request) {
            String ip = request.getHeader("X-Forwarded-For");
            if (StringUtils.isNoneBlank(ip) && !"unKnown".equalsIgnoreCase(ip)) {
                //多次反向代理后会有多个ip值，第一个ip才是真实ip
                int index = ip.indexOf(",");
                if (index != -1) {
                    return ip.substring(0, index);
                } else {
                    return ip;
                }
            }
            ip = request.getHeader("X-Real-IP");
            if (StringUtils.isNoneBlank(ip) && !"unKnown".equalsIgnoreCase(ip)) {
                return ip;
            }
            return request.getRemoteAddr();
        }
    
        /**
         * 获取request中token 对应的用户id,因为有过滤器的存在，会先校验request中token的有效性，所以这里就不校验了
         */
        public Integer getRequestUserId(HttpServletRequest request) {
            String requestHeader = request.getHeader(jwtProperties.getHeader());
            String authToken = requestHeader.substring(7);
            String userId = jwtTokenUtil.getUserIdFromToken(authToken);
            return Integer.parseInt(userId);
        }
    
        public User getRequestUser(HttpServletRequest request) {
            String requestHeader = request.getHeader(jwtProperties.getHeader());
            String authToken = requestHeader.substring(7);
            String userId = jwtTokenUtil.getUserIdFromToken(authToken);
            User user = new User();
            user = user.selectById(Integer.valueOf(userId));
            return user;
        }
    
    }