Java 读写配置文档线程安全

绝地灬酷狼 2022-05-30 02:39 134阅读 0赞

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

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);
        }
    }