package com.xiaoxing.upload;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Iterator;
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.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

/**
 * <h1>Apache Fileupload文件上傳（2014-5-3）</h1>
 * <p>1、如果你對(duì)本示例感興趣并想了解更多，歡迎加入Java私塾在線學(xué)習(xí)社區(qū)（329232140）</p>
 * <p>2、針對(duì)這個(gè)例子小修小改便可移植到你的實(shí)際項(xiàng)目中。</p>
 */
public class FileUploadServlet extends HttpServlet {

    private static final long serialVersionUID = 7579265950932321867L;

    // 設(shè)置文件默認(rèn)上傳目錄（如果你沒(méi)有在web.xml中配置的話(huà)）
    private String uploadDir = "c:/"; // 文件上傳目錄
    private String tempUploadDir = "c:/"; // 文件臨時(shí)存放目錄（會(huì)話(huà)銷(xiāo)毀后由監(jiān)聽(tīng)器自動(dòng)刪除）

    /*
     * (non-Javadoc)
     * @see javax.servlet.GenericServlet#init()
     * 如果在web.xml中配置了文件上傳目錄則優(yōu)先使用，判斷文件目錄是否存在，不存在就創(chuàng)建。
     */
    @Override
    public void init() throws ServletException {
        // 獲取本項(xiàng)目所在真實(shí)硬盤(pán)目錄
        String path = getClass().getProtectionDomain().getCodeSource().getLocation().getPath();
        path = path.substring(0, path.indexOf("WEB-INF"));
        // 判斷目標(biāo)是否存在，不存在就建立
        String uploadDir = path.concat(this.getInitParameter("uploadDir"));
        String tempUploadDir = path.concat(this.getInitParameter("tempUploadDir"));
        File f_uploadDir = new File(uploadDir);
        File f_tempUploadDir = new File(tempUploadDir);
        if (!f_uploadDir.exists()) {
            f_uploadDir.mkdirs();
        }
        if (!f_tempUploadDir.exists()) {
            f_tempUploadDir.mkdirs();
        }
        // 給變量賦值
        this.uploadDir = uploadDir;
        this.tempUploadDir = tempUploadDir;
    }

    /*
     * (non-Javadoc)
     * @see javax.servlet.http.HttpServlet#doGet(javax.servlet.http.HttpServletRequest, javax.servlet.http.HttpServletResponse)
     * 不接收get方式提交的數(shù)據(jù)，返回上傳失敗狀態(tài)碼。
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        this.setResponse(response);
        PrintWriter out = response.getWriter();
        out.print("{\"error\":\"-1\""); // 非法提交方式
    }

    /*
     * (non-Javadoc)
     * @see javax.servlet.http.HttpServlet#doPost(javax.servlet.http.HttpServletRequest, javax.servlet.http.HttpServletResponse)
     * 上傳文件請(qǐng)求都是通常POST提交
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        this.setResponse(response); // 設(shè)置響應(yīng)類(lèi)型，以便前端解析
        PrintWriter out = response.getWriter();
        String result = "";
        try {
            // 檢查本次是否一個(gè)文件上傳請(qǐng)求
            boolean isMultipart = ServletFileUpload.isMultipartContent(request);
            if (isMultipart) {
                DiskFileItemFactory factory = new DiskFileItemFactory(); // 創(chuàng)建一個(gè)工廠基于磁盤(pán)的文件項(xiàng)
                factory.setRepository(new File(tempUploadDir)); // 配置儲(chǔ)存庫(kù)（確保安全的臨時(shí)位置時(shí)）
                ServletFileUpload upload = new ServletFileUpload(factory); // 創(chuàng)建一個(gè)新的文件上傳處理程序
                upload.setSizeMax(1024 * 1024 * 100); // 設(shè)置總體要求尺寸限制（建議前后臺(tái)分別設(shè)置，因?yàn)榍昂笈_(tái)用到了不同的插件）
                List<FileItem> items = upload.parseRequest(request); // 解析請(qǐng)求
                Iterator<FileItem> iter = items.iterator(); // 處理上傳的項(xiàng)目
                while (iter.hasNext()) { //如果是一次性上傳多個(gè)文件，那這里會(huì)分別去保存
                    FileItem item = iter.next();
                    if (!item.isFormField()) { // 過(guò)濾表單里的非文件類(lèi)型字段
                        if (!"".equals(item.getName())) { // 過(guò)濾非文件類(lèi)型的input
                            String s_name = item.getName(); // 獲得原始文件名
                            int position = s_name.lastIndexOf(".");
                            String s_fileType = s_name.substring(position, s_name.length()); // 獲得文件后綴
                            String date = new SimpleDateFormat("yyyyMMdd").format(new Date());
                            String s = uploadDir.concat("/").concat(date).concat("/");
                            //這里按日期分目錄保存文件
                            File sf = new File(s);
                            if (!sf.exists()) {
                                sf.mkdirs();
                            }
                            String s_filePath = s.concat(UUID.randomUUID().toString()).concat(s_fileType);
                            File path = new File(s_filePath);
                            item.write(path);
                            result += s_filePath.concat(",");
                        } else {
                            result = "";
                            break;
                        }
                    }
                }
            } else {
                result = "";
            }
            String s_resultJSON = this.jointJSON(result); // 拼接返回前端JSON
            out.print(s_resultJSON);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            out.flush();
            out.close();
        }
    }

    /**
     * 拼接JSON，將保存文件的文件名和日期目錄返回給前端（前端可能需要這個(gè)路徑完成其他表單操作，比如將文件路徑存放到數(shù)據(jù)庫(kù)）
     * @param result JSON格式的字符串
     * @return
     * @throws UnsupportedEncodingException
     */
    private String jointJSON (String result) throws UnsupportedEncodingException {
        String str = "";
        if(!"".equals(result)) {
            String rs[] = result.split(",");
            StringBuffer buffer = new StringBuffer("{\"rows\":[");
            for (int i = 0; i < rs.length; i++) {
                String s_tmpName = rs[i];
                s_tmpName = s_tmpName.substring(uploadDir.length(), s_tmpName.length());
                buffer.append("{\"name\":\"").append(s_tmpName).append("\"},");
            }
            str = buffer.toString();
            str = str.substring(0, str.length() - 1).concat("]}");
        } else {
            str = "{\"error\":\"-2\""; //上傳失敗
        }
        return str;
    }

    /**
     * 設(shè)置響應(yīng)類(lèi)型ContentType為"application/x-json"
     * @param response
     */
    private void setResponse(HttpServletResponse response) {
        response.setCharacterEncoding("UTF-8");
        response.setContentType("application/json;charset=UTF-8");
        response.setHeader("cache-control", "no-cache");
    }

}

<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<title>jQuery Uploadify + Apache Fileupload異步上傳文件示例(2014-5-3)</title>
<link rel="stylesheet" type="text/css" href="/js/uploadify/uploadify.css">
<script src="/js/jquery-1.9.0.js"></script>
<script src="/js/uploadify/jquery.uploadify.min.js"></script>
<script type="text/javascript">
$(function() {
 $('#fileupload').uploadify({
     'method'   : 'post',
     'buttonText' : 'flash上傳文件',
     'fileSizeLimit' : '1024KB',
     'fileTypeExts' : '*.gif; *.jpg; *.png',
  'swf'      : '/js/uploadify/uploadify.swf',
  'uploader' : '/upload', //這是上傳圖片的路徑，也就是我在web.xml里配置的servlet
  'onUploadSuccess' : function(file, data, response) { //圖片上傳成功后返回?cái)?shù)據(jù)在這里處理
      var ary = eval("(" + data + ")").rows;
      for(var i = 0; i < ary.length; i++) {
          $("#J_div").append("<img alt='圖片' src='/upload/images" + ary[i].name + "' width='200px' height='200px'>");
      }
     }
 });
});
</script>
</head>
<body>
 <h2>jQuery Uploadify + Apache Fileupload異步上傳文件示例(2014-5-3)</h2>
 <p>1、可以限制上傳文件大小和類(lèi)型，理論上任何類(lèi)型的文件都可以上傳（自己根據(jù)api配置即可）；</p>
 <p>2、后臺(tái)使用Apache commons-fileupload-1.3.1.jar作為上傳工具包，本示例支持一次性多文件上傳；</p>
 <p>3、文件上傳目錄可以任意指定，請(qǐng)?jiān)趙eb.xml中配置；</p>
 <p>4、對(duì)于已經(jīng)上傳的圖片沒(méi)有查詢(xún)到這個(gè)頁(yè)面上，這部分留給你去做吧。</p>
 <p>Uploadify api 詳見(jiàn)http://www.uploadify.com/documentation/</p>
 <p style="color: red">*如果你對(duì)本示例感興趣并想了解更多，歡迎加入Java私塾在線學(xué)習(xí)社區(qū)（329232140）。</p>
 <input id="fileupload" type="file" name="img" multiple="multiple"/>
 <div id="J_div"></div>
</body>
</html>

<?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_3_0.xsd" metadata-complete="true" version="3.0">
  <welcome-file-list>
   <welcome-file>test_upload.html</welcome-file>
  </welcome-file-list>

  <servlet>
    <description>專(zhuān)門(mén)用來(lái)處理上傳操作的servlet</description>
        <servlet-name>FileUploadServlet</servlet-name>
        <servlet-class>com.xiaoxing.upload.FileUploadServlet</servlet-class>
        <init-param>
         <description>文件存放的正式目錄，可以自己配置</description>
         <param-name>uploadDir</param-name>
         <param-value>/upload/images/</param-value>
        </init-param>
        <init-param>
         <description>文件存放的臨時(shí)目錄，可以自己配置，里的文件由下面配置的監(jiān)聽(tīng)器自動(dòng)刪除。</description>
         <param-name>tempUploadDir</param-name>
         <param-value>/upload/temp</param-value>
        </init-param>
    </servlet>
    <servlet-mapping>
        <servlet-name>FileUploadServlet</servlet-name>
        <url-pattern>/upload</url-pattern>
    </servlet-mapping>

    <listener>
     <description>臨時(shí)文件資源清理，工具包自帶，不用我們來(lái)寫(xiě)</description>
     <listener-class>org.apache.commons.fileupload.servlet.FileCleanerCleanup</listener-class>
   </listener>

</web-app>