spring.turbo.core.SpringContext Maven / Gradle / Ivy

Go to download
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 *    ____             _            _____           _
 *   / ___| _ __  _ __(_)_ __   __ |_   _|   _ _ __| |__   ___
 *   \___ \| '_ \| '__| | '_ \ / _` || || | | | '__| '_ \ / _ \
 *    ___) | |_) | |  | | | | | (_| || || |_| | |  | |_) | (_) |
 *   |____/| .__/|_|  |_|_| |_|\__, ||_| \__,_|_|  |_.__/ \___/
 *         |_|                 |___/   https://github.com/yingzhuo/spring-turbo
 * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
package spring.turbo.core;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.springframework.beans.factory.NoUniqueBeanDefinitionException;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.boot.ApplicationArguments;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.context.MessageSource;
import org.springframework.core.OrderComparator;
import org.springframework.core.convert.ConversionService;
import org.springframework.core.env.Environment;
import org.springframework.core.io.ResourceLoader;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.lang.Nullable;
import org.springframework.util.CollectionUtils;
import org.springframework.validation.Validator;
import spring.turbo.util.Asserts;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

/**
 * @author 应卓
 * @see ApplicationContext
 * @since 1.0.0
 */
public final class SpringContext {

    private final ApplicationContext applicationContext;

    private SpringContext(ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }

    public static SpringContext of(ApplicationContext applicationContext) {
        return new SpringContext(applicationContext);
    }

    public ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    public BeanDefinitionRegistry getBeanDefinitionRegistry() {
        return (BeanDefinitionRegistry) applicationContext.getAutowireCapableBeanFactory();
    }

    public  ObjectProvider getObjectProvider(Class beanType) {
        return applicationContext.getBeanProvider(beanType);
    }

    public ResourceLoader getResourceLoader() {
        return applicationContext;
    }

    public ResourcePatternResolver getResourcePatternResolver() {
        return applicationContext;
    }

    public ApplicationEventPublisher getApplicationEventPublisher() {
        return applicationContext;
    }

    @Nullable
    public Environment getEnvironment() {
        return getBean(Environment.class).orElse(null);
    }

    @Nullable
    public ApplicationArguments getApplicationArguments() {
        return getBean(ApplicationArguments.class).orElse(null);
    }

    @Nullable
    public ConversionService getConversionService() {
        return getBean(ConversionService.class).orElse(null);
    }

    @Nullable
    public Validator getValidator() {
        return getBean(Validator.class).orElse(null);
    }

    @Nullable
    public MessageSource getMessageSource() {
        return getBean(MessageSource.class).orElse(null);
    }

    // -----------------------------------------------------------------------------------------------------------------

    public  Optional getBean(Class beanType) {
        try {
            return Optional.of(applicationContext.getBean(beanType));
        } catch (BeansException e) {
            return Optional.empty();
        }
    }

    public  Optional getBean(Class beanType, String beanName) {
        try {
            return Optional.of(applicationContext.getBean(beanName, beanType));
        } catch (BeansException e) {
            return Optional.empty();
        }
    }

    public  List getBeanList(Class beanType) {
        try {
            final List list = new ArrayList<>(applicationContext.getBeansOfType(beanType).values());
            OrderComparator.sort(list);
            return Collections.unmodifiableList(list);
        } catch (BeansException e) {
            return Collections.emptyList();
        }
    }

    public  List getBeanList(Class beanType, T defaultIfNotNull) {
        Asserts.notNull(defaultIfNotNull);

        List list = getBeanList(beanType);
        if (CollectionUtils.isEmpty(list)) {
            list = new ArrayList<>();
            list.add(defaultIfNotNull);
        }
        return Collections.unmodifiableList(list);
    }

    public  boolean containsBean(String beanName) {
        return applicationContext.containsBean(beanName);
    }

    public boolean containsBean(Class beanType) {
        try {
            applicationContext.getBean(beanType);
            return true;
        } catch (NoUniqueBeanDefinitionException e) {
            return true;
        } catch (NoSuchBeanDefinitionException e) {
            return false;
        }
    }

}