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 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 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;
    }

 

你可能感兴趣的:(java)