spring.turbo.module.configuration.env.processor.LoadmeYamlEnvironmentPostProcessor Maven / Gradle / Ivy

Go to download
package spring.turbo.module.configuration.env.processor;

import org.springframework.boot.ConfigurableBootstrapContext;
import org.springframework.boot.SpringApplication;
import org.springframework.core.env.ConfigurableEnvironment;

import static spring.turbo.module.configuration.util.PropertySourceUtils.loadYamlFormat;

/**
 * @author 应卓
 * @since 2.2.1
 */
@SuppressWarnings("DuplicatedCode")
public class LoadmeYamlEnvironmentPostProcessor extends AbstractLoadmeEnvironmentPostProcessor {

    public LoadmeYamlEnvironmentPostProcessor(ConfigurableBootstrapContext bootstrapContext) {
        super(bootstrapContext);
        super.setOrder(HIGHEST_PRECEDENCE + 101);
    }

    @Override
    public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {

        final var option = LoadmeOption.YAML;

        if (super.isNotHandled()) {
            var pair = option.load(application);

            if (pair.nothingToRead()) {
                return;
            }

            var propertySources = environment.getPropertySources();
            var fromClzPath = loadYamlFormat(pair.getClasspathResource(), LOADME + " (classpath)");
            var fromAppHome = loadYamlFormat(pair.getApplicationHomeResource(), LOADME + " (application home)");

            int addedCount = 0;

            if (fromAppHome != null) {
                propertySources.addLast(fromAppHome);
                addedCount++;
            }

            if (fromClzPath != null) {
                propertySources.addLast(fromClzPath);
                addedCount++;
            }

            if (addedCount > 0) {
                super.setHandled(option);
            }
        }
    }

}