org.sitoolkit.wt.infra.PropertyUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sit-wt-runtime Show documentation
There is a newer version: 2.3
package org.sitoolkit.wt.infra;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import java.util.TreeMap;

import org.apache.commons.beanutils.BeanUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class PropertyUtils {

    private static final Logger LOG = LoggerFactory.getLogger(PropertyUtils.class);

    private static final Map CACHE = new HashMap<>();

    public static Properties load(String resourceName, boolean ignoreResourceNotFound) {
        Properties prop = new Properties();
        URL url = null;

        if (resourceName.endsWith(".xml") || resourceName.endsWith(".properties")) {
            url = PropertyUtils.class.getResource(resourceName);
        } else {
            url = PropertyUtils.class.getResource(resourceName + ".properties");

            if (url == null) {
                url = PropertyUtils.class.getResource(resourceName + ".xml");
            }
        }

        if (url == null) {
            if (ignoreResourceNotFound) {
                return prop;
            } else {
                throw new ConfigurationException("プロパティファイルが見つかりません。" + resourceName);
            }
        }

        LOG.info("プロパティを読み込みます。{}", url);

        try {
            if (url.getFile().endsWith("properties")) {
                prop.load(url.openStream());
            } else {
                prop.loadFromXML(url.openStream());
            }
        } catch (IOException e) {
            throw new ConfigurationException(e);
        }

        return prop;
    }

    public static Map loadAsMap(String resourceName,
            boolean ignoreResourceNotFound) {
        Properties prop = load(resourceName, ignoreResourceNotFound);

        Map map = new HashMap<>();

        for (Entry entry : prop.entrySet()) {
            map.put(entry.getKey().toString().trim(), entry.getValue().toString().trim());
        }

        return map;

    }

    public static void save(Object obj, File path) {
        try (FileWriter writer = new FileWriter(path)) {

            Map map = new TreeMap<>();
            map.putAll(BeanUtils.describe(obj));

            Properties prop = new Properties();

            for (Entry entry : map.entrySet()) {
                prop.setProperty(entry.getKey(), StringUtils.defaultString(entry.getValue()));
            }

            prop.store(writer, "");

        } catch (Exception e) {
            throw new ConfigurationException("プロパティの保存で例外が発生しました", e);
        }

    }

    public static Properties loadFromPathWithCache(String path) {
        Properties prop = CACHE.get(path);

        if (prop == null) {
            prop = new Properties();
        }

        try (FileInputStream fis = new FileInputStream(path)) {

            prop.load(fis);
            CACHE.put(path, prop);

            return prop;
        } catch (IOException e) {
            throw new ConfigurationException(e);
        }

    }
}