java 配置文档工具类

package tm.vision.common;

import org.apache.log4j.Logger;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Properties;
import java.util.Scanner;

/*************************************************
 <类    名>   ConfigUtil
 <功能描述>   实现从配置文件里读取关键词对应的配置信息
 **************************************************/
public final class ConfigUtils {

    private static Logger logger = Logger.getLogger(ConfigUtils.class);

    /*** 构造 ***/
    private ConfigUtils() {
    }

    /*** 从流中获取关键字及对应的配置信息 ***/
    public static String getConfig(final InputStream stream, final String key) {
        return Objects.requireNonNullElse(getProperties(stream).getProperty(key), "");
    }

    /*** 从指定文件中获取关键字及对应的配置信息 ***/
    public static String getConfig(final String fileName, final String key) {
        return Objects.requireNonNullElse(getProperties(fileName).getProperty(key), "");
    }

    /*** 把配置信息写入指定配置文件中 ***/
    public static void setConfig(final String fileName, final String key, final String val) {
        List<String> list = new ArrayList<>();
        try (Scanner scanner = new Scanner(new FileInputStream(fileName), StandardCharsets.UTF_8)) {
            while (scanner.hasNext()) {
                String string = scanner.nextLine();
                if (string.startsWith(key + "=") || string.startsWith(key + " =")) {
                    list.add(key + "=" + val);
                    continue;
                }
                list.add(string);
            }
        } catch (IOException e) {
            logger.error("read config fail, filename = " + fileName);
        }

        try (BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(new
                BufferedOutputStream(new FileOutputStream(fileName)), StandardCharsets.UTF_8))) {
            for (String string : list) {
                writer.write(string);
                if (list.indexOf(string) != list.size() - 1) {
                    writer.newLine();
                }
            }
        } catch (IOException e) {
            logger.error("write config fail, filename = " + fileName);
        }
    }

    /*** 把配置信息写入config.properties文件中 ***/
    public static void setConfig(final String key, final String val) {
        setConfig(System.getProperty("user.dir") + "/setting/config.properties", key, val);
        logger.info("key = " + key + ", val = " + val);
    }

    /*** 从config.properties文件中读取string类型的value ***/
    public static String getStringConfig(final String key, final String defaultVal) {
        String val = getConfig(System.getProperty("user.dir") + "/setting/config.properties", key);
        if (val == null || val.isBlank()) {
            val = defaultVal;
            logger.info("get config is null! key =" + key);
        }
        return val;
    }

    /*** 从config.properties文件中读取int类型的value ***/
    public static int getIntegerConfig(final String key, final int defaultVal) {
        String val = getConfig(System.getProperty("user.dir") + "/setting/config.properties", key);
        int ret = defaultVal;
        // 格式为数字
        if (val.matches("^[0-9]\\d*$")) {
            ret = Integer.parseInt(val);
        }
        return ret;
    }

    /*** 从config.properties文件中读取文件路径类型的value ***/
    public static String getPathConfig(final String key, final String defaultVal) {
        String val = getConfig(System.getProperty("user.dir") + "/setting/config.properties", key);
        if (val == null || val.isBlank() || !new File(val).exists()) {
            val = defaultVal;
        }
        return val;
    }

    /*** 通过文件名获取所有property ***/
    private static Properties getProperties(final String fileName) {
        Properties properties = new Properties();
        try (BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(
                new BufferedInputStream(new FileInputStream(fileName)), StandardCharsets.UTF_8))) {
            properties.load(bufferedReader);
        } catch (IOException e) {
            logger.error("read config fail, filename = " + fileName);
        }
        return properties;
    }

    /*** 通过stream获取所有property ***/
    private static Properties getProperties(final InputStream stream) {
        Properties properties = new Properties();

        try {
            properties.load(stream);
        } catch (IOException e) {
            logger.error("read config fail in inputstream");
        } finally {
            try {
                stream.close();
            } catch (IOException e) {
                logger.error("close read config file fail");
            }
        }
        return properties;
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值