【前言】
我们项目经理要看一下贝瓦给我们提供了哪些数据，所以就导出一下咯
【正文】
废不多话，直接上代码
/**
* 从贝瓦获取指定频道下的作品并导出到excel文件中
*
* @param albumExportPreUrl 获取数据url的前部分
* @param page 当前页
*/
public void exportAlbumOrContent(String fileNames, String albumExportPreUrl,
String page) throws Exception {
//文件相关，UUID 防重名，以为对方一次只能给20条数据（真分页）
UUID uuid = UUID.randomUUID();
String fileName = fileNames + uuid.toString() + "专辑.xls";
File file = new File(fileName);
String contentOfCatUrl = albumExportPreUrl + page + "&access-token=" + bevaResource.getBevaToken();
;
JSONObject formResource = getJSONObjectBeva(contentOfCatUrl);
//获取表格的列名，没有tags，因为是空的
Map<String, String> bevaDataExcelTitle = BevaDataMapOfExcelTitle();//设置标题
JSONArray jsonArray = formResource.getJSONObject("data").getJSONArray("items");
//生成excel文件
saveFile(bevaDataExcelTitle, jsonArray, file);
}
/**
* 为url添加token，并且调用贝瓦的数据
*
* @param strUrl 基础url
* @return 贝瓦的数据
*/
private JSONObject getJSONObjectBeva(String strUrl) throws Exception {
strUrl = strUrl + "&access-token=" + bevaResource.getBevaToken();
strUrl = strUrl.replaceAll("[;\"]", "");//拼接的url 去除分隔符符合url要求
return restTemplate.getForEntity(strUrl, JSONObject.class).getBody();
}
设置标题：
/**
* 导出excel的列名
*
* @return
*/
private Map<String, String> BevaDataMapOfExcelTitle() {
Map<String, String> excelTitle = new LinkedHashMap<>();
excelTitle.put("id", "作品id");
excelTitle.put("name", "作品名称");
excelTitle.put("cont_type", "作品类型");
excelTitle.put("ad_play_type", "支持广告播放的情况");
excelTitle.put("description", "作品描述");
excelTitle.put("play_count", "作品的总播放次数");
excelTitle.put("vip_label", "是否需要vip和收费");
excelTitle.put("img_small", "小的图片");
excelTitle.put("img_middle", "中等的图片");
excelTitle.put("img_big", "大的图片");
excelTitle.put("published_at", "专辑发布时间");
excelTitle.put("created_at", "专辑最后修改时间");
excelTitle.put("updated_at", "专辑最后修改时间");
return excelTitle;
}
导出：
saveFile(bevaDataExcelTitle, jsonArray, file);
/**
* 导出文件
*
* @param headData
* @param listData
* @param file
* @return
*/
private File saveFile(Map<String, String> headData, JSONArray listData, File file) {
// 创建工作薄
HSSFWorkbook hssfWorkbook = new HSSFWorkbook();
// sheet:一张表的简称
// row:表里的行
// 创建工作薄中的工作表
HSSFSheet hssfSheet = hssfWorkbook.createSheet();
// 创建行
HSSFRow row = hssfSheet.createRow(0);
// 创建单元格，设置表头 创建列
HSSFCell cell = null;
// 初始化索引
int rowIndex = 0;
int cellIndex = 0;
// 创建标题行
row = hssfSheet.createRow(rowIndex);
rowIndex++;
// 遍历标题
for (String h : headData.keySet()) {
// 创建列
cell = row.createCell(cellIndex);
// 索引递增
cellIndex++;
// 逐列插入标题
cell.setCellValue(headData.get(h));
}
// 得到所有记录 行：列
JSONObject record = null;
if (listData != null) {
// 获取所有的记录 有多少条记录就创建多少行
for (int i = 0; i < listData.size(); i++) {
row = hssfSheet.createRow(rowIndex);
// 得到所有的行 一个record就代表 一行
record = listData.getJSONObject(i);
// 下一行索引
rowIndex++;
// 刷新新行索引
cellIndex = 0;
// 在有所有的记录基础之上，便利传入进来的表头,再创建N行
for (String h : headData.keySet()) {
cell = row.createCell(cellIndex);
cellIndex++;
// 按照每条记录匹配数据
cell.setCellValue(record.get(h) == null ? "" : record.get(h).toString());
}
}
}
try {
FileOutputStream fileOutputStreane = new FileOutputStream(file);
hssfWorkbook.write(fileOutputStreane);
fileOutputStreane.flush();
fileOutputStreane.close();
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return file;
}
会导出到文件file中的路径下，导出很多文件手动合并显然不太科学、网上看了一下excel本身提供了函数支持，java有代码，下面的代码我没有试验、看着挺不错、充一下门面：
/**
* TODO合并excel为一个文件
* @param outputFileName
* @param inputFileNameArray
*/
//@RequestMapping(value = "/mergExcel", method = RequestMethod.GET)
public void mergeExcel(@RequestParam(value = "outputFileName") String outputFileName,
@RequestBody String... inputFileNameArray) {
if (inputFileNameArray.length == 1) {
return;
}
try {
WritableWorkbook outputExcel = Workbook.createWorkbook(new File(outputFileName));
int index = 0;
for (String fileName : inputFileNameArray) {
// 创建excel文件的工作簿对象book
Workbook inputExcel = Workbook.getWorkbook(new FileInputStream(fileName));
// 获取excel文件工作簿的工作表数量sheets
Sheet[] sheets = inputExcel.getSheets();
for (Sheet sheet : sheets) {
WritableSheet writableSheet = outputExcel.createSheet(sheet.getName(), index);
copy(sheet, writableSheet);
index++;
}
}
/** **********将以上缓存中的内容写到EXCEL文件中******** */
outputExcel.write();
outputExcel.close();
} catch (Exception e) {
StringWriter sw = new StringWriter();
PrintWriter pw = new PrintWriter(sw);
e.printStackTrace(pw);
}
}
private static void copy(Sheet formSheet, WritableSheet toWritableSheet) {
// 行数
int rows = formSheet.getRows();
// 列数
int columns = formSheet.getColumns();
for (int rowIndex = 0; rowIndex < rows; rowIndex++) {
for (int columnIndex = 0; columnIndex < columns; columnIndex++) {
// 获取当前工作表.row_index,column_index单元格的cell对象
Cell cell = formSheet.getCell(columnIndex, rowIndex);
try {
CellFormat wcf = new WritableCellFormat();
if (cell.getCellFormat() != null) {
wcf = cell.getCellFormat();
}
toWritableSheet.addCell(new Label(columnIndex, rowIndex, cell.getContents(), wcf));
} catch (Exception e) {
StringWriter sw = new StringWriter();
PrintWriter pw = new PrintWriter(sw);
e.printStackTrace(pw);
}
}
}
}
【后语】
有一个动态获取token的过程，暂时不写了，下篇说或者看心情再说吧