@GetMapping("/download")
ResponseEntity<byte[]> download() throws IOException {
String head[] = {"员工编号", "用户名", "姓名", "密码", "身份证", "联系电话", "状态"};
HttpHeaders headers = new HttpHeaders();
headers.setCacheControl("no-cache, no-store, must-revalidate");
headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
headers.setContentLength(test(head).length);
headers.setContentDispositionFormData("attachment", URLEncoder.encode("员工信息表.xls", "utf-8"));
System.out.println(test(head).toString());
return ResponseEntity.ok()
.headers(headers)
.body(test(head));
}
public byte[] test(String arr[]){
List<Employee> petList = employeeMapper.selectAll();
HSSFWorkbook workbook = new HSSFWorkbook();
HSSFSheet sheet = workbook.createSheet("员工信息表");
HSSFRow row = sheet.createRow(0);
for (int i = 0; i < arr.length; i++) {
row.createCell(i).setCellValue(arr[i]);
}
int i = 1;
for (Employee employee : petList) {
row = sheet.createRow(i);
row.createCell(0).setCellValue(employee.getEmployeeId());
row.createCell(1).setCellValue(employee.getEmployeeName());
row.createCell(2).setCellValue(employee.getEmployeeUser());
row.createCell(3).setCellValue(employee.getEmployeePwd());
row.createCell(4).setCellValue(employee.getEmployeeIdcard());
row.createCell(5).setCellValue(employee.getEmployeePhone());
row.createCell(6).setCellValue(employee.getEmployeeStatus());
i++;
}
ByteArrayOutputStream stream = new ByteArrayOutputStream();
try {
workbook.write(stream);
} catch (IOException e) {
e.printStackTrace();
}
return stream.toByteArray();
}