All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • spring.turbo.core.env.SpringApplicationHolders Maven / Gradle / Ivy

    package spring.turbo.core.env;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.WebApplicationType;
import org.springframework.boot.env.EnvironmentPostProcessor;
import org.springframework.boot.system.ApplicationHome;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.core.PriorityOrdered;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.lang.Nullable;
import spring.turbo.core.SpringUtils;
import spring.turbo.util.collection.CollectionUtils;

import java.nio.file.Path;
import java.util.Set;

/**
 * ApplicationContext实例保存
 *
 * @author 应卓
 * @see SpringUtils
 * @since 1.0.2
 */
public final class SpringApplicationHolders {

    @Nullable
    private static ApplicationContext APPLICATION_CONTEXT;

    @Nullable
    private static Path APPLICATION_HOME;

    @Nullable
    private static Set APPLICATION_SOURCES;

    @Nullable
    private static WebApplicationType APPLICATION_WEB_APPLICATION_TYPE;

    /**
     * 私有构造方法
     */
    private SpringApplicationHolders() {
    }

    @Nullable
    public static ApplicationContext getApplicationContext() {
        return APPLICATION_CONTEXT;
    }

    @Nullable
    public static Path getApplicationHome() {
        return APPLICATION_HOME;
    }

    @Nullable
    public static Set getApplicationSources() {
        return APPLICATION_SOURCES;
    }

    @Nullable
    public static WebApplicationType getApplicationWebApplicationType() {
        return APPLICATION_WEB_APPLICATION_TYPE;
    }

    /**
     * 被SpringBoot回调的挂钩
     *
     * @see EnvironmentPostProcessor
     * @see ApplicationListener
     */
    private static class Hook implements EnvironmentPostProcessor, ApplicationListener, PriorityOrdered {

        /**
         * 私有构造方法
         */
        private Hook() {
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
            SpringApplicationHolders.APPLICATION_HOME = getAppHomeDir(application).toAbsolutePath();
            SpringApplicationHolders.APPLICATION_SOURCES = application.getAllSources();
            SpringApplicationHolders.APPLICATION_WEB_APPLICATION_TYPE = application.getWebApplicationType();
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public void onApplicationEvent(ContextRefreshedEvent event) {
            SpringApplicationHolders.APPLICATION_CONTEXT = event.getApplicationContext();
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public int getOrder() {
            return HIGHEST_PRECEDENCE;
        }

        private Path getAppHomeDir(SpringApplication springApplication) {
            var sourceClasses = springApplication.getAllSources()
                    .stream()
                    .filter(o -> o instanceof Class)
                    .map(o -> (Class) o)
                    .toList();

            var file = CollectionUtils.size(sourceClasses) == 1 ?
                    new ApplicationHome(sourceClasses.get(0)).getDir() :
                    new ApplicationHome().getDir();

            return file.toPath();
        }
    }

}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy