java文件下載(單文件下載,多文件打包下載)

最近項目有須要寫文件下載相關代碼,這邊提交記錄下相關代碼模塊,寫的不太好,後期再優化相關代碼,有好的建議,可留言,謝謝。java

1)單文件下載瀏覽器

    public String oneFileDownload(HttpServletRequest request,HttpServletResponse response){
        //針對需求須要與需求溝通下載文件傳遞參數
        //目前demo文件名是文件的hashCode值+後綴名的方式命名,能夠默認該hashCode值爲文件惟一id
        String fileName = request.getParameter("fileName");
        if(StringUtils.isBlank(fileName)){
            return "文件名爲空";
        }
        String filePath = request.getSession().getServletContext().getRealPath("/convert")+File.separator+fileName;//目前文件默認在該文件夾下,後續變更需修改
        File downloadFile = new File(filePath);
        if(downloadFile.exists()){
            OutputStream out = null;
            FileInputStream fis = null;
            BufferedInputStream bis = null;
            try {
                if(downloadFile.isDirectory()){
                    return "路徑錯誤僅指向文件夾";
                }
                out = response.getOutputStream();// 建立頁面返回方式爲輸出流,彈出下載框
                fis = new FileInputStream(downloadFile);
                bis = new BufferedInputStream(fis);
                response.setContentType("application/pdf; charset=UTF-8");//設置編碼字符
                response.setHeader("Content-disposition", "attachment;filename=" + fileName);
                byte[] bt = new byte[2048];
                int size = 0;
                while((size=bis.read(bt)) != -1){
                    out.write(bt, 0, size);
                }

            } catch (Exception e) {
                e.printStackTrace();
            }finally {
                try {
                    //關閉流
                    out.flush();
                    if(out != null){
                        out.close();
                    }
                    if(bis != null){
                        bis.close();
                    }
                    if(fis != null){
                        fis.close();
                    }
                } catch (Exception e2) {
                    e2.printStackTrace();
                }

            }
            return "下載成功";
        }else{
            
            return "文件不存在!";    
        }

    }

2)多文件打包下載緩存

  a)下載指定文件夾下的文件,若是嵌套文件夾也支持(但文件名須要惟一)app

    public String zipDownload(HttpServletRequest request,HttpServletResponse response) throws IOException{
        
        String sourceFilePath = request.getSession().getServletContext().getRealPath("/convert");//文件路徑
        String destinFilePath = request.getSession().getServletContext().getRealPath("/fileZip");
        String fileName = FileUtil.zipFiles(sourceFilePath, destinFilePath);
        File file = new File(destinFilePath+File.separator+fileName);
        response.setHeader("Content-disposition", "attachment;filename=" + fileName);
        if(file.exists()){
            FileInputStream fis = null;
            BufferedInputStream bis = null;
            OutputStream out = response.getOutputStream();
            try {
                fis = new FileInputStream(file);
                bis = new BufferedInputStream(fis);
                byte[] bufs = new byte[1024 * 10];
                int read = 0;
                while ((read = bis.read(bufs, 0, 1024 * 10)) != -1) {
                    out.write(bufs, 0, read);
                }
            } catch (FileNotFoundException e) {
                e.printStackTrace();
            }finally {
                if(bis != null){
                    bis.close();
                }
                if(out!=null){
                    out.close();
                }
                File zipFile = new File(destinFilePath+File.separator+fileName);
                if(zipFile.exists()){
                    zipFile.delete();
                }
            }
        }else{
            return "文件壓縮失敗";
        }
        return "下載成功";
    }

b)下載指定文件夾下的全部文件,支持樹型結構函數

    public String zipDownloadRelativePath(HttpServletRequest request,HttpServletResponse response) {
        String msg ="";//下載提示信息
        String root = request.getSession().getServletContext().getRealPath("/convert");//文件路徑
        Vector<File> fileVector = new Vector<File>();//定義容器裝載文件
        File file = new File(root);
        File [] subFile = file.listFiles();
        //判斷文件性質並取文件
        for(int i = 0; i<subFile.length; i++){
            if(!subFile[i].isDirectory()){//不是文件夾而且不爲空
                fileVector.add(subFile[i]);//裝載文件
            }else{//是文件夾,再次遞歸遍歷文件夾裏面的文件
                File [] files = subFile[i].listFiles();
                Vector v = FileUtil.getPathAllFiles(subFile[i],new Vector<File>());
                fileVector.addAll(v);//把迭代的文件裝到容器中
            }
        }
        //設置文件下載名稱
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss");
        String fileName = dateFormat.format(new Date())+".zip";
        response.setHeader("Content-disposition", "attachment;filename=" + fileName);//若是有中文須要轉碼
        //定義相關流
        //用於瀏覽器下載響應
        OutputStream out = null;
        //用於讀壓縮好的文件
        BufferedInputStream bis = null;//用緩存性能會好一些
        //用於壓縮文件
        ZipOutputStream zos = null;
        //建立一個空的zip文件
        String zipBasePath = request.getSession().getServletContext().getRealPath("/fileZip");
        String zipFilePath = zipBasePath+File.separator+fileName;
        File zipFile = new File(zipFilePath);
        try {
            if(!zipFile.exists()){//不存在建立一個新的
                zipFile.createNewFile();
            }
            out = response.getOutputStream();
            //建立文件輸出流
            zos = new ZipOutputStream(new FileOutputStream(zipFile));
            //循環文件,一個一個按順序壓縮
            for(int i = 0;i< fileVector.size();i++){
                System.out.println(fileVector.get(i).getName()+"大小爲:"+fileVector.get(i).length());
                FileUtil.zipFileFun(fileVector.get(i),root,zos);
            }
            //壓縮完成之後必需要在此處執行關閉 不然下載文件會有問題
            if(zos != null){
                zos.closeEntry();
                zos.close();
                }
            byte[] bt = new byte[10*1024];
            int size = 0;
            bis = new BufferedInputStream(new FileInputStream(zipFilePath),10*1024);
            while((size=bis.read(bt,0,10*1024)) != -1){//沒有讀完
                out.write(bt,0,size);
            }
            
        } catch (Exception e) {
            e.printStackTrace();
        }finally {//關閉相關流
            try {
                //避免出異常時沒法關閉
                if(zos != null){
                    zos.closeEntry();
                    zos.close();
                    }
                
                if(bis != null){
                    bis.close();
                }
                
                if(out != null){
                    out.close();
                }
                if(zipFile.exists()){
                    zipFile.delete();//刪除
                }
            } catch (Exception e2) {
                System.out.println("流關閉出錯!");
            }

        }
        return "下載成功";
    }

下載輔助工具類工具

package com.hhwy.fileview.utils;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Vector;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;




public class FileUtil {
    /**
     * 把某個文件路徑下面的文件包含文件夾壓縮到一個文件下
     * @param file
     * @param rootPath 相對地址
     * @param zipoutputStream
     */
    public static void zipFileFun(File file,String rootPath,ZipOutputStream zipoutputStream){
        if(file.exists()){//文件存在才合法
            if(file.isFile()){
                //定義相關操做流
                FileInputStream fis = null;
                BufferedInputStream bis = null;
                try {
                    //設置文件夾
                    String relativeFilePath = file.getPath().replace(rootPath+File.separator, "");
                    //建立輸入流讀取文件
                    fis = new FileInputStream(file);
                    bis = new BufferedInputStream(fis,10*1024);
                    //將文件裝入zip中,開始打包
                    ZipEntry zipEntry;
                    if(!relativeFilePath.contains("\\")){
                        zipEntry = new ZipEntry(file.getName());//此處值不能重複,要惟一,不然同名文件會報錯
                    }else{
                        zipEntry = new ZipEntry(relativeFilePath);//此處值不能重複,要惟一,不然同名文件會報錯
                    }
                    zipoutputStream.putNextEntry(zipEntry);
                    //開始寫文件
                    byte[] b = new byte[10*1024];
                    int size = 0;
                    while((size=bis.read(b,0,10*1024)) != -1){//沒有讀完
                        zipoutputStream.write(b,0,size);
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }finally {
                    try {
                        //讀完之後關閉相關流操做
                        if(bis != null){
                            bis.close();
                        }
                        if(fis != null){
                            fis.close();
                        }
                    } catch (Exception e2) {
                        System.out.println("流關閉錯誤!");
                    }
                }
            }
//            else{//若是是文件夾
//                try {
//                    File [] files = file.listFiles();//獲取文件夾下的全部文件
//                    for(File f : files){
//                        zipFileFun(f,rootPath, zipoutputStream);
//                    }
//                } catch (Exception e) {
//                    e.printStackTrace();
//                }
//                
//            }
        }
    }
    
    /*
     * 獲取某個文件夾下的全部文件
     */
    public static Vector<File> getPathAllFiles(File file,Vector<File> vector){
        if(file.isFile()){//若是是文件,直接裝載
            System.out.println("在迭代函數中文件"+file.getName()+"大小爲:"+file.length());
            vector.add(file);
        }else{//文件夾
            File[] files = file.listFiles();
            for(File f : files){//遞歸
                if(f.isDirectory()){
                    getPathAllFiles(f,vector);
                }else{
                    System.out.println("在迭代函數中文件"+f.getName()+"大小爲:"+f.length());
                    vector.add(f);
                }
            }
        }
        return vector;
    }
    
    /**
     * 壓縮文件到指定文件夾
     * @param sourceFilePath 源地址
     * @param destinFilePath 目的地址
     */
    public static String zipFiles(String sourceFilePath,String destinFilePath){
        File sourceFile = new File(sourceFilePath);
        FileInputStream fis = null;
        BufferedInputStream bis = null;
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss");
        String fileName = dateFormat.format(new Date())+".zip";
        String zipFilePath = destinFilePath+File.separator+fileName;
        if (sourceFile.exists() == false) {
            System.out.println("待壓縮的文件目錄:" + sourceFilePath + "不存在.");
        } else {
            try {
                File zipFile = new File(zipFilePath);
                if (zipFile.exists()) {
                    System.out.println(zipFilePath + "目錄下存在名字爲:" + fileName + ".zip" + "打包文件.");
                } else {
                    File[] sourceFiles = sourceFile.listFiles();
                    if (null == sourceFiles || sourceFiles.length < 1) {
                        System.out.println("待壓縮的文件目錄:" + sourceFilePath + "裏面不存在文件,無需壓縮.");
                    } else {
                        //讀取sourceFile下的全部文件
                        Vector<File> vector = FileUtil.getPathAllFiles(sourceFile, new Vector<File>()); 
                        fos = new FileOutputStream(zipFile);
                        zos = new ZipOutputStream(new BufferedOutputStream(fos));
                        byte[] bufs = new byte[1024 * 10];
                        
                        for (int i = 0; i < vector.size(); i++) {
                            // 建立ZIP實體,並添加進壓縮包
                            ZipEntry zipEntry = new ZipEntry(vector.get(i).getName());//文件不可重名,不然會報錯
                            zos.putNextEntry(zipEntry);
                            // 讀取待壓縮的文件並寫進壓縮包裏
                            fis = new FileInputStream(vector.get(i));
                            bis = new BufferedInputStream(fis, 1024 * 10);
                            int read = 0;
                            while ((read = bis.read(bufs, 0, 1024 * 10)) != -1) {
                                zos.write(bufs, 0, read);
                            }
                        }
                    }
                }
            } catch (FileNotFoundException e) {
                e.printStackTrace();
                throw new RuntimeException(e);
            } catch (IOException e) {
                e.printStackTrace();
                throw new RuntimeException(e);
            } finally {
                // 關閉流
                try {
                    if (null != bis)
                        bis.close();
                    if (null != zos)
                        zos.closeEntry();
                        zos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                    throw new RuntimeException(e);
                }
            }
        }
        return fileName;
    }
    
}

這邊自測試所有能夠正常使用,代碼質量不高,若是代碼方面優化有好的建議,請多多留言。性能

相關文章
相關標籤/搜索