欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

导出CSV文件工具类

程序员文章站 2022-05-26 19:19:39
...

 CSVUtils.java


import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.List;

import org.apache.commons.beanutils.BeanUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;

public class CSVUtils {
	
	/**解决乱码问题
	 * @see http://www.cnblogs.com/tangkai/p/3818084.html
	 */
	private final static byte bom[] = { (byte) 0xEF, (byte) 0xBB, (byte) 0xBF };
	private final static String charset = "UTF-8";

	/**
	 * @param titles 
	 * @param titleTags
	 * @param list
	 * @param outPutPath
	 * @param filename
	 * @return
	 */
	public static <T> void writeCSVFile(String[] titles, String[] titleTags, List<T> list, OutputStream out) {
		BufferedWriter csvWtriter = null;
		try {
			csvWtriter = new BufferedWriter(new OutputStreamWriter(out, charset), 1024);
			csvWtriter.write(new String(bom, charset));
			// 写入文件头部
			writeHead(titles, csvWtriter);
			
			// 写入文件内容
			for (T row : list) {
				writeRow(titleTags, row, csvWtriter);
			}
			csvWtriter.flush();
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			IOUtils.closeQuietly(csvWtriter);
		}
	}
	
    private static void writeHead(String[] titles, BufferedWriter csvWriter) throws IOException {
    	// 写入文件头部
    	for (String title : titles) {
    		StringBuffer sb = new StringBuffer();
    		String rowStr = sb.append("\"").append(title).append("\",").toString();
    		csvWriter.write(rowStr);
    	}
    	csvWriter.newLine();
    }
    
    private static <T> void writeRow(String[] titleTags, T row, BufferedWriter csvWriter) throws Exception {
        // 写入文件头部
        for (String tag : titleTags) {
            StringBuffer sb = new StringBuffer();
            String value = BeanUtils.getProperty(row, tag);
            if(StringUtils.isNotBlank(value)){
            	// 替换值中双引号
            	if(value.contains("\"")){
            		value = value.replace("\"", "");
            	}
            	// \t解决数字0开头字符串,0显示不出来的问题,比如邮编
            	if(value.startsWith("0") && !value.contains(".")){
            		value = value + "\t";
            	}
            }
            String rowStr = sb.append("\"").append(value==null?"":value).append("\",").toString();
            csvWriter.write(rowStr);
        }
        csvWriter.newLine();
    }
}

示例

String[] titles = new String[]{"订单号","邮箱","国家","省份","城市","街道地址","邮编",
			"电话","姓名"}; // 导出列标题
String[] titleTags = new String[]{"ordernumber","email","country","province","city","address","postalcode",
			"telephone","name"}; // 对应bean中的字段名称

List<Order> list = null;
ByteArrayOutputStream out = new ByteArrayOutputStream();
CSVUtils.writeCSVFile(titles, titleTags, list, out);