本地文件数据源

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.TypeReference;
import lombok.extern.log4j.Log4j2;

import java.io.*;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * 本地文件数据源
 */
@Log4j2
public class FileDataResource {
    // 数据文件中涉及的表名
    public static final String TABLES = "tab_order";

    public static void main(String[] args){
        try{
            // 初始化文件
            FileDataResource.initDataResource();
            //往文件里写内容
            Map<String,Object> data = new HashMap<String,Object>();
            data.put("orderNo","2017070601024512564");
            data.put("createTime","2017-12-06 12:11:09");
            data.put("uploadTag","1");
            //addDataToTbl("tab_order",data);
            //System.out.println(queryAllDataByTblNameAndParam("tab_order","orderNo","2017070601024512562"));
            //editDataValByTblOtherParam("tab_order","orderNo","2017070601024512563","uploadTag","8");
        }catch (Exception e){
            e.printStackTrace();
        }
    }

    /**
     * 初始化数据文件
     * @return
     */
    public static boolean initDataResource(){
        File file = getDataBaseFile();
        PrintStream pStream = null;

        try {
            if(!file.exists()){
                // 初始化创建文件
                file.createNewFile();
            }
            // 解析所有表
            String[] tables = TABLES.split(",");
            // 读取数据库文件
            String context = readFileCountInfo(file);
            if("".equals(context)){
                JSONObject jObject = new JSONObject();
                for(String tbl:tables){
                    JSONArray jArray = new JSONArray();
                    jObject.put(tbl,jArray);
                }
                // 写入文件
                pStream = new PrintStream(new FileOutputStream(file));
                pStream.print(jObject);
            }
        }catch (Exception e){
            e.printStackTrace();
            log.error(e);
        }finally {
            if(pStream!=null){
                pStream.close();
            }
        }
        return true;
    }

    /**
     * 修改表里某个数据的值
     * @param tableName 表名
     * @param keyFieldName  关键字段
     * @param keyFieldVal  关键字段的值
     * @param fieldName 要修改的字段名
     * @param changeVal 修改后值
     * @return
     */
    public static boolean editDataValByTblOtherParam(String tableName,String keyFieldName,String keyFieldVal,String fieldName,String changeVal){
        File file = getDataBaseFile();
        // 将文件内容解析成JSON数据
        JSONObject jsonObject = readFileParseToJosnObject();
        JSONArray features = jsonObject.getJSONArray(tableName);
        for (int i = 0; i < features.size(); i++) {
            JSONObject info = features.getJSONObject(i);
            // 找到关键字段关键行
            if(keyFieldVal.equals(info.getString(keyFieldName))){
                // 修改需要修改的字段值
                info.put(fieldName,changeVal);
            }
        }
        // 将内容重新写回文件里面
        writeFileContentInfo(file,jsonObject.toString());
        return true;
    }

    /**
     * 修改表里某个数据的值
     * @param tableName 表名
     * @param fieldName 字段名
     * @param beforeVal 修改前值
     * @param changeVal 修改后值
     * @return
     */
    public static boolean editDataValByTblParam(String tableName,String fieldName,String beforeVal,String changeVal){
        File file = getDataBaseFile();
        // 将文件内容解析成JSON数据
        JSONObject jsonObject = readFileParseToJosnObject();
        JSONArray features = jsonObject.getJSONArray(tableName);
        for (int i = 0; i < features.size(); i++) {
            JSONObject info = features.getJSONObject(i);
            String jsonVal = info.getString(fieldName);
            if(beforeVal.equals(jsonVal)){
                info.put(fieldName,changeVal);
            }
        }
        // 将内容重新写回文件里面
        writeFileContentInfo(file,jsonObject.toString());
        return true;
    }

    /**
     * 读取文件并将内容转为josn对象
     * @return
     */
    public static JSONObject readFileParseToJosnObject(){
        File file = getDataBaseFile();
        // 读取数据库文件
        String context = readFileCountInfo(file);
        // 将文件内容解析成JSON数据
        return JSONObject.parseObject(context);
    }

    /**
     * 往指定表名写数据
     * @param tableName 表名
     * @param data  数据内容
     * @return
     */
    public static boolean addDataToTbl(String tableName,Map<String,Object> data){
        log.info("----- addDataToTbl start -----");
        File file = getDataBaseFile();
        // 读取数据库文件
        String context = readFileCountInfo(file);
        // 将文件内容解析成JSON数据
        JSONObject json = readFileParseToJosnObject();
        ArrayList<Map<String,Object>> dataList = null;
        if(json!=null){
            // 查询指定json内容的数据
            JSONArray jsonArray = json.getJSONArray(tableName);
            jsonArray.add(data);
        }
        // 将内容重新写回文件里面
        writeFileContentInfo(file,json.toString());
        log.info("----- addDataToTbl end -----");
        return true;
    }

    /**
     * 查询对应表字段的符合数据内容
     * @param tableName 表名
     * @param fieldName 字段名
     * @param param 数据内容
     * @return
     */
    public static List<Map<String,Object>> queryAllDataByTblNameAndParam(String tableName,String fieldName,String param){
        List<Map<String,Object>> returnDateList = new ArrayList<Map<String,Object>>();
        // 查询文件内指定table的所有数据
        List<Map<String,Object>> tableDataList = queryAllDataByTblName(tableName);
        // 遍历table中的数据
        for(Map<String,Object> data:tableDataList){
            for(Map.Entry<String, Object> entry:data.entrySet()){
                // 筛选指定的内容
                if(fieldName.equals(entry.getKey().toString()) && param.equals(entry.getValue().toString())){
                    returnDateList.add(data);
                }
            }
        }
        return returnDateList;
    }

    /**
     * 根据表名查询所有数据
     * @param tableName
     * @return
     */
    public static List<Map<String,Object>> queryAllDataByTblName(String tableName){
        File file = getDataBaseFile();
        // 读取数据库文件
        String context = readFileCountInfo(file);
        // 将数据文件内容解析成json对象
        JSONObject json = JSONObject.parseObject(context);
        ArrayList<Map<String,Object>> dataList = null;
        if(json!=null){
            Object jsonArray = json.get(tableName);
            // 将获取的表内所有数据转成Map类型
            dataList = JSON.parseObject(jsonArray+"", new TypeReference<ArrayList<Map<String,Object>>>(){});
        }
        return dataList;
    }

    /**
     * 将内容写到文件中
     * @param file  读取的文件
     * @param content   需要写入的内容
     */
    public static void writeFileContentInfo(File file,String content){
        BufferedWriter bw = null;
        try {
            // 根据文件路径创建缓冲输出流
            bw = new BufferedWriter(new FileWriter(file));
            bw.write(content);
            bw.flush();
        }catch (Exception e) {
            e.printStackTrace();
            log.error(e);
        } finally {
            // 关闭流
            if (bw != null) {
                try {
                    bw.close();
                }catch (IOException e) {
                    bw = null;
                }
            }
        }
    }

    /**
     * 读取文件内容并返回文件内容String字符串
     * @param file  数据文件
     * @return
     */
    public static String readFileCountInfo(File file){
        BufferedReader reader = null;
        String dbContent = "";
        try{
            reader = new BufferedReader(new InputStreamReader(new FileInputStream(file)));
            int index = 1;
            String line = null;
            while ((line = reader.readLine()) != null) {
                dbContent += line;
                index++;
            }
        }catch(Exception e){
            e.printStackTrace();
            log.error(e);
        }finally{
            if(reader!=null){
                try {
                    reader.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return dbContent;
    }

    /**
     * 获取数据文件
     * @return
     */
    public static File getDataBaseFile(){
        String rootPath = FileDataResource.getRootPath();
        String filePath = rootPath + "/localDataBase.db";
        File file = new File(filePath);
        return file;
    }

    /**
     * 获取工程路径
     * @return
     */
    public static String getRootPath() {
        String classPath = FileDataResource.class.getResource("/").getPath();
        //String classPath = "M:\\logs";
        String rootPath  = "";
        //windows下
        if("\\".equals(File.separator)){
            int index = classPath.indexOf("/WEB-INF/classes");
            if(index==-1){
                rootPath = classPath;
            }else{
                rootPath  = classPath.substring(1,classPath.indexOf("/WEB-INF/classes"));
            }
            rootPath = rootPath.replace("/", "\\");
        }
        //linux下
        if("/".equals(File.separator)){
            int index = classPath.indexOf("/WEB-INF/classes");
            if(index==-1){
                rootPath = classPath;
            }else{
                rootPath  = classPath.substring(0,classPath.indexOf("/WEB-INF/classes"));
            }
            rootPath = rootPath.replace("\\", "/");
        }
        System.out.println("----rootPath----"+rootPath);
        return rootPath;
    }

}