@RequestMapping(value = "/exportUserSourceConsumeFundList", method = RequestMethod.GET)
public void doExportUserSourceConsumeFundList(HttpServletResponse response, UserProfileVO userProfile, Integer exportType) throws Exception {
LOGGER.info("用户消费金额导出:usersource/exportUserSourceConsumeFundList");
if (null != userProfile.getBeginTime()) {
userProfile.setBeginDateTime(DateUtil.parseDate(userProfile.getBeginTime()));
}
if (null != userProfile.getEndTime()) {
userProfile.setEndDateTime(DateUtil.parseDate(userProfile.getEndTime()));
}
List<UserProfileVO> userSourceConsumeFundList = userProfileBO.exportUserSourceConsumeFundList(userProfile);
String[] columns = {"账户用户名","用户真实姓名","手机号","消费金额","加金额","注册时间"};
String[] dataFields = {"username","name","mobile","fundAmount","goldAmount","registerTime"};
// 设置response参数,可以打开下载页面
response.reset();
response.setContentType("application/vnd.ms-excel;charset=utf-8");
ServletOutputStream out = response.getOutputStream();
if (exportType == ExportFileType.XLS.getKey()) {
Workbook wb = new HSSFWorkbook();
response.setHeader("Content-Disposition", "attachment;filename=" + new String((EXPORT_NAME_USER_CONSUME_FUND + DateUtil.formatStandardDate(new Date()) + EXTENSION_XLS).getBytes(), "iso-8859-1"));
FileUtil.<UserProfileVO>exportExcel(out, columns, dataFields, userSourceConsumeFundList, wb, EXPORT_NAME_USER_CONSUME_FUND);
} else if (exportType == ExportFileType.XLSX.getKey()) {
Workbook wb = new XSSFWorkbook();
response.setHeader("Content-Disposition", "attachment;filename=" + new String((EXPORT_NAME_USER_CONSUME_FUND + DateUtil.formatStandardDate(new Date()) + EXTENSION_XLSX).getBytes(), "iso-8859-1"));
FileUtil.<UserProfileVO>exportExcel(out, columns, dataFields, userSourceConsumeFundList,wb, EXPORT_NAME_USER_CONSUME_FUND);
} else if (exportType == ExportFileType.CSV.getKey()) {
response.setHeader("Content-Disposition", "attachment;filename=" + new String((EXPORT_NAME_USER_CONSUME_FUND + DateUtil.formatStandardDate(new Date()) + EXTENSION_CSV).getBytes(), "iso-8859-1"));
FileUtil.<UserProfileVO>exportCsvFile(out, columns, dataFields, userSourceConsumeFundList, EXPORT_NAME_USER_CONSUME_FUND);
}
}
public static <T> void exportExcel(OutputStream out,String[] columns, String[] dataFields, List<T> dataList, Workbook wb, String fileName) throws Exception {
// 创建第一个sheet(页),并命名
Sheet sheet = wb.createSheet(fileName);
// 创建第一行
Row row = sheet.createRow((short) 0);
Cell cell = null;
// 创建title
for (int i = 0; i < columns.length; i++) {
cell = row.createCell(i);
cell.setCellValue(columns[i]);
}
for (int i = 0; i < dataList.size(); i++) {
row = sheet.createRow(i + 1);
T userTemp = dataList.get(i);
HashMap<String, Object> userMap = HashMapUtil.objToHash(userTemp);
for (int j = 0; j < dataFields.length; j++) {
cell = row.createCell(j);
String dataField = dataFields[j];
Object value = userMap.get(dataField);
if (value != null) {
cell.setCellValue(value.toString());
} else {
cell.setCellValue("");
}
}
}
ByteArrayOutputStream os = new ByteArrayOutputStream();
try {
wb.write(os);
} catch (IOException e) {
e.printStackTrace();
}
byte[] content = os.toByteArray();
InputStream is = new ByteArrayInputStream(content);
BufferedInputStream bis = null;
BufferedOutputStream bos = null;
try {
bis = new BufferedInputStream(is);
bos = new BufferedOutputStream(out);
byte[] buff = new byte[2048];
int bytesRead;
// Simple read/write loop.
while (-1 != (bytesRead = bis.read(buff, 0, buff.length))) {
bos.write(buff, 0, bytesRead);
}
} catch (final IOException e) {
e.getMessage();
} finally {
if (bis != null) {
bis.close();
}
if (bos != null){
bos.close();
}
}
}