csv導出導入工具類 commons-csv導出

需要在pom文件添加commons-csv

       <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-csv</artifactId>
            <version>1.6</version>
        </dependency>

 

 

package utils;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVPrinter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * csv導入導出工具類
 * @author kpzc
 * @date 2018年12月29日 下午3:45:00
 * @desc 
 */
public class CSVUtils {
   // private static final Logger logger = LoggerFactory.getLogger(CSVUtils.class);

    /**
     * io流導出
     * @author kpzc
     * @date 2018年12月29日 下午3:48:34
     * @param file csv文件(路徑+文件名)，csv文件不存在會自動創建
     * @param dataList 數據，字符串用逗號分隔
     * @return 返回導出是否成功 true成功 false失敗
     */
    public static boolean exportCsv(File file, List<String> dataList){
        boolean isSucess=false;
        
        FileOutputStream out=null;
        OutputStreamWriter osw=null;
        BufferedWriter bw=null;
        try {
            out = new FileOutputStream(file);
            osw = new OutputStreamWriter(out, "GBK");//解決FileOutputStream中文亂碼問題  解決MS office亂碼問題
            bw =new BufferedWriter(osw);
            if(dataList!=null && !dataList.isEmpty()){
                for(String data : dataList){
                    bw.append(data).append("\r");
                }
            }
            isSucess=true;
        } catch (Exception e) {
            isSucess=false;
        }finally{
            if(bw!=null){
                try {
                    bw.close();
                    bw=null;
                } catch (IOException e) {
                    e.printStackTrace();
                } 
            }
            if(osw!=null){
                try {
                    osw.close();
                    osw=null;
                } catch (IOException e) {
                    e.printStackTrace();
                } 
            }
            if(out!=null){
                try {
                    out.close();
                    out=null;
                } catch (IOException e) {
                    e.printStackTrace();
                } 
            }
        }
        return isSucess;
    }
    
    /**
     * 導入
     * @author kpzc
     * @date 2018年12月29日 下午3:48:11
     * @param file  csv文件(路徑+文件)
     * @return 返回List<String>列表
     */
    public static List<String> importCsv(File file){
        List<String> dataList=new ArrayList<String>();
        BufferedReader br=null;
        try { 
            br = new BufferedReader(new FileReader(file));
            String line = ""; 
            while ((line = br.readLine()) != null) { 
                dataList.add(line);
            }
        }catch (Exception e) {
            e.printStackTrace();
        }finally{
            if(br!=null){
                try {
                    br.close();
                    br=null;
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return dataList;
    }
    
    /**
     * apache commons-csv導出
     * 注意jdk要在1.7及以上使用
     * map的數據個數要與header的個數相等 並且一一對應，可參照main方法
     * @author kpzc
     * @date 2019年1月4日 上午10:12:20
     * @param filePath 文件存儲路徑
     * @param list 數據列表
     * @param header 表頭
     */
    public static void write(String filePath,List<LinkedHashMap<String, String>> list,String... header) {
        try {
            FileOutputStream fos = new FileOutputStream(filePath);
            OutputStreamWriter osw = new OutputStreamWriter(fos, "GBK");
            CSVFormat csvFormat = CSVFormat.DEFAULT.withHeader(header);
            CSVPrinter csvPrinter = new CSVPrinter(osw, csvFormat);
            //跟上面兩行代碼是一樣的
            //CSVPrinter csvPrinter = CSVFormat.DEFAULT.withHeader(header).print(osw);
            for (Map<String, String> map : list) {
                csvPrinter.printRecord(map.values());
            }
            csvPrinter.flush();
            csvPrinter.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    
    public static void main(String[] args) {
        //io流導出
        File file=new File("E:/aa.csv");
        List<String> dataList=new ArrayList<String>();
        dataList.add("1,2,3,'/N',4");
        dataList.add("1,2,3,'/N',4");
        exportCsv(file, dataList);
        
        // apache commons-csv導出
        String filePath = "E://aaa.csv";
        String header1 = "姓名";
        String header2 = "性別";
        String header3 = "編號";
        String header4 = "描述";
        
        List<LinkedHashMap<String, String>> recordList = new ArrayList<LinkedHashMap<String, String>>();
        for (int i = 0; i < 5; i++) {
            LinkedHashMap<String, String> map = new LinkedHashMap<String, String>();
            map.put("name", "zhangsan");
            map.put("sex", "男");
            map.put("code", "001");
            map.put("aa", "aaa");
            recordList.add(map);
        }
        write(filePath, recordList,header1,header2,header3,header4);
    }
}