com.fccfc.framework.common.utils.PropertyHolder.java Source code

Introduction

Here is the source code for com.fccfc.framework.common.utils.PropertyHolder.java

Source

/**
 * 
 * APDPlat - Application Product Development Platform
 * Copyright (c) 2013, ??, yang-shangchuan@qq.com
 * 
 * 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 <http://www.gnu.org/licenses/>.
 * 
 */

package com.fccfc.framework.common.utils;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;

import org.springframework.core.io.ClassPathResource;

import com.fccfc.framework.common.utils.logger.Logger;

/**
 * ?
 * 
 * @author ??
 */
public class PropertyHolder {
    private static Logger log = new Logger(PropertyHolder.class);

    private static final Map<String, String> PROPERTIES = new HashMap<>();

    static {
        init();
    }

    public static Map<String, String> getProperties() {
        return PROPERTIES;
    }

    private static void load(InputStream inputStream, Map<String, String> map) {
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream, "utf-8"))) {
            String line;
            while ((line = reader.readLine()) != null) {
                line = line.trim();
                if ("".equals(line) || line.startsWith("#")) {
                    continue;
                }
                int index = line.indexOf("=");
                if (index == -1) {
                    log.error("?" + line);
                    continue;
                }
                if (index > 0 && line.length() > index + 1) {
                    String key = line.substring(0, index).trim();
                    String value = line.substring(index + 1, line.length()).trim();
                    map.put(key, value);
                } else {
                    log.error("?" + line);
                }
            }
        } catch (IOException ex) {
            log.error("?:" + ex.getMessage());
            throw new RuntimeException(ex);
        }
    }

    /**
     * ?APDPlatLoggerImpl ??LOG.setLocale(getLogLanguage())
     * ???
     */
    private static void init() {
        String systemConfig = "/config.properties";
        ClassPathResource cr = null;
        try {
            cr = new ClassPathResource(systemConfig);
            load(cr.getInputStream(), PROPERTIES);
            log.info("?:" + systemConfig);
        } catch (Exception e) {
            log.info("?" + systemConfig + "!", e);
        }

        String extendPropertyFiles = PROPERTIES.get("extend.property.files");
        if (extendPropertyFiles != null && !"".equals(extendPropertyFiles.trim())) {
            String[] files = extendPropertyFiles.trim().split(",");
            for (String file : files) {
                try {
                    cr = new ClassPathResource(file);
                    load(cr.getInputStream(), PROPERTIES);
                    log.info("?" + file);
                } catch (Exception e) {
                    log.info("?" + file + "?", e);
                }
            }
        }
        log.info("?");
        log.info("*********************************************");
        PROPERTIES.keySet().forEach(propertyName -> {
            log.info("  " + propertyName + " = " + PROPERTIES.get(propertyName));
        });
        log.info("***********************************************************");
    }

    public static Boolean getBooleanProperty(String name) {
        return getBooleanProperty(name, null);
    }

    public static Boolean getBooleanProperty(String name, Boolean defaultValue) {
        String value = PROPERTIES.get(name);
        return CommonUtil.isNotEmpty(value) ? "true".equals(value) : defaultValue;
    }

    public static Integer getIntProperty(String name) {
        return getIntProperty(name, null);
    }

    public static Integer getIntProperty(String name, Integer defaultValue) {
        String value = PROPERTIES.get(name);
        return CommonUtil.isNotEmpty(value) ? Integer.parseInt(value) : defaultValue;
    }

    public static String getProperty(String name) {
        return PROPERTIES.get(name);
    }

    public static String getProperty(String name, String defaultValue) {
        String value = PROPERTIES.get(name);
        return CommonUtil.isNotEmpty(value) ? value : defaultValue;
    }

    public static void setProperty(String name, String value) {
        PROPERTIES.put(name, value);
    }
}