com.github.zeldigas.spring.env.HoconPropertySourceLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-hocon-property-source Show documentation

Property source loader for Spring that add support for HOCON format files as property sources

There is a newer version: 0.4.0

Show newest version

package com.github.zeldigas.spring.env;

import com.typesafe.config.*;
import org.springframework.boot.env.PropertySourceLoader;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.Resource;

import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedHashMap;
import java.util.Map;

/**
 * Strategy to load '.conf' files in
 * HOCON
 * format into a {@link PropertySource}.
 */
public class HoconPropertySourceLoader implements PropertySourceLoader {

    @Override
    public String[] getFileExtensions() {
        return new String[]{"conf"};
    }

    @Override
    public PropertySource load(String name, Resource resource, String profile) throws IOException {
        if (profile != null) {
            return null;
        }
        Config config = ConfigFactory.parseReader(new InputStreamReader(resource.getInputStream()),
                ConfigParseOptions.defaults().setSyntax(ConfigSyntax.CONF))
                .resolve();
        LinkedHashMap properties = new LinkedHashMap<>();
        toFlatMap(properties, config);
        return new MapPropertySource(name, properties);
    }

    private void toFlatMap(LinkedHashMap properties, Config config) {
        toFlatMap(properties, "", config);
    }

    private void toFlatMap(Map properties, String key, Config config) {
        final String prefix = "".equals(key) ? "" : key + ".";

        for (Map.Entry entry : config.entrySet()){
            String propertyKey = prefix + entry.getKey();
            ConfigValue value = entry.getValue();
            addConfigValue(properties, propertyKey, value);
        }
    }

    private void addConfigValue(Map properties, String key, ConfigValue value) {
        if (value instanceof ConfigList) {
            processListValues(properties, key, (ConfigList) value);
        } else if (value instanceof ConfigObject) {
            processObjectValues(properties, key, (ConfigObject) value);
        } else {
            addScalarValue(properties, key, value);
        }
    }

    private void processListValues(Map properties, String key, ConfigList value) {
        int i = 0;
        for (ConfigValue element : value) {
            addConfigValue(properties, String.format("%s[%d]", key, i++), element);
        }
    }

    private void processObjectValues(Map properties, String key, ConfigObject value) {
        toFlatMap(properties, key, value.toConfig());
    }

    private void addScalarValue(Map properties, String key, ConfigValue value) {
        properties.put(key, value.unwrapped());
    }

}