upload.java

重点不是信息管理系统

package com.yijia_ctgu.servlet;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.text.SimpleDateFormat;
import java.util.Iterator;
import java.util.List;

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

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.servlet.*;
import org.apache.commons.fileupload.disk.*;
import org.apache.commons.io.output.DeferredFileOutputStream;
import org.json.simple.JSONObject;

import com.yijia_ctgu.DB.*;
public class Upload extends HttpServlet {
	File photoBasePath ; // 用于存放上传文件的目录  
    File tempPath =new File("d:\\tmp\\"); // 用于存放临时文件的目录  
    int maxMemorySize=4194304;
	/**
	 * Constructor of the object.
	 */
	public Upload() {
		super();
	}

	/**
	 * Destruction of the servlet. <br>
	 */
	public void destroy() {
		super.destroy(); // Just puts "destroy" string in log
		// Put your code here
	}

	/**
	 * The doGet method of the servlet. <br>
	 *
	 * This method is called when a form has its tag value method equals to get.
	 * 
	 * @param request the request send by the client to the server
	 * @param response the response send by the server to the client
	 * @throws ServletException if an error occurred
	 * @throws IOException if an error occurred
	 */
	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {

		doPost(request,response);
	}

	/**
	 * The doPost method of the servlet. <br>
	 *
	 * This method is called when a form has its tag value method equals to post.
	 * 
	 * @param request the request send by the client to the server
	 * @param response the response send by the server to the client
	 * @throws ServletException if an error occurred
	 * @throws IOException if an error occurred
	 */
	public void doPost(HttpServletRequest req, HttpServletResponse res)
			throws ServletException, IOException {
		HttpSession session=req.getSession();
		//System.out.println(session.getServletContext().getRealPath("/"));
		photoBasePath=new File(session.getServletContext().getRealPath("/")+"photo/");
		res.setCharacterEncoding("utf-8");
	    res.setContentType("text/html");
	    PrintWriter out = res.getWriter();
	    //System.out.println(req.getServletPath());
	    JSONObject obj=new JSONObject();
		String id=req.getParameter("id");
		String tableName=req.getParameter("tableName");
		//System.out.println(id+" "+tableName);
		try{
			DiskFileItemFactory factory = new DiskFileItemFactory(maxMemorySize,tempPath);
	        // Configure the factory here, if desired.
	        ServletFileUpload upload = new ServletFileUpload(factory);
	        // Configure the uploader here, if desired.
	        List fileItems = upload.parseRequest(req);
	        Iterator iter = fileItems.iterator();
	        while (iter.hasNext()) {
	            FileItem item = (FileItem) iter.next();
	            if (!item.isFormField()) {
	                String fieldName = item.getFieldName();
	                String fileName = item.getName();
	                byte[] bytes=fileName.getBytes();
	                fileName=new String(bytes,"utf-8");
	                //String contentType = item.getContentType();
	                //boolean isInMemory = item.isInMemory();
	                //long sizeInBytes = item.getSize();
	                //System.out.println(fieldName+" "+fileName+" "+contentType+" "+sizeInBytes);
	                if(fieldName.equals("photo")){ 
	                	photoBasePath.mkdirs();
	                	Timestamp now=new Timestamp((new java.util.Date()).getTime());  
	                    SimpleDateFormat fmt= new SimpleDateFormat("yyyyMMddHHmmss");
	                    String  pfileName=fmt.format(now).toString().trim(); 
	                	DB.getTable(tableName).getRow(id).set("photo",pfileName+fileName);
		                item.write(new File(photoBasePath,pfileName+fileName));
		                obj.put("success", true);
		                obj.put("message", "上传照片成功");
	                }       	
	            }
	        }
			
		}catch(SQLException e){
			obj.put("success", false);
			obj.put("error", "数据库错误");
		}catch(FileUploadException e){
			obj.put("success", false);
			obj.put("error", "上传过程错误");
		}catch(Exception e){
			obj.put("success", false);
			obj.put("error", "未知的错误");
		}
		out.println(obj);
	}

	/**
	 * Initialization of the servlet. <br>
	 *
	 * @throws ServletException if an error occure
	 */
	public void init() throws ServletException {
		// Put your code here
	}

}