在定时任务中注入Service或者mapper

ゝ一世哀愁。 2023-06-23 13:13 14阅读 0赞

## 1.使用ApplicationContextProvider ##

package com.nari.iot.common.config;
    
    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.stereotype.Component;
    
    /**
     * @ProjectName:
     * @Package: com.backstage.config
     * @ClassName: ApplicationContextProvider
     * @Description: 获取bean对象的工具类
     * @Author: zmj
     * @Version: 1.0
     */
    
    /**
     * Author:zmj
     * 提供bean注入的另一种实现
     */
    @Component
    public class ApplicationContextProvider implements ApplicationContextAware {
        /**
         * 上下文对象实例
         */
        @Autowired
        private static ApplicationContext applicationContext;
    
        @Override
        public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
            this.applicationContext = applicationContext;
        }
    
        /**
         * 获取applicationContext
         *
         * @return
         */
        public static ApplicationContext getApplicationContext() {
            return applicationContext;
        }
    
        /**
         * 通过name获取 Bean.
         *
         * @param name
         * @return
         */
        public static Object getBean(String name) {
            return getApplicationContext().getBean(name);
        }
    
        /**
         * 通过class获取Bean.
         *
         * @param clazz
         * @param <T>
         * @return
         */
        public static <T> T getBean(Class<T> clazz) {
            return getApplicationContext().getBean(clazz);
        }
    
        /**
         * 通过name,以及Clazz返回指定的Bean
         *
         * @param name
         * @param clazz
         * @param <T>
         * @return
         */
        public static <T> T getBean(String name, Class<T> clazz) {
            return getApplicationContext().getBean(name, clazz);
        }
    }

## 2.在定时任务中引入 ##

private IotAirInfoService iotAirInfoService = ApplicationContextProvider.getBean(IotAirInfoService.class);

## 3.修改启动类 ##

public class IotApplicatiopn extends SpringBootServletInitializer {
        public static void main(String[] args) {
            SpringApplication.run(IotApplicatiopn.class);
        }
        @Override
        protected SpringApplicationBuilder configure(SpringApplicationBuilder builder) {
            return builder.sources(IotApplicatiopn.class);
        }
    }