导出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);
上一篇: Tensorflow分布式训练
下一篇: oracle触发器详解与使用