解析CSV文件工具

CSV文件可以用Excel打开，其格式类似于二维数组，格式如下所示：
，具体实体文件可见附件“CSV文件样例”。


下面是解析类的主要代码，主要用于将CSV文件解析成为List<Bean>结构

package com.zhengtian.study.tools.util;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import net.sf.json.JSONObject;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.csvreader.CsvReader;

/**
 * $Id: CsvUtil.java 276213 2013-01-06 10:55:38Z zhengtian $
 * 
 * @author zhengtian
 * @time 2013-1-2
 */
public class CsvUtil {
	private static final Logger log = LoggerFactory.getLogger(CsvUtil.class);

	// csv文件格式
	private static final String csv_charset = "SJIS";
	// csv文件分隔符(逗号)
	public static final String csv_split_comma = ",";

	/**
	 * 将CSV文件转换为Bean列表
	 * 
	 * @param csvFilePath
	 *            CSV文件的本路路径
	 * @param csv_split
	 *            CSV文件中各个字段间的分隔符
	  * @param isReadHead
	 *            是否读取CSV文件头部
	 * @param propMapping
	 *            CSV文件字段顺序(从0开始)与Bean字段的映射
	 * @param clazz
	 *            Bean的class
	 * @return
	 */
	public static <T> List<T> getBeanList(String csvFilePath, String csv_split, boolean isReadHead, Map<Integer, String> propMapping,
			Class<T> clazz) {
		List<T> list = new ArrayList<T>();

		List<String[]> records = readCsv(csvFilePath, csv_split,isReadHead);
		if (records == null || records.size() == 0) {

		} else {
			for (String[] record : records) {
				T bean = convertCsvToBean(record, propMapping, clazz);
				if (bean != null) {
					list.add(bean);
				}
			}
		}

		return list;
	}

	/**
	 * 读取CSV文件
	 * 
	 * @param csvFilePath
	 *            CSV文件服务器本地路径
	 * @param csv_split
	 *            CSV文件分隔符
	 * @param isReadHead
	 *            是否读取CSV文件头部
	 * @return
	 */
	private static List<String[]> readCsv(String csvFilePath, String csv_split, boolean isReadHead) {
		List<String[]> csvList = new ArrayList<String[]>();
		if (new File(csvFilePath).exists()) {
			// 读取文件
			CsvReader reader = null;
			try {
				reader = new CsvReader(csvFilePath, csv_split.toCharArray()[0], Charset.forName(csv_charset));
				// 跳过表头
				if (isReadHead) {
					reader.readHeaders();
				}
				// 逐行读入除表头的数据
				while (reader.readRecord()) {
					csvList.add(reader.getValues());
				}
			} catch (FileNotFoundException e) {
				log.error("", e);
			} catch (IOException e) {
				log.error("", e);
			} finally {
				// 关闭流
				reader.close();
			}
		}
		return csvList;
	}

	/**
	 * 将CSV对象转换为Bean
	 * 
	 * @param records
	 * @param propMapping
	 * @param clazz
	 * @return
	 */
	public static <T> T convertCsvToBean(String[] records, Map<Integer, String> propMapping, Class<T> clazz) {
		if (records == null || records.length == 0 || propMapping == null || propMapping.size() == 0) {
			return null;
		}
		// 组装JSON对象
		JSONObject jsonObj = new JSONObject();
		for (int i = 0; i < records.length; i++) {
			String record = records[i];

			if (StringUtils.isEmpty(record)) {
				continue;
			}
			Integer key = new Integer(i);
			if (!propMapping.containsKey(key)) {
				continue;
			}
			JsonUtils.setValue(jsonObj, propMapping.get(key), record);
		}
		return (T) JSONObject.toBean(jsonObj, clazz);
	}
}

该类需要导入jar包，具体附件中有，直接下载即可，现在将pom文件贴出来，如下所示：

<dependency>
			<groupId>commons-httpclient</groupId>
			<artifactId>commons-httpclient</artifactId>
			<version>3.1</version>
		</dependency>
		<dependency>
			<groupId>net.sf.json-lib</groupId>
			<artifactId>json-lib</artifactId>
			<version>2.4</version>
			<classifier>jdk15</classifier>
		</dependency>
		<dependency>
			<groupId>net.sourceforge.javacsv</groupId>
			<artifactId>javacsv</artifactId>
			<version>2.0</version>
		</dependency>
			<dependency>
			<groupId>org.slf4j</groupId>
			<artifactId>slf4j-log4j12</artifactId>
			<version>1.6.1</version>
			<exclusions>
				<exclusion>
					<artifactId>log4j</artifactId>
					<groupId>log4j</groupId>
				</exclusion>
			</exclusions>
		</dependency>

评论

2 楼 zheng12tian 2013-11-23  

xfbxag 写道

JsonUtils能分享下吗

不好意思，换公司了，代码没了

1 楼 xfbxag 2013-11-06  

JsonUtils能分享下吗