FileUploadServlet 资料上传

FileUploadServlet 文件上传

package com.appdev.bsf.server.servlet;

import java.io.File;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import javax.servlet.ServletContext;
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.FileItemFactory;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;

import com.appdev.bsf.common.client.datasource.DictionaryGwtRpcService;
import com.ibm.icu.text.SimpleDateFormat;

public class FileUploadServlet extends HttpServlet {
??? private static final long serialVersionUID = 1L;
??? private DictionaryGwtRpcService service;
??? private ServletFileUpload upload;
??? private String year;
??? private String month;
??? private String day;

??? @Override
??? public void init() throws ServletException {
??? ??? FileItemFactory factory = new DiskFileItemFactory();
??? ??? upload = new ServletFileUpload(factory);

??? ??? ServletContext application;
??? ??? WebApplicationContext wac;

??? ??? application = getServletContext();
??? ??? wac = WebApplicationContextUtils.getWebApplicationContext(application);
??? ??? service = (DictionaryGwtRpcService) wac.getBean("dictionaryGwtRpcServiceImpl");

??? ??? Date date = new Date();
??? ??? SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
??? ??? String[] dateArr = sdf.format(date).split("-");
??? ??? year = dateArr[0];
??? ??? month = dateArr[1];
??? ??? day = dateArr[2];
??? }

??? @Override
??? protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException,
??? ??? ??? IOException {
??? ??? response.setContentType("text/html;charset=utf-8");

??? ??? Integer id = null;
??? ??? String clsUrl = null;
??? ??? Map<String, String> data = null;
??? ??? try {
??? ??? ??? id = Integer.valueOf(request.getParameter("id"));
??? ??? ??? clsUrl = request.getParameter("clsUrl");
??? ??? ??? data = new HashMap<String, String>();
??? ??? } catch (Exception e) {
??? ??? ??? e.printStackTrace();
??? ??? }

??? ??? List<?> items = null;
??? ??? upload.setSizeMax(1024 * 1024 * 2);
??? ??? try {

??? ??? ??? items = upload.parseRequest(request);
??? ??? } catch (FileUploadException ex) {
??? ??? ??? throw new ServletException(ex);
??? ??? }

??? ??? Iterator<?> iter = items.iterator();

??? ??? while (iter.hasNext()) {
??? ??? ??? FileItem item = (FileItem) iter.next();
??? ??? ??? if (item.isFormField() == false) {
??? ??? ??? ??? operatorFile(item, id, clsUrl, data);
??? ??? ??? }
??? ??? }
??? }

??? private void operatorFile(FileItem item, Integer id, String clsUrl, Map<String, String> data) {
??? ??? String fileName = null;
??? ??? // 获得上传路径
??? ??? String path = "/uploads/" + year + "/" + month + "/" + day;
??? ??? String savePath = getServletContext().getRealPath(path);

??? ??? // 获得文件扩展名
??? ??? String ext = item.getName().substring(item.getName().lastIndexOf("."));

??? ??? // 取的file域的值的名字，不带路径
??? ??? int pos = item.getName().lastIndexOf("\");
??? ??? String fieldValue = item.getName().substring(pos + 1);
??? ??? // 或直接保存成文件

??? ??? // 判断当天的文件夹是否存在,按年/月/日创建,如果不存在则创建
??? ??? if (!new File(savePath).isDirectory()) {
??? ??? ??? new File(savePath).mkdirs();
??? ??? }
??? ??? // 判断是否有扩展名exe
??? ??? if ("exe".equals(ext)) {
??? ??? ??? return;
??? ??? }
??? ??? File file = new File(savePath, fieldValue);
??? ??? try {
??? ??? ??? fileName = item.getFieldName();
??? ??? ??? // 写文件
??? ??? ??? item.write(file);
??? ??? ??? // 存路径
??? ??? ??? data.put(fileName, path + "/" + item.getName());

??? ??? ??? // 数据库写值
??? ??? ??? service.updateFiledUrl(clsUrl, id, data);
??? ??? } catch (Exception e) {
??? ??? ??? e.printStackTrace();
??? ??? }// 直接保存文件
??? }
}