io.smallrye.config.ProfilePropertiesConfigSourceProvider Maven / Gradle / Ivy

Go to download
package io.smallrye.config;

import java.util.Collections;
import java.util.List;
import java.util.OptionalInt;

import org.eclipse.microprofile.config.spi.ConfigSource;
import org.eclipse.microprofile.config.spi.ConfigSourceProvider;

public class ProfilePropertiesConfigSourceProvider implements ConfigSourceProvider {
    private final List configSources;

    public ProfilePropertiesConfigSourceProvider(String propertyFileName, boolean optional, ClassLoader classLoader) {
        final ConfigurableConfigSource configurableConfigSource = new ConfigurableConfigSource(
                new ConfigSourceFactory() {
                    @Override
                    public Iterable getConfigSources(final ConfigSourceContext context) {
                        final ConfigValue value = context.getValue(ProfileConfigSourceInterceptor.SMALLRYE_PROFILE);
                        if (value != null) {
                            final String profileName = value.getValue();
                            final PropertiesConfigSourceProvider propertiesConfigSourceProvider = new PropertiesConfigSourceProvider(
                                    addProfileToPropertyFileName(propertyFileName, profileName), optional, classLoader);
                            return propertiesConfigSourceProvider.getConfigSources(classLoader);
                        }

                        return Collections.emptyList();
                    }

                    @Override
                    public OptionalInt getPriority() {
                        return OptionalInt.of(101);
                    }
                });

        this.configSources = Collections.singletonList(configurableConfigSource);
    }

    @Override
    public List getConfigSources(final ClassLoader classLoader) {
        return configSources;
    }

    private static String addProfileToPropertyFileName(final String propertyFileName, final String profile) {
        final int dot = propertyFileName.lastIndexOf(".");
        if (dot != -1) {
            return propertyFileName.substring(0, dot) + "-" + profile + propertyFileName.substring(dot);
        } else {
            return propertyFileName + "-" + profile;
        }
    }
}