POI 表格数据导出

@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();
    }

  

猜你喜欢

转载自www.cnblogs.com/Qi1007/p/10569410.html