Java實現檔案上傳下載以及檢視功能

2022-03-01 19:01:22

專案的目錄結構

程式碼

IOUtils.java

package cn.edu.zyt.util;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

public class IOUtils {
    private IOUtils() {
    }
    
    public static void In2Out(InputStream in,OutputStream out) throws IOException{
        byte [] bs = new byte[1024];
        int i = 0;
        while((i=in.read(bs))!=-1){
            out.write(bs,0,i);
        }
    }
    
    public static void close(InputStream in,OutputStream out){
        if(in!=null){
            try {
                in.close();
            } catch (IOException e) {
                e.printStackTrace();
            }finally{
                in = null;
            }
        }
        if(out!=null){
            try {
                out.close();
            } catch (IOException e) {
                e.printStackTrace();
            }finally{
                out = null;
            }
        }
    }
}

DownServlet.java

package cn.edu.zyt.web;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;

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

import cn.edu.zyt.util.IOUtils;

public class DownServlet extends HttpServlet {

    public void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        String filename = request.getParameter("file");
        
        response.setHeader("Content-Disposition", "attachment;filename="+URLEncoder.encode(filename,"utf-8"));
        response.setContentType(this.getServletContext().getMimeType(filename));//MIME型別
        
        InputStream in = new FileInputStream(this.getServletContext().getRealPath(filename));
        OutputStream out = response.getOutputStream();
        IOUtils.In2Out(in, out);
        IOUtils.close(in, null);
    }

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

}

UploadHandleServlet.java

package cn.edu.zyt.web;


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 UploadHandleServlet 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:ab1.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演演算法打散儲存
    * @Method: makePath
  
    *
    * @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;  //upload23  upload35
        //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

<?xml version="1.0" encoding="UTF-8"?>
<web-app xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://java.sun.com/xml/ns/javaee" xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/web-app_2_5.xsd" version="2.5">
  
  <servlet>
    <servlet-name>DownServlet</servlet-name>
    <servlet-class>cn.edu.zyt.web.DownServlet</servlet-class>
  </servlet>
  
  <servlet-mapping>
    <servlet-name>DownServlet</servlet-name>
    <url-pattern>/servlet/DownServlet</url-pattern>
  </servlet-mapping>
  
  
  <servlet>
    <servlet-name>UploadHandleServlet</servlet-name>
    <servlet-class>cn.edu.zyt.web.UploadHandleServlet</servlet-class>
</servlet>

<servlet-mapping>
    <servlet-name>UploadHandleServlet</servlet-name>
    <url-pattern>/servlet/UploadHandleServlet</url-pattern>
</servlet-mapping>
</web-app>

upload.jsp

<%@ page language="java" contentType="text/html; charset=UTF-8"
    pageEncoding="UTF-8"%>
<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
<title>Insert title here</title>
</head>
<body>
    <form
        action="${pageContext.request.contextPath}/servlet/UploadHandleServlet"
        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>

down.jsp

<%@ page language="java" import="java.util.*" pageEncoding="utf-8"%>
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">
<html>
<head>
</head>
<body>
    <a href="${pageContext.request.contextPath }/1.jpg" rel="external nofollow" >檢視美女1號</a>
    <a href="${pageContext.request.contextPath }/1.txt" rel="external nofollow" >檢視檔案內容</a>
    <a href="${pageContext.request.contextPath }/毛概終版.doc" rel="external nofollow" >毛概終板</a>
    <a
        href="${pageContext.request.contextPath }/servlet/DownServlet?file=2.jpg" rel="external nofollow" >美女2號</a>
    <a
        href="${pageContext.request.contextPath }/servlet/DownServlet?file=1.txt" rel="external nofollow" >檔案</a>
    <a
        href="${pageContext.request.contextPath }/servlet/DownServlet?file=result.wav" rel="external nofollow" >視訊下載</a>
</body>
</html>