java 保存csv 工具类_Java生成CSV工具类

package com.test.app.utils;

import java.io.BufferedWriter;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.OutputStreamWriter;

import java.lang.reflect.Field;

import java.lang.reflect.Method;

import java.util.List;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.util.CollectionUtils;

/**

* @Description: CSV生成工具类

* @Author: hunger.zhu

* @CreateDate: 2019/4/10 15:21

*/

public class CsvUtils {

private static final Logger logger = LoggerFactory.getLogger(CsvUtils.class);

/**

* 写出CSV文件，每一行数据为一个Object对象

* @param data List* @param header 第一行标题

* @param filePath 写出文件的绝对路径

* @param split CSV文件分隔符，一般为","

* @throws Exception

*/

public static void writeCSV(List> data, Listheader, String filePath, String split) throws Exception {

BufferedWriter bw = null;

try {

bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(filePath), "UTF-8"));

// 填充对象属性值

if (!CollectionUtils.isEmpty(header)) {

// 添加头部(首行)

for (int i = 0; i < header.size(); i++) {

bw.append(header.get(i));

if (i != header.size() - 1) {

bw.append(split);

}

}

bw.newLine();

}

// 填充数据

if (!CollectionUtils.isEmpty(data)) {

Class extends Object> objClass = data.get(0).getClass();

Field[] fields = objClass.getDeclaredFields();

for (int i = 0; i < data.size(); i++) {

for (int j = 0; j < fields.length; j++) {

Method[] methods = objClass.getDeclaredMethods();

for (Method method : methods) {

if (method.getName().equalsIgnoreCase("get" + fields[j].getName())) {

String property = (String) method.invoke(data.get(i), null);

bw.append(property == null ? "" : property);

break;

}

}

if (j != fields.length - 1) {

bw.append(split);

}

}

if (i != data.size() - 1) {

bw.newLine();

}

}

}

} catch (Exception e) {

logger.error("Write CSV exception------", e);

throw e;

} finally {

if (bw != null) {

try {

bw.close();

} catch (IOException e) {

logger.error("Close OutputStream exception------", e);

}

}

}

}

/**

* 写出CSV文件，每一行数据为一个List集合

* @param data List* @param header 第一行标题

* @param filePath 写出文件的绝对路径

* @param split CSV文件分隔符，一般为","

* @throws Exception

*/

public static void writeCSVFromList(List> data, Listheader, String filePath,

String split) throws Exception {

BufferedWriter bw = null;

try {

bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(filePath), "UTF-8"));

// 填充对象属性值

if (!CollectionUtils.isEmpty(header)) {

// 添加头部(首行)

for (int i = 0; i < header.size(); i++) {

bw.append(header.get(i));

if (i != header.size() - 1) {

bw.append(split);

}

}

bw.newLine();

}

// 填充数据

if (!CollectionUtils.isEmpty(data)) {

for (int i = 0; i < data.size(); i++) {

Listlist = data.get(i);

for (int j = 0; j < list.size(); j++) {

bw.append(list.get(j));

if (j != list.size() - 1) {

bw.append(split);

}

}

if (i != data.size() - 1) {

bw.newLine();

}

}

}

} catch (Exception e) {

logger.error("Write CSV exception------", e);

throw e;

} finally {

if (bw != null) {

try {

bw.close();

} catch (IOException e) {

logger.error("Close OutputStream exception------", e);

}

}

}

}

}