Java 读写配置文档 线程安全

因为业务需求不能使用数据库, 只能用文档存储数据, 但是又要保证线程安全. 所以就使用读写锁去完成这个需求, 代码如下:

package com.lb.template.util;

import java.io.BufferedInputStream;
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.OutputStream;
import java.util.Properties;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;

/**
 * 文件存储帮助类, 使用读写锁保证数据一致性
 *
 * @author admin
 *
 */
public class PropertyHelper {

    // TODO 后续得加自动编码判断
    /**
     * 配置文件编码
     */
    private static final String CONFIG_FILE_CODE = "GBK";

    /**
     * 文件路径
     */
    private String storeFilePath;

    private Properties props = new Properties();

    /**
     * 使用读写锁, 保证 读读共享, 读写互斥, 写写互斥
     */
    private ReadWriteLock rwl = new ReentrantReadWriteLock();

    /**
     * 构造的时候读取配置文件
     *
     * @param filePath
     */
    public PropertyHelper(String filePath) {
        this.storeFilePath = filePath;

        File storeFile = new File(filePath);

        if (storeFile.exists()) {
            try {
                InputStream in = new BufferedInputStream(new FileInputStream(filePath));
                props.load(new InputStreamReader(in, CONFIG_FILE_CODE));
            } catch (Exception e) {
                System.out.println("数据存储文件加载失败, 系统退出........" + e);
                System.exit(0);
            }
        } else {
            try {
                storeFile.createNewFile();
            } catch (IOException e) {
                System.out.println("数据存储文件创建失败, 系统退出........." + e);
                System.exit(0);
            }
        }
    }

    /**
     * 读取属性文件中相应键的值
     *
     * @param key
     *            主键
     * @return String
     */
    public String get(String key) {
        rwl.readLock().lock();

        try {
            return props.getProperty(key);
        } finally {
            rwl.readLock().unlock();
        }
    }

    /**
     * 更新（或插入）一对properties信息(主键及其键值) 如果该主键已经存在，更新该主键的值； 如果该主键不存在，则插件一对键值。
     *
     * @param keyname
     *            键名
     * @param keyvalue
     *            键值
     */
    public void putOrUpdate(String keyname, String keyvalue) {
        rwl.writeLock().lock();

        try {
            // 调用 Hashtable 的方法 put，使用 getProperty 方法提供并行性。
            // 强制要求为属性的键和值使用字符串。返回值是 Hashtable 调用 put 的结果。
            OutputStream fos = new FileOutputStream(storeFilePath);
            props.setProperty(keyname, keyvalue);
            // 以适合使用 load 方法加载到 Properties 表中的格式，
            // 将此 Properties 表中的属性列表（键和元素对）写入输出流
            props.store(fos, "Update '" + keyname + "' value");
        } catch (IOException e) {
            System.err.println("属性文件更新错误");
        } finally {
            rwl.writeLock().unlock();
        }
    }

    // 测试代码
    public static void main(String[] args) {
        String filePath = "C:\\Users\\admin\\Desktop\\数据导入\\xml\\数据拷贝\\custom.txt";
        PropertyHelper fileStoreHelper = new PropertyHelper(filePath);
        String string = fileStoreHelper.get("insertSql");
        System.out.println(string);
    }
}