代码优化

linwe

2024-07-29 aeb7d068be92312dcdcea75e1240bcf2a78dd0fe

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package org.springblade.common.utils;
 
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
 
import javax.servlet.http.HttpServletRequest;
 
/**
 * spring 工具类
 * @author zhongrj
 */
@Component
public class SpringUtils implements ApplicationContextAware {
 
 
    private static ApplicationContext applicationContext;
 
    @Override
    public void setApplicationContext(ApplicationContext applicationContext)throws BeansException {
        if (SpringUtils.applicationContext == null) {
            SpringUtils.applicationContext = applicationContext;
        }
    }
 
    public static HttpServletRequest getRequestHeaders(){
        ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        // get the request
        HttpServletRequest request = requestAttributes.getRequest();
        // 返回request 对象
        return request;
    }
 
 
    /**
     * 获取角色别名
     * @param key
     * @return
     */
    public static String getRequestParam(String key){
        // get the request
        HttpServletRequest requestHeaders = getRequestHeaders();
        // 返回request 对象
        return requestHeaders.getHeader(key);
    }
 
    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }
 
 
    public static Object getBean(String name) {
        return getApplicationContext().getBean(name);
    }
 
 
    public static <T> T getBean(Class<T> clazz) {
        return getApplicationContext().getBean(clazz);
    }
 
    public static <T> T getBean(String name, Class<T> clazz) {
        return getApplicationContext().getBean(name, clazz);
    }
 
    /**
     * 获取当前环境
     */
    public static String getActiveProfile() {
        return applicationContext.getEnvironment().getActiveProfiles()[0];
    }
}