最近項目有須要寫文件下載相關代碼,這邊提交記錄下相關代碼模塊,寫的不太好,後期再優化相關代碼,有好的建議,可留言,謝謝。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; } }
這邊自測試所有能夠正常使用,代碼質量不高,若是代碼方面優化有好的建議,請多多留言。性能