根据csv模板写入数据
1.确认业务数据接口返回的数据格式,将csv模板解析为自己需要的jon数据格式;
举例:
业务接口数据返回格式:
{
“csvDataList”: [
{
“name”: “name1”,
“id”: 1
},
{
“name”: “name2”,
“id”: 2
}
]
}
配置csv模板 csvExample.csv:
第一列:导出csv文件的头部;
第二列:业务数据接口返回结构的对应list字段;
第三列:业务数据接口返回结构对应list的每个字段,此处的字段跟第一行的头部对应;
解析模板为:
{"head":["姓名","id"],
"listDesc":"csvDataList",
"valueList":[{"index":0,"valueDesc":"name"},{"index":1,"valueDesc":"id"}]}
2.读取模板写入数据:
/**
* 写一行数据方法
*
* @param row
* @param csvWriter
* @throws IOException
*/
private static void writeRow(List<Object> row, BufferedWriter csvWriter,boolean isHead) throws IOException {
// 写入文件头部
StringBuffer sb = new StringBuffer();
for (Object data : row) {
if(isHead){
sb.append(data).append(",");
continue;
}
if(data instanceof String && StringUtils.isEmpty((String)data)){
sb.append(data).append(",");
continue;
}
//加上制表符"\t",解决导出结果变为科学技术法的问题
sb.append(data).append("\t,");
}
csvWriter.write(sb.toString());
csvWriter.newLine();
}
public static void main(String[] args) throws Exception {
//csv解析模板
String csvTemplete = "";
CsvVo csvVo = JsonUtils.readValue(csvTemplete,CsvVo.class);
File csvFile = new File("/data/temp/test.csv");
//删除并创建新文件
if (csvFile.exists()) {
csvFile.delete();
}
csvFile.createNewFile();
//使正确读取分隔符","
BufferedWriter csvWtriter = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(csvFile), "UTF-8"), 1024);
//处理中文乱码
csvWtriter.write(new String(new byte[] { (byte) 0xEF, (byte) 0xBB,(byte) 0xBF }));
/** 写入文件头部 */
writeRow(csvVo.getHead(), csvWtriter,true);
boolean isNext = true;
while (isNext);{
//调用业务数据接口获取返回结果
String resp = callGetData();
//将resp根据csvVo解析为对应的数据
List<List<Object>> parsedDataList = praseResp(resp,csvVo);
for (List<Object> param : parsedDataList) {
writeRow(param, csvWtriter,false);
}
}
csvWtriter.close();
csvWtriter.flush();
}
private static List<List<Object>> praseResp(String resp, CsvVo csvVo) {
//todo 根据自己的接口返回结果解析数据,返回结果为:List<行数据的list>
return null;
}
private static String callGetData() {
// TODO: 2020-04-17 请求获取业务数据的接口
return null;
}
- tips:
如果写入数据本身存在双引号或者逗号,需要做特殊处理:
boolean quoteFlag = false;
valueString = valueString.replaceAll("\r|\n", "");
if(valueString.contains("\"")){ //若发现有双引号 需要将字符串中的一个双引号替换为两个 并且需前后加双引号
valueString = valueString.replaceAll("\"","\"\"");
valueString = "\"" + valueString + "\"";
quoteFlag = true;
}
if(valueString.contains(",") && !quoteFlag){ //若发现有逗号 需前后加引号
valueString = "\"" + valueString + "\"";
}
如果存在中文编码问题:
File csvFile = new File("/data/temp/test.csv");
//删除并创建新文件
if (csvFile.exists()) {
csvFile.delete();
}
csvFile.createNewFile();
//使正确读取分隔符","
BufferedWriter csvWtriter = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(csvFile), "UTF-8"), 1024);
//处理中文乱码
csvWtriter.write(new String(new byte[] { (byte) 0xEF, (byte) 0xBB,(byte) 0xBF }));
如果导出数据变为科学计数法,则需要在数据上添加制表符"\t":
StringBuffer sb = new StringBuffer();
sb.append(exportData).append("\t,");