springboot單文件下載和多文件壓縮zip下載

單文件下載

//下載單個文件
public void downloadFile(HttpServletResponse response){
        String path = "D:\test\ce\1.txt"
        File file = new File(path);
        if(file.exists()){

            String fileName = file.getName();
            response.setHeader("Content-Disposition", "attachment;fileName=" + fileName);
            download(response,file);

        }
    }


public void download(HttpServletResponse response,File file){


        FileInputStream fis = null;
        BufferedInputStream bis = null;
        OutputStream os = null;

        try {
            os = response.getOutputStream();
            fis = new FileInputStream(file);
            bis = new BufferedInputStream(fis);
            byte[] buffer = new byte[bis.available()];
            int i = bis.read(buffer);
            while(i != -1){
                os.write(buffer, 0, i);
                i = bis.read(buffer);
            }

        } catch (Exception e) {
            e.printStackTrace();
        }
        try {
            bis.close();
            fis.close();
            os.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

多文件壓縮下載

//多個文件，壓縮成zip後下載
public void downloadMoreFile(HttpServletResponse response) {

        
        String test1= "D:\test\ce\1.txt";
        String test2= "D:\test\ce\2.txt";

        File tfile= new File(test1);
        File cfile= new File(test2);

        List<File> files = new ArrayList<>();
        files.add(tfile);
        files.add(cfile);
        if (tfile.exists() && cfile.exists()) {

            String zipTmp = "D:\test\ce\1.zip";
            zipd(zipTmp,files,response);

         
        }
    }

public void zipd(String zipTmp,List<File> files,HttpServletResponse response){
        File zipTmpFile = new File(zipTmp);
        try {
            if (zipTmpFile.exists()) {
                zipTmpFile.delete();
            }
            zipTmpFile.createNewFile();

            response.reset();
            // 創建文件輸出流
            FileOutputStream fous = new FileOutputStream(zipTmpFile);
            ZipOutputStream zipOut = new ZipOutputStream(fous);
            zipFile(files, zipOut);
            zipOut.close();
            fous.close();
            downloadZip(zipTmpFile, response);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

public static HttpServletResponse downloadZip(File file, HttpServletResponse response) {
        if (file.exists() == false) {
            System.out.println("待壓縮的文件目錄：" + file + "不存在.");
        } else {
            try {
                // 以流的形式下載文件。
                InputStream fis = new BufferedInputStream(new FileInputStream(file.getPath()));
                byte[] buffer = new byte[fis.available()];
                fis.read(buffer);
                fis.close();
                // 清空response
                response.reset();

                OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
                response.setContentType("application/octet-stream");

                // 如果輸出的是中文名的文件，在此處就要用URLEncoder.encode方法進行處理
                response.setHeader("Content-Disposition",
                        "attachment;filename=" + new String(file.getName().getBytes("GB2312"), "ISO8859-1"));
                toClient.write(buffer);
                toClient.flush();
                toClient.close();
            } catch (Exception ex) {
                ex.printStackTrace();
            } finally {
                try {
                    File f = new File(file.getPath());
                    f.delete();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
        return response;
    }