io.github.linyimin0812.profiler.common.settings.ProfilerSettings Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-profiler-common Show documentation

There is a newer version: 3.1.4

Show newest version

package io.github.linyimin0812.profiler.common.settings;

import io.github.linyimin0812.profiler.common.logger.LogFactory;
import io.github.linyimin0812.profiler.common.logger.Logger;
import io.github.linyimin0812.profiler.common.utils.AgentHomeUtil;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Properties;

/**
 * @author linyimin
 **/
public class ProfilerSettings {

    private final static Logger logger = LogFactory.getStartupLogger();
    private final static Properties properties = new Properties();

    static {
        loadProperties(AgentHomeUtil.home() + "config" + File.separator + "spring-startup-analyzer.properties");
    }

    public static String getProperty(String key, String defaultValue) {
        if (isNotBlank(System.getProperty(key))) {
            logger.info(ProfilerSettings.class, "Key: {} from command line, value is {}", key, System.getProperty(key));
            return System.getProperty(key);
        }

        if(properties.containsKey(key)) {
            logger.info(ProfilerSettings.class, "Key: {} from configuration file, value is {}", key, properties.getProperty(key));
            return properties.getProperty(key);
        }

        logger.info(ProfilerSettings.class, "Key: {} from default value, value is {}", key, defaultValue);

        return defaultValue;
    }

    public static String getProperty(String key) {
        if (isNotBlank(System.getProperty(key))) {
            return System.getProperty(key);
        }
        return properties.getProperty(key);
    }

    public static boolean contains(String key) {
        return properties.containsKey(key);
    }

    public static boolean isNotBlank(String text) {
        return text != null && !text.isEmpty();
    }

    public static void loadProperties(String path) {

        clear();

        try (FileInputStream fileInputStream = new FileInputStream(path)) {
            properties.load(fileInputStream);
            logger.info(ProfilerSettings.class, "loaded settings from {}", path);
        } catch (IOException e) {
            logger.error(ProfilerSettings.class, "load settings from {} error.", path, e);
        }
    }

    public static void clear() {
        properties.clear();
    }

}