com.bugsnag.ScheduledTaskBeanLocator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bugsnag-spring Show documentation
'Official Bugsnag notifier for Spring applications'
There is a newer version: 3.7.2
package com.bugsnag;

import static org.springframework.scheduling.annotation.ScheduledAnnotationBeanPostProcessor.DEFAULT_TASK_SCHEDULER_BEAN_NAME;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanNotOfRequiredTypeException;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.springframework.beans.factory.NoUniqueBeanDefinitionException;
import org.springframework.beans.factory.config.AutowireCapableBeanFactory;
import org.springframework.beans.factory.config.NamedBeanHolder;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.scheduling.TaskScheduler;

import java.util.concurrent.ScheduledExecutorService;

class ScheduledTaskBeanLocator implements ApplicationContextAware {

    private ApplicationContext beanFactory;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.beanFactory = applicationContext;
    }

    TaskScheduler resolveTaskScheduler() {
        return resolveSchedulerBean(TaskScheduler.class);
    }

    ScheduledExecutorService resolveScheduledExecutorService() {
        return resolveSchedulerBean(ScheduledExecutorService.class);
    }

    /**
     * Resolves a bean, first by searching for any bean in the
     * {@link org.springframework.beans.factory.BeanFactory}
     * which matches the type, and then by returning any bean which matches the
     * given bean name. If no bean can be found which satisifies either of these
     * conditions, null will be returned.
     * 
     * This broadly follows the approach used in
     * {@link org.springframework.scheduling.annotation.ScheduledAnnotationBeanPostProcessor}.
     *
     * @param schedulerType the scheduler clas
     * @param            the bean type
     * @return the resolved bean, or null if it could not be found
     */
    private  T resolveSchedulerBean(Class schedulerType) {
        try {
            return resolveSchedulerBeanByType(schedulerType);
        } catch (NoUniqueBeanDefinitionException ex) { // multiple beans available, use name
            return resolveSchedulerBeanByName(schedulerType);
        } catch (NoSuchBeanDefinitionException ex2) {
            return null;
        }
    }

    private  T resolveSchedulerBeanByType(Class schedulerType) {
        if (this.beanFactory instanceof AutowireCapableBeanFactory) {
            AutowireCapableBeanFactory factory = (AutowireCapableBeanFactory) this.beanFactory;
            NamedBeanHolder holder = factory.resolveNamedBean(schedulerType);
            return holder.getBeanInstance();
        } else {
            return this.beanFactory.getBean(schedulerType);
        }
    }

    private  T resolveSchedulerBeanByName(Class schedulerType) {
        try {
            return beanFactory.getBean(DEFAULT_TASK_SCHEDULER_BEAN_NAME, schedulerType);
        } catch (NoSuchBeanDefinitionException exc) {
            return null;
        } catch (BeanNotOfRequiredTypeException exc) {
            return null;
        }
    }
}