lodsve.core.properties.ParamsHome Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lodsve-all Show documentation
lodsve全部聚合
There is a newer version: 2.7.5-RELEASE
/*
 * Copyright (C) 2018  Sun.Hao
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */

package lodsve.core.properties;

import lodsve.core.io.support.LodsveResourceLoader;
import lodsve.core.utils.StringUtils;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.EncodedResource;
import org.springframework.core.io.support.PropertiesLoaderUtils;

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

/**
 * 初始化ParamsHome路径.
 *
 * @author sunhao(sunhao.java @ gmail.com)
 * @version 1.0 2017/11/30 下午6:48
 */
public class ParamsHome {
    private static ParamsHome instance = new ParamsHome();

    private static final String ROOT_PARAM_KEY = "config.root";
    /**
     * 容器启动参数
     */
    public static final String JVM_PARAM_PATH = "params.home";
    /**
     * 环境变量
     */
    private static final String ENV_PARAM_PATH = "PARAMS_HOME";
    /**
     * web.xml中配置
     */
    static final String PARAMS_HOME_NAME = "paramsHome";
    /**
     * 此文件只配置在JVM启动参数中，key-value将覆盖原来的值，而且这里的文件需要时绝对路径[仅仅在开发环境起作用!!!]
     */
    private static final String EXT_PARAMS_FILE_NAME = "ext.params.file";

    private static final String ROOT_PARAM_FILE_NAME = "root.properties";
    private static final String PREFIX_FILE = "file:";
    private static final String PREFIX_CLASSPATH = "classpath:";
    private static final String PREFIX_ZOOKEEPER = "zookeeper:";
    private static final Properties EXT_PARAMS_PROPERTIES = new Properties();

    /**
     * 配置文件路径
     */
    private static String PARAMS_ROOT = null;

    /**
     * 构造器私有，不可在外部进行初始化实例
     */
    private ParamsHome() {

    }

    public String getParamsRoot() {
        return PARAMS_ROOT;
    }

    public void cleanParamsRoot() {
        PARAMS_ROOT = null;
    }

    public void init(String webXmlParamsPath) {
        //1.默认是环境变量
        String paramsPath = System.getenv(ENV_PARAM_PATH);

        //2.JVM启动参数获取
        if (StringUtils.isEmpty(paramsPath)) {
            paramsPath = System.getProperty(JVM_PARAM_PATH);
        }

        //3.在web.xml中配置的
        if (StringUtils.isEmpty(paramsPath)) {
            paramsPath = webXmlParamsPath;
        }

        if (StringUtils.isEmpty(paramsPath)) {
            throw new RuntimeException("读取配置文件错误！需要设置环境变量[PARAMS_HOME]或者容器启动参数[params.home]或者web.xml参数，并且环境变量 > 容器启动参数 > web.xml中配置!");
        }

        if (!StringUtils.startsWith(paramsPath, PREFIX_FILE) &&
                !StringUtils.startsWith(paramsPath, PREFIX_CLASSPATH) &&
                !StringUtils.startsWith(paramsPath, PREFIX_ZOOKEEPER)) {
            // 没有前缀的情况下，默认是file:
            paramsPath = PREFIX_FILE + paramsPath;
        }

        System.out.println(String.format("解析得到的paramsPath为【%s】", paramsPath));

        String rootPropertiesPath = paramsPath + File.separator + ROOT_PARAM_FILE_NAME;
        Resource paramsResource = new LodsveResourceLoader().getResource(org.springframework.util.StringUtils.cleanPath(rootPropertiesPath));

        Properties properties;
        try {
            properties = PropertiesLoaderUtils.loadProperties(paramsResource);
        } catch (IOException e) {
            throw new RuntimeException("加载配置文件[" + paramsResource + "]发生IO异常");
        }

        String root = properties.getProperty(ROOT_PARAM_KEY);
        if (StringUtils.isEmpty(root)) {
            throw new RuntimeException("配置文件中没有rootKey:[" + ROOT_PARAM_KEY + "]");
        }

        PARAMS_ROOT = org.springframework.util.StringUtils.cleanPath(paramsPath + File.separator + root);
        System.out.println(String.format("获取到的配置文件路径为:'%s'", PARAMS_ROOT));

        // 获取devMode的值
        if (isDevMode()) {
            initExtResources();
        }
    }

    private boolean isDevMode() {
        String filePath = PARAMS_ROOT + File.separator + "framework" + File.separator + "application.properties";
        Resource resource = new FileSystemResource(filePath);
        if (!resource.exists()) {
            return false;
        }

        Properties prop = new Properties();
        try {
            PropertiesLoaderUtils.fillProperties(prop, new EncodedResource(resource, "UTF-8"));
            return Boolean.valueOf(prop.getProperty("application.devMode"));
        } catch (IOException e) {
            return false;
        }
    }

    private static void initExtResources() {
        String filePath = System.getProperty(EXT_PARAMS_FILE_NAME);
        if (StringUtils.isBlank(filePath)) {
            return;
        }

        System.out.println(String.format("获取到%s路径为'{%s}'!", EXT_PARAMS_FILE_NAME, filePath));

        Resource resource = new FileSystemResource(filePath);
        if (!resource.exists()) {
            System.err.println(String.format("找不到外部文件'[%s]'!", filePath));
            return;
        }

        try {
            PropertiesLoaderUtils.fillProperties(EXT_PARAMS_PROPERTIES, new EncodedResource(resource, "UTF-8"));
        } catch (IOException e) {
            e.printStackTrace();
            System.err.println(String.format("读取路径为'%s'的文件失败！失败原因是'%s'!", filePath, e.getMessage()));
        }
    }

    public void coveredWithExtResource(Properties systemProperties) {
        if (EXT_PARAMS_PROPERTIES.isEmpty()) {
            return;
        }

        for (Map.Entry entry : EXT_PARAMS_PROPERTIES.entrySet()) {
            Object key = entry.getKey();
            Object value = entry.getValue();

            if (systemProperties.containsKey(key)) {
                systemProperties.put(key, value);
            }
        }
    }

    public static ParamsHome getInstance() {
        return instance;
    }
}