import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

/**
 * @Description 一個類實現(xiàn)了ApplicationContextAware接口后，就可以獲得ApplicationContext中的所有bean
 *              用于解決某些類因為有被new出來的實例導致@Autowired失效的問題
 * @Author wxp
 * @Date 2024/7/9 12:47
 */
@Component
public class BeanUtils implements ApplicationContextAware {
    
    protected static ApplicationContext applicationContext;

    /**
     * 實現(xiàn)ApplicationContextAware接口的回調(diào)方法，設(shè)置上下文環(huán)境
     * @param arg spring上下文對象
     * @throws BeansException 拋出spring異常
     */
    @Override
    public void setApplicationContext(ApplicationContext arg) throws BeansException {
        if (applicationContext == null) {
            applicationContext = arg;
        }
    }

    /**
     * 獲取spring上下文對象
     * @return 上下文對象
     */
    public static ApplicationContext getContext() {
        return context;
    }

    /**
     * 根據(jù)beanName獲取bean
     * @param beanName bean的名稱
     * @return bean對象
     */
    public Object getBean(String beanName) {
        return context.getBean(beanName);
    }

    /**
     * 根據(jù)beanName和類型獲取bean
     * @param beanName bean名稱
     * @param clazz    bean的Class類型
     * @param <T>      bean的類型
     * @return bean對象
     */
    public <T> T getBean(String beanName, Class<T> clazz) {
        return context.getBean(beanName, clazz);
    }

    /**
     * 根據(jù)類型獲取bean
     * @param clazz bean的Class類型
     * @param <T>   bean的類型
     * @return bean對象
     */
    public <T> T getBean(Class<T> clazz) {
        return context.getBean(clazz);
    }
}

JWTUtil jwtUtil = BeanUtils.getBean(JWTUtil.class);