SpringBoot：文件下载、pdf预览、文件压缩

文件下载

编写工具类，编辑http响应体格式。其中Content-type为application/octet-stream代表二进制流，为下载文件的格式。详见对照表：https://tool.oschina.net/commons/
Utils：

public class FileExporter
{
    public static ResponseEntity<FileSystemResource> export(File file)
    {
        if (file == null) {
            return null;
        }
        HttpHeaders headers = new HttpHeaders();
        headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
        headers.add("Content-Disposition", "attachment; filename="+file.getName());
        headers.add("Pragma", "no-cache");
        headers.add("Expires", "0");
        headers.add("Last-Modified", new Date().toString());
        headers.add("ETag", String.valueOf(System.currentTimeMillis()));
        return ResponseEntity .ok() .headers(headers) .contentLength(file.length()) .contentType(MediaType.parseMediaType("application/octet-stream")) .body(new FileSystemResource(file));
    }
}

Controller：

    @RequestMapping(value = "/template/download",method = RequestMethod.GET)
    public ResponseEntity<FileSystemResource> downloadATemplate(Integer courseId)
    {
        File dir = new File(getClass().getResource(".").getFile(),"course"+courseId);
        if(!dir.exists()) return null;
        File[] files = dir.listFiles();
        if(files==null||files.length==0) return null;
        return FileExporter.export(files[0]);
    }

PDF预览

Http响应里Content-type为application/pdf。
Controller：

	@RequestMapping("/homework/check")
    public void homeworkCheckPage(Long userId, Integer courseId, HttpServletResponse response)
    {
        response.setContentType("application/pdf");//改成application/octet-stream就成了下载文件
        FileInputStream in;
        OutputStream out;
        try {
            in = new FileInputStream(new File(getClass().getResource("./course_"+courseId+"_report").getFile(),userId+".pdf"));
            out = response.getOutputStream();
            byte[] b = new byte[512];
            while ((in.read(b)) != -1) {
                out.write(b);
            }
            out.flush();
            in.close();
            out.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

文件压缩

要求将多个文件批量下载，方法为压缩成Zip后进行下载。文件压缩采用递归算法，遍历目录里的文件后进行压缩。编写工具类实现。
Utils：

public class ZipCompression
{
	/**
     * 压缩文件
     * @param zipFileName 压缩文件保存名
     * @param target 目标文件路径
     */
    public static void filetoZip(String zipFileName,File target)throws RuntimeException{
        ZipOutputStream zos = null ;
        try {
            zos = new ZipOutputStream(new FileOutputStream(new File(zipFileName)));
            do_file_toZip(target,zos,target.getName());//开始压缩
        } catch (Exception e) {
            throw new RuntimeException("压缩失败",e);
        }finally{
            if(zos != null){
                try {
                    zos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * 递归压缩方法
     * @param sourceFile   源文件
     * @param zos         zip输出流
     * @param name        压缩后的名称
     */
    private static void do_file_toZip(File sourceFile, ZipOutputStream zos, String name) throws Exception {
        byte[] buf = new byte[1024 * 2];
        if (sourceFile.isFile()) {
            // 向zip输出流中添加一个zip实体，构造器中name为zip实体的文件的名字
            zos.putNextEntry(new ZipEntry(name));
            // copy文件到zip输出流中
            int len;
            FileInputStream in = new FileInputStream(sourceFile);
            while ((len = in.read(buf)) != -1) {
                zos.write(buf, 0, len);
            }
            zos.closeEntry();
            in.close();
        } else {
            File[] listFiles = sourceFile.listFiles();
            if (listFiles == null || listFiles.length == 0) {
                // 需要保留原来的文件结构时,需要对空文件夹进行处理
                zos.putNextEntry(new ZipEntry(name + "/"));// 空文件夹的处理
                zos.closeEntry();// 没有文件，不需要文件的copy
            } else {
                for (File file : listFiles) {
                    // 判断是否需要保留原来的文件结构
                    // 注意：file.getName()前面需要带上父文件夹的名字加一斜杠,
                    // 不然最后压缩包中就不能保留原来的文件结构,即：所有文件都跑到压缩包根目录下了
                    do_file_toZip(file, zos, name + "/" + file.getName());
                }
            }
        }
    }
}