【SpringBoot】工具类手动获取bean和配置的方法

众所周知，springboot中，一般通过自动注入即可获取到bean或配置，但在一些场景，没办法注入的时候，比如说抽象类设置属性。

@Slf4j
public abstract class BaseController {
    private String secureKey = null;
    private String saltKey = null;

    /**
     * *****
     method
     * *****
     */
}

此处，需要为secureKey和saltKey注入进配置的内容

data.encryption.secureKey=fdssadsadwqdsdsacaadfsafdsadas
data.sign.saltKey=we21rfcxzcdas

 

怎么办呢？于是就手写了工具类：

@SuppressWarnings("unchecked")
@Component
public class SpringUtils implements ApplicationContextAware {

    private static ApplicationContext applicationContext;
    
    private static Environment env;
    
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        SpringUtils.applicationContext = applicationContext;
    }

    public static <T> T getBean(String beanName) {
        if (applicationContext.containsBean(beanName)) {
            return (T) applicationContext.getBean(beanName);
        } else {
            return null;
        }
    }

    public static <T> Map<String, T> getBeansOfType(Class<T> baseType) {
        return applicationContext.getBeansOfType(baseType);
    }
    
    public static String readConfig(String key) {
        if(env == null) {
            env = SpringUtils.getBeansOfType(Environment.class).get("environment");
        }
         return env.getProperty(key);
    }
}

 

工具类实现ApplicationContextAware接口，重写setApplicationContext方法，注意类上的 @Component

应用启动时，spring将applicationContext对象通过setApplicationContext方法注入给静态的全局私有对象applicationContext

此时，即可使用静态方法,通过调用applicationContext的getBean来获取我们想要的bean

 

获取配置呢？配置一般通过Environment对象手动获取，Environmen对象没办法注入进来，所以readConfig方法中，读取配置前，通过单例模式从上面的getBean方法中获取Environmen，放入静态变量中。

这样既可通过SpringUntil的静态方法读取bean和配置

@Slf4j
public abstract class BaseController {
    private String secureKey = null;
 4     private String getSecureKey() {
 5         if(secureKey == null) {
 6             secureKey = SpringUtils.readConfig("data.encryption.secureKey");
 7         }
 8         return secureKey;
 9     }
    private String saltKey = null;
11     private String getSaltKey() {
12         if(saltKey == null) {
13             saltKey = SpringUtils.readConfig("data.sign.saltKey");
14         }
15         return saltKey;
16     }

    public static final String REDIS_KEY_REQUESTID = "SMSPAY_REQUESTID_stationAlipayBill_tradePay_";

    /**
     * *****
     method
     * *****
     */
}