import java.io.*;
import javax.servlet.*;
import javax.servlet.http.*;
import org.apache.commons.fileupload.*;
import java.util.*;

public class UploadServlet extends HttpServlet
{
	public void doPost(HttpServletRequest request,
		HttpServletResponse response) throws ServletException,IOException
	{		
		response.setContentType("text/html;charset=gb2312");
		PrintWriter out = response.getWriter();
   				    		
		//設置保存上傳文件的目錄
		String uploadDir = getServletContext().getRealPath("/upload");
		if (uploadDir == null)
		{
			out.println("無法訪問存儲目錄！");
			return;
		}
		File fUploadDir = new File(uploadDir);
		if(!fUploadDir.exists())
		{
			if(!fUploadDir.mkdir())
			{
				out.println("無法創建存儲目錄!");
				return;
			}
		}
    
		if (!DiskFileUpload.isMultipartContent(request)) 
		{
			out.println("只能處理multipart/form-data類型的數據!");
			return ;
		}
    		
		DiskFileUpload fu = new DiskFileUpload();
		//最多上傳200M數據
		fu.setSizeMax(1024 * 1024 * 200);
		//超過1M的字段數據采用臨時文件緩存
		fu.setSizeThreshold(1024 * 1024);
		//采用默認的臨時文件存儲位置
		//fu.setRepositoryPath(...);
		//設置上傳的普通字段的名稱和文件字段的文件名所采用的字符集編碼
		fu.setHeaderEncoding("gb2312");
		
		//得到所有表單字段對象的集合
		List fileItems = null;
		try
		{
			fileItems = fu.parseRequest(request);
		}
		catch (FileUploadException e) 
		{
			out.println("解析數據時出現如下問題：");
			e.printStackTrace(out);
			return;
		}
		
		//處理每個表單字段
		Iterator i = fileItems.iterator();
		while (i.hasNext()) 
		{
			FileItem fi = (FileItem) i.next();
			if (fi.isFormField()) 
			{
				String content = fi.getString("GB2312");
				String fieldName = fi.getFieldName();
				request.setAttribute(fieldName,content);    				
			}
			else
			{
				try 
				{
					String pathSrc = fi.getName();
					/*如果用戶沒有在FORM表單的文件字段中選擇任何文件，
					那么忽略對該字段項的處理*/
					if(pathSrc.trim().equals(""))
					{
						continue;
					}
					int start = pathSrc.lastIndexOf('\\');
					String fileName = pathSrc.substring(start + 1);
					File pathDest = new File(uploadDir, fileName);    	

					fi.write(pathDest);
					String fieldName = fi.getFieldName();
					request.setAttribute(fieldName, fileName);
				}
				catch (Exception e) 
				{
    				out.println("存儲文件時出現如下問題：");
    				e.printStackTrace(out);
    				return;
				}
				finally  //總是立即刪除保存表單字段內容的臨時文件
				{
					fi.delete();
				}
			}
		}
		
		//顯示處理結果
		out.println("用戶：" + request.getAttribute("author") + "<br>");
		out.println("來自：" + request.getAttribute("company") + "<br>");
		
		/*將上傳的文件名組合成“file1,file2”這種形式顯示出來，如果沒有上傳
		 *任何文件，則顯示“無”，如果只上傳了第二個文件，顯示為“file2”。*/
		StringBuffer filelist = new StringBuffer();
		String file1 = (String)request.getAttribute("file1");
		makeUpList(filelist,file1);
		String file2 = (String)request.getAttribute("file2");
		makeUpList(filelist,file2);
		out.println("成功上傳的文件：" + 
			(filelist.length()==0 ? "無" : filelist.toString()));
	}
	
	/**
	 *將一段字符串追加到一個結果字符串中。如果結果字符串的初始內容不為空，
	 *在追加當前這段字符串之前先最加一個逗號（,）。在組合sql語句的查詢條件時，
	 *經常要用到類似的方法，第一條件前沒有"and"，而后面的條件前都需要用"and"
	 *作連詞，如果沒有選擇第一個條件，第二個條件就變成第一個，依此類推。
	 *
	 *@param result 要將當前字符串追加進去的結果字符串
	 *@param fragment 當前要追加的字符串
	 */
	private void makeUpList(StringBuffer result,String fragment)
	{
		if(fragment != null)
		{
			if(result.length() != 0)
			{
				result.append(",");
			}
			result.append(fragment);
		}		
	}
}