因为明天要用,所以匆匆记录了下,后期补充; 主要参照:
package opencsv;
import com.opencsv.CSVWriter;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.lang.reflect.Field;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
/**
* @Description
* @Author T480
* @Version
* @Date 2021/8/27
*/
public class ExportByOpenCsvUtils {
/**
* 导出Csv
*
* @param csvName
* 要导出的excel名称
* @param list
* 要导出的数据集合
* @param fieldMap
* 中英文字段对应Map,即要导出的excel表头
* @param response
* 使用response可以导出到浏览器
* @return
*/
public static <T> void export(String path, String csvName, List<T> list, LinkedHashMap<String, String> fieldMap) {
// 设置默认文件名为当前时间:年月日时分秒
if (StringUtils.isBlank(csvName)) {
csvName = new SimpleDateFormat("yyyy-MM-dd hhmmss").format(new Date()).toString();
}else{
csvName = csvName + new SimpleDateFormat("_yyyy-MM-dd hhmmss").format(new Date()).toString();
}
File csv = new File(path);
if (!csv.exists()) {
try {
csv.createNewFile();
} catch (IOException e) {
e.printStackTrace();
}
}
try {
CSVWriter writer = new CSVWriter(new OutputStreamWriter(new FileOutputStream(csv), "GBK"),
CSVWriter.DEFAULT_SEPARATOR, CSVWriter.DEFAULT_QUOTE_CHARACTER);
fillCsv(writer, list, fieldMap);
writer.flush();
writer.close();
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 根据字段名获取字段对象
*
* @param fieldName
* 字段名
* @param clazz
* 包含该字段的类
* @return 字段
*/
private static Field getFieldByName(String fieldName, Class<?> clazz) {
//logger.info("根据字段名获取字段对象:getFieldByName(), fieldName: "+fieldName);
// 拿到本类的所有字段
Field[] selfFields = clazz.getDeclaredFields();
// 如果本类中存在该字段,则返回
for (Field field : selfFields) {
// 如果本类中存在该字段,则返回
if (field.getName().equals(fieldName)) {
return field;
}
}
// 否则,查看父类中是否存在此字段,如果有则返回
Class<?> superClazz = clazz.getSuperclass();
if (superClazz != null && superClazz != Object.class) {
// 递归
return getFieldByName(fieldName, superClazz);
}
// 如果本类和父类都没有,则返回空
return null;
}
/**
* 根据字段名获取字段值
*
* @param fieldName 字段名
* @param o 对象
* @return 字段值
* @throws Exception 异常
*
*/
private static Object getFieldValueByName(String fieldName, Object o) throws Exception {
//logger.info("根据字段名获取字段值:getFieldValueByName()");
Object value = null;
// 根据字段名得到字段对象
Field field = getFieldByName(fieldName, o.getClass());
// 如果该字段存在,则取出该字段的值
if (field != null) {
field.setAccessible(true);// 类中的成员变量为private,在类外边使用属性值,故必须进行此操作
value = field.get(o);// 获取当前对象中当前Field的value
} else {
throw new Exception(o.getClass().getSimpleName() + "类不存在字段名 " + fieldName);
}
System.out.println("value: "+ value);
return value;
}
/**
* 根据带路径或不带路径的属性名获取属性值,即接受简单属性名,
* 如userName等,又接受带路径的属性名,如student.department.name等
*
* @param fieldNameSequence 带路径的属性名或简单属性名
* @param o 对象
* @return 属性值
* @throws Exception 异常
*
*/
private static Object getFieldValueByNameSequence(String fieldNameSequence, Object o) throws Exception {
//logger.info("根据带路径或不带路径的属性名获取属性值,即接受简单属性名:getFieldValueByNameSequence(): "+fieldNameSequence);
Object value = null;
// 将fieldNameSequence进行拆分
String[] attributes = fieldNameSequence.split("\\.");
if (attributes.length == 1) {
value = getFieldValueByName(fieldNameSequence, o);
} else {
// 根据数组中第一个连接属性名获取连接属性对象,如student.department.name
Object fieldObj = getFieldValueByName(attributes[0], o);
// 截取除第一个属性名之后的路径
String subFieldNameSequence = fieldNameSequence.substring(fieldNameSequence.indexOf(".") + 1);
// 递归得到最终的属性对象的值
value = getFieldValueByNameSequence(subFieldNameSequence, fieldObj);
}
return value;
}
/**
* 向csv中填充数据
*
* @param writer csv文件的名称
* @param list 数据源
* @param fieldMap 中英文字段对应关系的Map
* @throws Exception 异常
*
*/
private static <T> void fillCsv(CSVWriter writer, List<T> list, LinkedHashMap<String, String> fieldMap) throws Exception {
//logger.info("向csv file中填充数据");
// 定义存放英文字段名和中文字段名的数组
String[] enFields = new String[fieldMap.size()];
String[] cnFields = new String[fieldMap.size()];
// 填充数组
int count = 0;
for (Map.Entry<String, String> entry : fieldMap.entrySet()) {
enFields[count] = entry.getKey();
cnFields[count] = entry.getValue();
count++;
}
// 填充表头
String [] includeHeaders = new String[count];
for (int i = 0; i < cnFields.length; i++) {
includeHeaders[i] = cnFields[i];
}
writer.writeNext(includeHeaders);
// 填充内容
String [] data = new String[count];
for (int index = 0; index < list.size(); index++) {
// 获取单个对象
T item = list.get(index);
for (int i = 0; i < enFields.length; i++) {
Object objValue = getFieldValueByNameSequence(enFields[i], item);
String fieldValue = objValue == null ? "" : objValue.toString();
data[i] = fieldValue;
}
writer.writeNext(data);
}
}
}
测试:
ExportByOpenCsvUtils exportByOpenCsvUtils = new ExportByOpenCsvUtils();
User user = new User();
user.setId(1);
user.setName("第一个");
User user1 = new User();
user1.setId(2);
user1.setName("第二个");
String path = "D:\\code\\test.csv";
String csvName = "测试csv";
List<User> userList = new ArrayList<>();
userList.add(user);
userList.add(user1);
LinkedHashMap<String, String> fieldMap = new LinkedHashMap<String, String>();
fieldMap.put("id","id");
fieldMap.put("name","名字");
ExportByOpenCsvUtils.export(path,csvName,userList,fieldMap);