package com.yqjr.nbs.car.af.ln.util;

import com.yqjr.nbs.car.af.ln.dto.InvoiceInfoDto;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;


/**
 * txt文件讀寫工具類
 * @date 2018-03-29
 * 
*/
  
public class WriterAndReadUtil {

    /**
     * @param url 讀取文件路徑
    **/
    public List<String> read(String url) throws Exception{
        File file = new File(url);
        //FileReader fr = null;
        InputStreamReader fr = null;
        BufferedReader br = null;
        List<String> contents= new ArrayList<String>();
        try {
            fr = new InputStreamReader(new FileInputStream(file), "UTF-8");
            br = new BufferedReader(fr);
            String line = br.readLine();
            while (line != null) {
                System.out.println(line);
                contents.add(line);
                line = br.readLine();
            }
        } catch (IOException e) {
            e.printStackTrace();
            throw e;
        } finally {
            try {
                br.close();
                fr.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return contents;
    }

    /**
     * @param contents 寫入txt文件內(nèi)容
     * @param url 文件存儲(chǔ)路徑
     * @param YYYYMM 文件每行內(nèi)容中包含的時(shí)間
     */
    public static void write(List<InvoiceInfoDto> contents, String url, String YYYYMM) throws Exception{
        File file = new File(url);
        File parent = file.getParentFile(); 
        if(parent==null||!parent.exists()){ 
            parent.mkdirs(); 
        }
        FileWriter fw = null;
        BufferedWriter bw = null;
        try {
            fw = new FileWriter(file);
            bw = new BufferedWriter(fw);
            for (int i = 0; i < contents.size(); i++) {
                String content = contents.get(i).getRegisterno();
                content += "@|@";
                content += String.valueOf(contents.get(i).getCode());
                content += "@|@";
                content += YYYYMM;
                content += "@|@";
                content += contents.get(i).getEmail();
                bw.write(content);
                bw.newLine();// 光標(biāo)換行
            }
            bw.flush();// 清空緩沖區(qū)內(nèi)容
        } catch (IOException e) {
            e.printStackTrace();
            throw e;
        } finally {
            try {
                bw.close();
                fw.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 測(cè)試
     * @param args
     * @throws Exception 
     */
    public static void main(String[] args) throws Exception {
        WriterAndReadUtil writerAndReadUtil = new WriterAndReadUtil();
        String url = "E:\\read.txt";
        writerAndReadUtil.read(url);
        System.out.print("read");
        List<InvoiceInfoDto> arg = new ArrayList<InvoiceInfoDto>();
        InvoiceInfoDto l = new InvoiceInfoDto();
        l.setRegisterno("asdsads");
        l.setCode("456");
        arg.add(l);
        writerAndReadUtil.write(arg, "E:\\write.txt", "201803");
        System.out.println();
    }
}