Java實現文件的上傳和下載

文件上傳和下載

文件上傳下載的基本過程主要包含兩個步驟：
1.通過IO流將文件上傳/下載到預設的文件夾內；
2.將文件相關的信息（經過處理后的文件名，后綴，大小等）保存到數據庫中。

上傳模塊

public class UploadHandleServlet extends HttpServlet {
  public void service(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();
        // 設置工廠的緩沖區的大小，當上傳的文件大小超過緩沖區的大小時，就會生成一個
        // 臨時文件存放到指定的臨時目錄當中。設置緩沖區的大小為100KB，其默認大小是10KB
        factory.setSizeThreshold(1024 * 100);
        // 設置上傳時生成的臨時文件的保存目錄
        factory.setRepository(tmpFile);
        // 2、創建一個文件上傳解析器
        ServletFileUpload upload=new ServletFileUpload(factory);
        // 解決上傳文件名的中文亂碼
        upload.setHeaderEncoding("UTF-8");
        // 3、判斷提交上來的數據是否是上傳表單的數據
        if (!ServletFileUpload.isMultipartContent(request)) {
        	// 按照傳統方式獲取數據
        	return;
        }
        // 設置上傳單個文件的大小的最大值，目前是設置為1024*1024*100字節，
        // 也就是100MB
        upload.setFileSizeMax(1024 * 1024 * 100);
        // 設置上傳文件總量的最大值，最大值=同時上傳的多個文件的大小的
        // 最大值的和，目前設置為1000MB
        upload.setSizeMax(1024 * 1024 * 1000);
        // 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");					
            } else {// 如果fileitem中封裝的是上傳文件
            	// 得到上傳的文件名稱，
            	String filename = item.getName();
            	System.out.println(filename);
            	if (filename == null || 
            	filename.trim().equals("")) {
            		continue;
            	}
                // 注意：不同的瀏覽器提交的文件名是不一樣的，有些瀏覽器提交上來的
                // 文件名是帶有路徑的，如： d:\storge\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 = 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 = "文件上傳成功！";
            	// 文件上傳成功后，將對應的信息保存到數據庫SYS_FILE表
                Integer userId 
                        = Integer.valueOf(request.getParameter("userId"));
            	Date date = new Date(System.currentTimeMillis());
            	String fileId = System.currentTimeMillis() + "";
            	String fileKey = userId + "";
            	Session session = SessionUtil.getSession();
            	String sql = "insert INTO SYS_FILE VALUES(FILE_SEQ.nextval,";
                       sql+="1,1,?,?,?,sysdate,1,sysdate)";
            	SQLQuery query = session.createSQLQuery(sql);
            	query.setString(0, fileKey);
            	query.setString(1, saveFilename);
            	query.setInteger(2, userId);
            	Transaction ts = session.beginTransaction();
            	try {
                    query.executeUpdate();
                    ts.commit();
                    System.out.println("插入成功");
            	} catch (HibernateException e) {
                    e.printStackTrace();
                    ts.rollback();
                    System.out.println("插入失敗");
            	} finally {
                    session.close();
            	}
             }
            }
        } catch (FileUploadBase
            .FileSizeLimitExceededException e) {
        	e.printStackTrace();
        	request.setAttribute(
        	    "message", "單個文件超出最大值！");
        	return;
        } catch (FileUploadBase.SizeLimitExceededException e) {
        	e.printStackTrace();
        	request.setAttribute(
        	    "message","上傳文件的總大小超出限制的最大值！");
        	return;
        } catch (Exception e) {
        	message = "文件上傳失敗！";
        	e.printStackTrace();
        }
        request.setAttribute("message", message);
        }
    /**
     * @Method: makeFileName
     * @Description: 生成上傳文件的文件名，文件名以：uuid+"_"+文件的原始名稱
     * @param filename文件的原始名稱
     * @return uuid+"_"+文件的原始名稱
     */
  private String makeFileName(String filename) {
     // 為防止文件覆蓋的現象發生，要為上傳文件產生一個唯一的文件名
     return UUID.randomUUID().toString() + "_" + filename;
  }
 
}

顯示當前（用戶id為1，業務類型id為1）的所有文件的servlet：

public class ListFileServlet extends HttpServlet {
  public void service(HttpServletRequest request, HttpServletResponse response)
  throws ServletException, IOException {
        Integer userId = Integer.valueOf(request.getParameter("userId"));
        Integer busiTypeId 
                    = Integer.valueOf(request.getParameter("busiTypeId"));
        // 獲取上傳文件的目錄
        String uploadFilePath = this.getServletContext()
                        .getRealPath("/WEB-INF/upload");
        // 存儲要下載的文件名
        Map<String, String> fileNameMap = new HashMap<String, String>();
        // 遞歸遍歷filepath目錄下的所有文件和目錄，將文件的文件名存儲到map集合中
        listfile(new File(uploadFilePath),fileNameMap,userId,busiTypeId);
        // File既可以代表一個文件也可以代表一個目錄
        Iterator<Map.Entry<String, String>> it
                        = fileNameMap.entrySet().iterator();
        while (it.hasNext()) {
          Map.Entry<String, String> entry = it.next();
        }
        // 將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,Integer userId,
    Integer busiTypeId){
        Session session = SessionUtil.getSession();
        String sql = "select name from SYS_ANNEX where user_id=?"
               sql += "and busi_type_id=?";
        SQLQuery query = session.createSQLQuery(sql);
        query.setInteger(0, userId);
        query.setInteger(1, busiTypeId);
        List<String> list = query.list();
        for (String fileName : list) {
 
       /**
        * 處理文件名，上傳后的文件是以uuid_文件名的形式去重新命
        * 名的，去除文件名的uuid_部分file.getName().indexOf("_")
        * 檢索字符串中第一次出現"_"字符的位置，如果文件名
        * 類似於：1554846223-75648-3264滑稽.jpg
        * file.getName().substring(file.getName().indexOf("_")+1)
        * 處理之后就可以得到滑稽.jpg這部分名字
        */
        String realName = fileName.substring(fileName.indexOf("_") + 1);
        // file.getName()得到的是文件的原始名稱，這個名稱是
        // 唯一的，因此可以作為key，realName是處理過后的名稱，
        // 有可能會重復
        map.put(fileName, realName);
        }
        session.close();
   }
 
}

下載模塊

public class DownLoadServlet extends HttpServlet {
  public void service(HttpServletRequest request, HttpServletResponse response)
  throws ServletException, IOException {
        // 得到要下載的文件名
        String fileName = request.getParameter("filename");
        // 上傳的文件都是保存在/WEB-INF/upload目錄下的子目錄當中
        String fileSaveRootPath = this.getServletContext()
                                .getRealPath("/WEB-INF/upload");
        // 通過文件名找出文件的所在目錄
        String path = fileSaveRootPath;
        // 得到要下載的文件
        File file = new File(path + "\\" + fileName);
        // 如果文件不存在
        if (!file.exists()) {
        	request.setAttribute("message", "您要下載的資源已被刪除！");
        	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();
  }
}

刪除模塊

public class DeleteFileServlet extends HttpServlet{
    @Override
    protected void service(HttpServletRequest request, HttpServletResponse response)
	throws ServletException, IOException {
        //獲取文件名
        String fileName = request.getParameter("fileName");
        // 數據庫刪除記錄
        Session session=SessionUtil.getSession();
        String sql="delete from SYS_FILE where name=?";
        SQLQuery query=session.createSQLQuery(sql);
        query.setString(0,fileName);
        Transaction ts=session.beginTransaction();
        try{
            query.executeUpdate();
            ts.commit();
        }catch(HibernateException e){
            e.printStackTrace();
            ts.rollback();
            return;
        }finally{
            session.close();
        }
 
        // 數據庫成功刪除記錄后，再從硬盤里刪除文件
        //上傳的文件都是保存在/WEB-INF/upload目錄下的子目錄當中
        String fileSaveRootPath = this.getServletContext()
                                .getRealPath("/WEB-INF/upload");
        //通過文件名找出文件的所在目錄
        String path = fileSaveRootPath;
        //得到要刪除的文件
        File file = new File(path + "\\" + fileName);
        //刪除文件
        file.delete();
 
	}
}

Session工具類

用於創建查詢

public class SessionUtil {
    private static SessionFactory sf;
    static{
    	Configuration conf=new Configuration();
    	conf.configure("hibernate.cfg.xml");
    	sf=conf.buildSessionFactory();
    }
    public static Session getSession(){
    	return sf.openSession();
    }
    public static SessionFactory getSf(){
    	return sf;
    }
}

表結構SYS_FILE

create table SYS_FILE(
   FILE_ID             VARCHAR2(20)         not null, --文件id
   AREA_ID              NUMBER(5),                                  
   BUSI_TYPE_ID         NUMBER(5),           --業務類型id，這里我們默認為1
   FILE_KEY             VARCHAR2(20)         not null,   
   NAME                 VARCHAR2(100)        not null, --文件名
   USER_ID              NUMBER(12)           not null, --用戶id，這里我們默認為1
   CREATE_DATE          DATE                 not null, --創建時間
   STATE_ID             NUMBER(5)            not null,
   STATE_DATE           DATE                 not null,
   constraint PK_SYS_FILE primary key (FILE_ID)
);