这两个功能都是很常见的,有工具类可以直接用,你要做的,只是传入数据而已
1. 导出为Excel
1.引入 poi 依赖
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>3.17</version>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>3.17</version>
</dependency>
2. 引入工具类 ExcelUtil
package com.gxx.nonmotorvehicle.util;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.sql.Timestamp;
import java.text.SimpleDateFormat;
import java.util.*;
public class ExcelUtil {
private static SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
/**
* @param response
* @param fileName excel文件名
* @param headMap 表头map
* @param dataList 表格数据
*/
public static void exportXlsx(HttpServletResponse response, String fileName, Map<String, String> headMap, List<Map<String, Object>> dataList) {
Workbook workbook = exportXlsx(fileName, headMap, dataList);
exportXlsx(response,fileName,workbook);
}
/**
* @param response
* @param fileName excel文件名
*/
public static void exportXlsx(HttpServletResponse response, String fileName, Workbook workbook) {
response.setContentType("application/binary;charset=ISO8859_1");
OutputStream outputStream = null;
try {
outputStream = response.getOutputStream();
String fn = new String(fileName.getBytes(), "ISO8859_1");
response.setHeader("Content-disposition", "attachment; filename=" + fn + ".xlsx");
workbook.write(outputStream);
} catch (Exception e) {
e.printStackTrace();
} finally {
if (outputStream != null) {
try {
outputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 导出数据
*
* @param headMap
* @param dataList
*/
public static Workbook exportXlsx(String sheetName, Map<String, String> headMap, List<Map<String, Object>> dataList) {
Workbook workbook = new XSSFWorkbook();
Sheet sheet = workbook.createSheet(sheetName);
int rowIndex = 0, columnIndex = 0;
Set<String> keys = headMap.keySet();
//表头
Row row = sheet.createRow(rowIndex++);
for (String key : keys) {
Cell cell = row.createCell(columnIndex++);
cell.setCellValue(headMap.get(key));
}
//内容
if (dataList != null && !dataList.isEmpty()) {
for (Map<String, Object> map : dataList) {
row = sheet.createRow(rowIndex++);
columnIndex = 0;
for (String key : keys) {
Cell cell = row.createCell(columnIndex++);
setCellValue(cell, map.get(key));
}
}
}
return workbook;
}
private static void setCellValue(Cell cell, Object obj) {
if (obj == null) {
return;
}
if (obj instanceof String) {
cell.setCellValue((String) obj);
} else if (obj instanceof Date) {
Date date = (Date) obj;
if (date != null) {
cell.setCellValue(format.format(date));
}
} else if (obj instanceof Calendar) {
Calendar calendar = (Calendar) obj;
if (calendar != null) {
cell.setCellValue(format.format(calendar.getTime()));
}
} else if (obj instanceof Timestamp) {
Timestamp timestamp = (Timestamp) obj;
if (timestamp != null) {
cell.setCellValue(format.format(new Date(timestamp.getTime())));
}
} else if (obj instanceof Double) {
cell.setCellValue((Double) obj);
} else {
cell.setCellValue(obj.toString());
}
}
}
3. 在 service 层定义列名,以及数据
@Override
public void exportRoadInfo(HttpServletResponse response) {
//1.获取数据
List<Map<String,Object>> datas = this.list();
//2.定义列名
Map<String,String> headerMap = new LinkedHashMap<>();
headerMap.put("crossing_name","路口");
headerMap.put("count","违法数量");
//3.调用工具类
ExcelUtil.exportXlsx(response, "路口违法信息"+new SimpleDateFormat("yyyyMMdd").format(new Date()), headerMap, datas);
}
2. 根据链接下载图片并打包
public void downloadPicture(HttpServletResponse response) throws ParseException {
//1.获取 url 集合
List<Map<String,Object>> datas = this.list(areaCode, violationType, startTime, endTime);
List<String> imageUrls = new ArrayList<>();
int i = 1;
for(Map mp : datas) {
if(i > 5000) break;
imageUrls.add(mp.get("img_path"));
i++;
}
//2.定义包名
LocalDateTime dateTime = LocalDateTime.now();
DateTimeFormatter format = DateTimeFormatter.ofPattern("yyyyMMddHHmmss");
String zipName = dateTime.format(format) + ".zip";
//3.设置请求头
response.setHeader("content-type", "application/octet-stream");
response.setHeader("Content-disposition", "attachment;filename=" + zipName);
response.setCharacterEncoding("utf-8");
//4.下载图片
ZipOutputStream zipOut = null;
try {
zipOut = new ZipOutputStream(response.getOutputStream());
for (String imageUrl : imageUrls) {
//4.1 设置图片名称
String fileName = imageUrl.substring(imageUrl.lastIndexOf("/") + 1);
zipOut.putNextEntry(new ZipEntry(fileName));
//4.2 请求下载图片
URL url = new URL(imageUrl);
InputStream in = new DataInputStream(url.openStream());
byte[] buff = new byte[1024];
int len;
while ((len = in.read(buff)) != -1) {
zipOut.write(buff, 0, len);
}
zipOut.closeEntry();
in.close();
}
} catch (IOException e) {
e.printStackTrace();
} finally {
if (zipOut != null) {
try {
zipOut.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}