基於servlet的檔案上傳和下載

star77266989發表於2015-08-27

一、開發環境搭建

  建立一個FileUploadAndDownLoad專案,加入Apache的commons-fileupload檔案上傳元件的相關Jar包

 

*檔案上傳:

fileupload.jsp

<%@ page language="java" pageEncoding="UTF-8"%>
<!DOCTYPE HTML>
<html>
  <head>
    <title>檔案上傳</title>
  </head>
  
  <body>
    <form action="${pageContext.request.contextPath}/servlet/UploadFileServlet" enctype="multipart/form-data" method="post">
        上傳使用者:<input type="text" name="username"><br/>
        上傳檔案1:<input type="file" name="file1"><br/>
        上傳檔案2:<input type="file" name="file2"><br/>
        <input type="submit" value="提交">
    </form>
  </body>
</html>

 

上述的程式碼雖然可以成功將檔案上傳到伺服器上面的指定目錄當中,但是檔案上傳功能有許多需要注意的小細節問題,以下列出的幾點需要特別注意的

  1、為保證伺服器安全,上傳檔案應該放在外界無法直接訪問的目錄下,比如放於WEB-INF目錄下。

  2、為防止檔案覆蓋的現象發生,要為上傳檔案產生一個唯一的檔名。

  3、為防止一個目錄下面出現太多檔案,要使用hash演算法打散儲存。

  4、要限制上傳檔案的最大值。

  5、要限制上傳檔案的型別,在收到上傳檔名時,判斷字尾名是否合法。

UploadFileServlet.java

package com.hmx.controller;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.UUID;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadBase;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

public class UploadFileServlet extends HttpServlet {

	    public void doGet(HttpServletRequest request, HttpServletResponse response)
	            throws ServletException, IOException {
	                //得到上傳檔案的儲存目錄,將上傳的檔案存放於WEB-INF目錄下,不允許外界直接訪問,保證上傳檔案的安全
	                String savePath = this.getServletContext().getRealPath("/WEB-INF/upload");
	                //上傳時生成的臨時檔案儲存目錄
	                String tempPath = this.getServletContext().getRealPath("/WEB-INF/temp");
	                File tmpFile = new File(tempPath);
	                if (!tmpFile.exists()) {
	                    //建立臨時目錄
	                    tmpFile.mkdir();
	                }
	                
	                //訊息提示
	                String message = "";
	                try{
	                    //使用Apache檔案上傳元件處理檔案上傳步驟:
	                    //1、建立一個DiskFileItemFactory工廠
	                    DiskFileItemFactory factory = new DiskFileItemFactory();
	                    //設定工廠的緩衝區的大小,當上傳的檔案大小超過緩衝區的大小時,就會生成一個臨時檔案存放到指定的臨時目錄當中。
	                    factory.setSizeThreshold(1024*100);//設定緩衝區的大小為100KB,如果不指定,那麼緩衝區的大小預設是10KB
	                    //設定上傳時生成的臨時檔案的儲存目錄
	                    factory.setRepository(tmpFile);
	                    //2、建立一個檔案上傳解析器
	                    ServletFileUpload upload = new ServletFileUpload(factory);
	                    //監聽檔案上傳進度
	                    upload.setProgressListener(new ProgressListener(){
	                        public void update(long pBytesRead, long pContentLength, int arg2) {
	                            System.out.println("檔案大小為:" + pContentLength + ",當前已處理:" + pBytesRead);
	                            /**
	                             * 檔案大小為:14608,當前已處理:4096
	                                檔案大小為:14608,當前已處理:7367
	                                檔案大小為:14608,當前已處理:11419
	                                檔案大小為:14608,當前已處理:14608
	                             */
	                        }
	                    });
	                     //解決上傳檔名的中文亂碼
	                    upload.setHeaderEncoding("UTF-8"); 
	                    //3、判斷提交上來的資料是否是上傳表單的資料
	                    if(!ServletFileUpload.isMultipartContent(request)){
	                        //按照傳統方式獲取資料
	                        return;
	                    }
	                    
	                    //設定上傳單個檔案的大小的最大值,目前是設定為1024*1024位元組,也就是1MB
	                    upload.setFileSizeMax(1024*1024);
	                    //設定上傳檔案總量的最大值,最大值=同時上傳的多個檔案的大小的最大值的和,目前設定為10MB
	                    upload.setSizeMax(1024*1024*10);
	                    //4、使用ServletFileUpload解析器解析上傳資料,解析結果返回的是一個List<FileItem>集合,每一個FileItem對應一個Form表單的輸入項
	                    List<FileItem> list = upload.parseRequest(request);
	                    for(FileItem item : list){
	                        //如果fileitem中封裝的是普通輸入項的資料
	                        if(item.isFormField()){
	                            String name = item.getFieldName();
	                            //解決普通輸入項的資料的中文亂碼問題
	                            String value = item.getString("UTF-8");
	                            //value = new String(value.getBytes("iso8859-1"),"UTF-8");
	                            System.out.println("名字:"+name + "=" + value);
	                        }else{
	                        	//如果fileitem中封裝的是上傳檔案
	                            //得到上傳的檔名稱,
	                            String filename = item.getName();
	                            System.out.println(filename);
	                            if(filename==null || filename.trim().equals("")){
	                                continue;
	                            }
	                            //注意:不同的瀏覽器提交的檔名是不一樣的,有些瀏覽器提交上來的檔名是帶有路徑的,如:  c:\a\b\1.txt,而有些只是單純的檔名,如:1.txt
	                            //處理獲取到的上傳檔案的檔名的路徑部分,只保留檔名部分
	                            filename = filename.substring(filename.lastIndexOf("\\")+1);
	                            //得到上傳檔案的副檔名
	                            String fileExtName = filename.substring(filename.lastIndexOf(".")+1);
	                            //如果需要限制上傳的檔案型別,那麼可以通過檔案的副檔名來判斷上傳的檔案型別是否合法
	                            System.out.println("上傳的檔案的副檔名是:"+fileExtName);
	                            //獲取item中的上傳檔案的輸入流
	                            InputStream in = item.getInputStream();
	                            //得到檔案儲存的名稱
	                            String saveFilename = makeFileName(filename);
	                            //得到檔案的儲存目錄
	                            String realSavePath = makePath(saveFilename, savePath);
	                            //建立一個檔案輸出流
	                            FileOutputStream out = new FileOutputStream(realSavePath + "\\" + saveFilename);
	                            //建立一個緩衝區
	                            byte buffer[] = new byte[1024];
	                            //判斷輸入流中的資料是否已經讀完的標識
	                            int len = 0;
	                            //迴圈將輸入流讀入到緩衝區當中,(len=in.read(buffer))>0就表示in裡面還有資料
	                            while((len=in.read(buffer))>0){
	                                //使用FileOutputStream輸出流將緩衝區的資料寫入到指定的目錄(savePath + "\\" + filename)當中
	                                out.write(buffer, 0, len);
	                            }
	                            //關閉輸入流
	                            in.close();
	                            //關閉輸出流
	                            out.close();
	                            //刪除處理檔案上傳時生成的臨時檔案
	                            //item.delete();
	                            message = "檔案上傳成功!";
	                        }
	                    }
	                }catch (FileUploadBase.FileSizeLimitExceededException e) {
	                    e.printStackTrace();
	                    request.setAttribute("message", "單個檔案超出最大值!!!");
	                    request.getRequestDispatcher("/message.jsp").forward(request, response);
	                    return;
	                }catch (FileUploadBase.SizeLimitExceededException e) {
	                    e.printStackTrace();
	                    request.setAttribute("message", "上傳檔案的總的大小超出限制的最大值!!!");
	                    request.getRequestDispatcher("/message.jsp").forward(request, response);
	                    return;
	                }catch (Exception e) {
	                    message= "檔案上傳失敗!";
	                    e.printStackTrace();
	                }
	                request.setAttribute("message",message);
	                request.getRequestDispatcher("/message.jsp").forward(request, response);
	    }
	    
	    /**
	    * @Method: makeFileName
	    * @Description: 生成上傳檔案的檔名,檔名以:uuid+"_"+檔案的原始名稱
	    * @param filename 檔案的原始名稱
	    * @return uuid+"_"+檔案的原始名稱
	    */ 
	    private String makeFileName(String filename){  //2.jpg
	        //為防止檔案覆蓋的現象發生,要為上傳檔案產生一個唯一的檔名
	        return UUID.randomUUID().toString() + "_" + filename;
	    }
	    
	    /**
	     * 為防止一個目錄下面出現太多檔案,要使用hash演算法打散儲存
	    * @param filename 檔名,要根據檔名生成儲存目錄
	    * @param savePath 檔案儲存路徑
	    * @return 新的儲存目錄
	    */ 
	    private String makePath(String filename,String savePath){
	        //得到檔名的hashCode的值,得到的就是filename這個字串物件在記憶體中的地址
	        int hashcode = filename.hashCode();
	        int dir1 = hashcode&0xf;  //0--15
	        int dir2 = (hashcode&0xf0)>>4;  //0-15
	        //構造新的儲存目錄
	        String dir = savePath + "\\" + dir1 + "\\" + dir2;  //upload\2\3  upload\3\5
	        //File既可以代表檔案也可以代表目錄
	        File file = new File(dir);
	        //如果目錄不存在
	        if(!file.exists()){
	            //建立目錄
	            file.mkdirs();
	        }
	        return dir;
	    }

	    public void doPost(HttpServletRequest request, HttpServletResponse response)
	            throws ServletException, IOException {

	        doGet(request, response);
	    }
	}

 在Web.xml檔案

 <servlet>
    <description>This is the description of my J2EE component</description>
    <display-name>This is the display name of my J2EE component</display-name>
    <servlet-name>UploadFileServlet</servlet-name>
    <servlet-class>com.hmx.controller.UploadFileServlet</servlet-class>
  </servlet>



<servlet-mapping>
    <servlet-name>UploadFileServlet</servlet-name>
    <url-pattern>/servlet/UploadFileServlet</url-pattern>
  </servlet-mapping>

 

 

檔案下載

列出提供下載的檔案資源

  我們要將Web應用系統中的檔案資源提供給使用者進行下載,首先我們要有一個頁面列出上傳檔案目錄下的所有檔案,當使用者點選檔案下載超連結時就進行下載操作,編寫一個ListFileServlet,用於列出Web應用系統中所有下載檔案。

listfile.jsp

<%@ page language="java" import="java.util.*" pageEncoding="UTF-8"%>
<%@taglib prefix="c" uri="http://java.sun.com/jsp/jstl/core" %>
<!DOCTYPE HTML>
<html>
  <head>
    <title>下載檔案顯示頁面</title>
  </head>
  
  <body>
  <form action="${pageContext.request.contextPath}/servlet/ListFileServlet" enctype="multipart/form-data" method="post">
        <input type="submit" value="提交">
    </form>
      <!-- 遍歷Map集合 -->
    <c:forEach var="me" items="${fileNameMap}">
        <c:url value="/servlet/DownLoadServlet" var="downurl">
            <c:param name="filename" value="${me.key}"></c:param>
        </c:url>
        ${me.value}<a href="${downurl}">下載</a>
        <br/>
    </c:forEach>
  </body>
</html>

 

 

  ListFileServlet的程式碼如下:

package com.hmx.controller;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class ListFileServlet extends HttpServlet {


	    public void doGet(HttpServletRequest request, HttpServletResponse response)
	            throws ServletException, IOException {
	        //獲取上傳檔案的目錄
	        String uploadFilePath = this.getServletContext().getRealPath("/WEB-INF/upload");
	        //儲存要下載的檔名
	        Map<String,String> fileNameMap = new HashMap<String,String>();
	        //遞迴遍歷filepath目錄下的所有檔案和目錄,將檔案的檔名儲存到map集合中
	        listfile(new File(uploadFilePath),fileNameMap);//File既可以代表一個檔案也可以代表一個目錄
	        //將Map集合傳送到listfile.jsp頁面進行顯示
	        request.setAttribute("fileNameMap", fileNameMap);
	        request.getRequestDispatcher("/listfile.jsp").forward(request, response);
	    }
	    
	    /**
	    * @Method: listfile
	    * @Description: 遞迴遍歷指定目錄下的所有檔案
	    * @param file 即代表一個檔案,也代表一個檔案目錄
	    * @param map 儲存檔名的Map集合
	    */ 
	    public void listfile(File file,Map<String,String> map){
	        //如果file代表的不是一個檔案,而是一個目錄
	        if(!file.isFile()){
	            //列出該目錄下的所有檔案和目錄
	            File files[] = file.listFiles();
	            //遍歷files[]陣列
	            for(File f : files){
	                //遞迴
	                listfile(f,map);
	            }
	        }else{
	            /**
	             * 處理檔名,上傳後的檔案是以uuid_檔名的形式去重新命名的,去除檔名的uuid_部分
	                file.getName().indexOf("_")檢索字串中第一次出現"_"字元的位置,如果檔名類似於:9349249849-88343-8344_阿_凡_達.avi
	                那麼file.getName().substring(file.getName().indexOf("_")+1)處理之後就可以得到阿_凡_達.avi部分
	             */
	            String realName = file.getName().substring(file.getName().indexOf("_")+1);
	            //file.getName()得到的是檔案的原始名稱,這個名稱是唯一的,因此可以作為key,realName是處理過後的名稱,有可能會重複
	            map.put(file.getName(), realName);
	        }
	    }
	    
	    public void doPost(HttpServletRequest request, HttpServletResponse response)
	            throws ServletException, IOException {
	        doGet(request, response);
	    }
	}

 

DownLoadServlet.java

package com.hmx.controller;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.URLEncoder;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class DownLoadServlet extends HttpServlet {

	    public void doGet(HttpServletRequest request, HttpServletResponse response)
	            throws ServletException, IOException {
	        //得到要下載的檔名
	        String fileName = request.getParameter("filename");  //23239283-92489-阿凡達.avi
	        //fileName = new String(fileName.getBytes("iso8859-1"),"UTF-8");
	        //上傳的檔案都是儲存在/WEB-INF/upload目錄下的子目錄當中
	        String fileSaveRootPath=this.getServletContext().getRealPath("/WEB-INF/upload");
	        //通過檔名找出檔案的所在目錄
	        String path = findFileSavePathByFileName(fileName,fileSaveRootPath);
	        //得到要下載的檔案
	        File file = new File(path + "\\" + fileName);
	        //如果檔案不存在
	        if(!file.exists()){
	            request.setAttribute("message", "您要下載的資源已被刪除!!");
	            request.getRequestDispatcher("/message.jsp").forward(request, response);
	            return;
	        }
	        //處理檔名
	        String realname = fileName.substring(fileName.indexOf("_")+1);
	        //設定響應頭,控制瀏覽器下載該檔案
	        response.setHeader("content-disposition", "attachment;filename=" + URLEncoder.encode(realname, "UTF-8"));
	        //讀取要下載的檔案,儲存到檔案輸入流
	        FileInputStream in = new FileInputStream(path + "\\" + fileName);
	        //建立輸出流
	        OutputStream out = response.getOutputStream();
	        //建立緩衝區
	        byte buffer[] = new byte[1024];
	        int len = 0;
	        //迴圈將輸入流中的內容讀取到緩衝區當中
	        while((len=in.read(buffer))>0){
	            //輸出緩衝區的內容到瀏覽器,實現檔案下載
	            out.write(buffer, 0, len);
	        }
	        //關閉檔案輸入流
	        in.close();
	        //關閉輸出流
	        out.close();
	    }
	    
	    /**
	    * @Method: findFileSavePathByFileName
	    * @Description: 通過檔名和儲存上傳檔案根目錄找出要下載的檔案的所在路徑
	    * @param filename 要下載的檔名
	    * @param saveRootPath 上傳檔案儲存的根目錄,也就是/WEB-INF/upload目錄
	    * @return 要下載的檔案的儲存目錄
	    */ 
	    public String findFileSavePathByFileName(String filename,String saveRootPath){
	        int hashcode = filename.hashCode();
	        int dir1 = hashcode&0xf;  //0--15
	        int dir2 = (hashcode&0xf0)>>4;  //0-15
	        String dir = saveRootPath + "\\" + dir1 + "\\" + dir2;  //upload\2\3  upload\3\5
	        File file = new File(dir);
	        if(!file.exists()){
	            //建立目錄
	            file.mkdirs();
	        }
	        return dir;
	    }
	    
	    public void doPost(HttpServletRequest request, HttpServletResponse response)
	            throws ServletException, IOException {
	        doGet(request, response);
	    }
	}

 

web.xml

 <servlet>
    <description>This is the description of my J2EE component</description>
    <display-name>This is the display name of my J2EE component</display-name>
    <servlet-name>ListFileServlet</servlet-name>
    <servlet-class>com.hmx.controller.ListFileServlet</servlet-class>
  </servlet>
  <servlet>
    <description>This is the description of my J2EE component</description>
    <display-name>This is the display name of my J2EE component</display-name>
    <servlet-name>DownLoadServlet</servlet-name>
    <servlet-class>com.hmx.controller.DownLoadServlet</servlet-class>
  </servlet>



  <servlet-mapping>
    <servlet-name>ListFileServlet</servlet-name>
    <url-pattern>/servlet/ListFileServlet</url-pattern>
  </servlet-mapping>
  <servlet-mapping>
    <servlet-name>DownLoadServlet</servlet-name>
    <url-pattern>/servlet/DownLoadServlet</url-pattern>
  </servlet-mapping>

 

相關文章