commons-csv的基本操作

本文主要介绍一些关系CSV的基本操作。

commons-csv 由两大核心对象组成 CSVParser（解析），CSVPrinter（写csv）下面附上工具类，以及测试类

package com.csv.demo;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVPrinter;

public class CSVUtils
{
	public static CSVParser getCSVParser(String filePath) throws IOException
	{
		CSVFormat format = CSVFormat.DEFAULT.withHeader();
		InputStreamReader isr = new InputStreamReader(new FileInputStream(filePath), "UTF-8");
		return new CSVParser(isr, format);
	}

	public static CSVPrinter getCSVPrinter(String filePath) throws IOException
	{
		CSVFormat format = CSVFormat.DEFAULT.withRecordSeparator("\n");
		OutputStreamWriter osw = new OutputStreamWriter(new FileOutputStream(filePath), "UTF-8");
		return new CSVPrinter(osw, format);
	}
}

测试类：演示将一个CSV写到另一个CSV

package com.csv.demo;

import java.io.IOException;
import java.util.Iterator;

import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVPrinter;
import org.apache.commons.csv.CSVRecord;
import org.apache.commons.io.IOUtils;

public class Test
{
	public static void main(String[] args) throws IOException
	{
		CSVParser parser = CSVUtils.getCSVParser("D://demo01.csv");
		CSVPrinter printer=CSVUtils.getCSVPrinter("D://demo02.csv");
		Iterator<CSVRecord> iterator = parser.iterator();
		printer.printRecord(parser.getHeaderMap().keySet());//写CSV第一行
		while(iterator.hasNext())
		{
			printer.printRecord(iterator.next());
		}
		IOUtils.closeQuietly(parser);
		IOUtils.closeQuietly(printer);
	}
}

CSVRecord有一个比较常用的方法get(String name)，可以选取CSV的某一列进行读写，只需指定首列字段名称即可。