CSV读写工具类

package util;

import com.csvreader.CsvReader;
import com.csvreader.CsvWriter;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
/**
 * 
 * CSV读写工具类
 * @className CSVUtil
 * @author Jeason
 * @date 2019年9月7日
 */
public class CSVUtil {
	
    private static char separator = ',';

    /**
     * 读取CSV文件
     * @param filename:全路径名
     */
    public static List<String[]> readCSV(String filename) throws Exception {
    	System.out.println("==================开始读取文件"+filename+"==================");
        CsvReader reader = null;
        List<String[]> dataList = new ArrayList<String[]>();
        try {
            //如果生产文件乱码,windows下用gbk,linux用UTF-8
            reader = new CsvReader(filename, separator, Charset.forName("UTF-8"));
            // 读取表头
            reader.readHeaders();
            String[] headArray = reader.getHeaders();//获取标题
            System.out.println("标题:" );
            for (String string : headArray) {
            	System.out.print(string + ",");
	    }
            System.out.println();
            // 逐条读取记录,直至读完
            System.out.println("内容:" );
            while (reader.readRecord()) {
            	String[] str = reader.getValues();
            	if (str != null && str.length > 0) {
            		dataList.add(str);
            		for (String s : str) {
            			System.out.print(s + ",");
			}
            		System.out.println();
            	}
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (null != reader) {
                reader.close();
            }
        }
        System.out.println("==================读取文件结束==================");
        return dataList;
    }

    /**
     * 生成CSV文件
     * @param dataList:数据集
     * @param filename:全路径名
     */
    public static boolean createCSV(List<String[]> dataList, String filename) throws Exception {
        boolean isSuccess = false;
        CsvWriter writer = null;
        FileOutputStream out = null;
        try {
            out = new FileOutputStream(filename, true);
            //如果生产文件乱码,windows下用gbk,linux用UTF-8
            writer = new CsvWriter(out, separator, Charset.forName("UTF-8"));
            for (String[] strs : dataList) {
                writer.writeRecord(strs);
            }
            isSuccess = true;
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (null != writer) {
                writer.close();
            }
            if (null != out) {
                try {
                    out.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return isSuccess;
    }
}

 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值