xheditor-文件上传-java-支持html5-application/octet-stream

package reyo.sdk.utils.file;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.io.Serializable;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
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.DiskFileUpload;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.lang.StringUtils;

public class UploadFileServlet extends HttpServlet {

	private static final long serialVersionUID = 1541334866883495283L;

	private static String baseDir = "/UploadFile/"; // 上传文件存储目录

	private static String fileExt = "jpg,jpeg,bmp,gif,png,txt";

	private static Long maxSize = 0l;

	private static String dirType = "1"; // 0:不建目录 1:按天存入目录 2:按月存入目录 3:按扩展名存目录
											// 建议使用按天存

	public void init() throws ServletException {

		baseDir = this.getInitParameter("baseDir"); // 获取web.xml中servlet的配置文件目录参数

		if (StringUtils.isEmpty(baseDir))
			baseDir = "/UploadFile/"; // 获取文件上传存储的相当路径

		String realBaseDir = this.getServletConfig().getServletContext().getRealPath(baseDir);

		File baseFile = new File(realBaseDir);

		if (!baseFile.exists()) {

			baseFile.mkdir();

		}

		fileExt = this.getInitParameter("fileExt"); // 获取文件类型参数

		if (StringUtils.isEmpty(fileExt))
			fileExt = "jpg,jpeg,gif,bmp,png,txt";

		String maxSize_str = this.getInitParameter("maxSize"); // 获取文件大小参数

		if (StringUtils.isNotEmpty(maxSize_str)) {

			maxSize = new Long(maxSize_str);

		} else {

			maxSize = Long.valueOf("5242880"); // 5M

		}

		dirType = this.getInitParameter("dirType"); // 获取文件目录类型参数

		if (StringUtils.isEmpty(dirType))
			dirType = "1";

		if (",0,1,2,3,".indexOf("," + dirType + ",") < 0)
			dirType = "1";

	}

	// 上传文件数据处理过程
	public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

		response.setContentType("text/html; charset=UTF-8");

		response.setHeader("Cache-Control", "no-cache");

		String err = "";

		String newFileName = "";

		if ("application/octet-stream".equals(request.getContentType())) { // HTML
																			// 5
																			// 上传

			try {

				String dispoString = request.getHeader("Content-Disposition");

				int iFindStart = dispoString.indexOf("name=\"") + 6;

				int iFindEnd = dispoString.indexOf("\"", iFindStart);

				iFindStart = dispoString.indexOf("filename=\"") + 10;

				iFindEnd = dispoString.indexOf("\"", iFindStart);

				String sFileName = dispoString.substring(iFindStart, iFindEnd);

				int i = request.getContentLength();

				byte buffer[] = new byte[i];

				int j = 0;

				while (j < i) { // 获取表单的上传文件

					int k = request.getInputStream().read(buffer, j, i - j);

					j += k;

				}

				if (buffer.length == 0) { // 文件是否为空

					printInfo(response, "上传文件不能为空", "");

					return;

				}

				if (maxSize > 0 && buffer.length > maxSize) { // 检查文件大小

					printInfo(response, "上传文件的大小超出限制", "");

					return;

				}

				String filepathString = getSaveFilePath(sFileName, response);

				if ("不允许上传此类型的文件".equals(filepathString))
					return; // 检查文件类型

				OutputStream out = new BufferedOutputStream(new FileOutputStream(this.getServletConfig().getServletContext().getRealPath("") + filepathString, true));

				out.write(buffer);

				out.close();

				newFileName = request.getContextPath() + filepathString;

			} catch (Exception ex) {

				System.out.println(ex.getMessage());

				newFileName = "";

				err = "错误: " + ex.getMessage();

			}

		} else {

			DiskFileUpload upload = new DiskFileUpload();

			try {

				List<FileItem> items = upload.parseRequest(request);

				Map<String, Serializable> fields = new HashMap<String, Serializable>();

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

				while (iter.hasNext()) {

					FileItem item = (FileItem) iter.next();

					if (item.isFormField())

						fields.put(item.getFieldName(), item.getString());

					else

						fields.put(item.getFieldName(), item);

				}

				FileItem uploadFile = (FileItem) fields.get("filedata"); // 获取表单的上传文件

				String fileNameLong = uploadFile.getName(); // 获取文件上传路径名称

				if (uploadFile.getSize() == 0) { // 文件是否为空

					printInfo(response, "上传文件不能为空", "");

					return;

				}

				if (maxSize > 0 && uploadFile.getSize() > maxSize) { // 检查文件大小

					printInfo(response, "上传文件的大小超出限制", "");

					return;

				}

				String filepathString = getSaveFilePath(fileNameLong, response);

				if ("不允许上传此类型的文件".equals(filepathString))
					return; // 检查文件类型

				File savefile = new File(this.getServletConfig().getServletContext().getRealPath("") + filepathString);

				uploadFile.write(savefile); // 存储上传文件

				newFileName = request.getContextPath() + filepathString;

			} catch (Exception ex) {

				System.out.println(ex.getMessage());

				newFileName = "";

				err = "错误: " + ex.getMessage();

			}

		}

		printInfo(response, err, newFileName);

	}

	public String getSaveFilePath(String sFileName, HttpServletResponse response) throws IOException {

		String extensionName = sFileName.substring(sFileName.lastIndexOf(".") + 1); // 获取文件扩展名

		if (("," + fileExt.toLowerCase() + ",").indexOf("," + extensionName.toLowerCase() + ",") < 0) { // 检查文件类型

			printInfo(response, "不允许上传此类型的文件", "");

			return "不允许上传此类型的文件";

		}

		String fileFolder = ""; // 0:不建目录, 1:按天存入目录, 2:按月存入目录,
								// 3:按扩展名存目录.建议使用按天存。

		if (dirType.equalsIgnoreCase("1"))
			fileFolder = new SimpleDateFormat("yyyyMMdd").format(new Date());

		if (dirType.equalsIgnoreCase("2"))
			fileFolder = new SimpleDateFormat("yyyyMM").format(new Date());

		if (dirType.equalsIgnoreCase("3"))
			fileFolder = extensionName.toLowerCase();

		String saveDirPath = baseDir + fileFolder + "/"; // 文件存储的相对路径

		String saveFilePath = this.getServletConfig().getServletContext().getRealPath("") + saveDirPath; // 文件存储在容器中的绝对路径

		File fileDir = new File(saveFilePath); // 构建文件目录以及目录文件

		if (!fileDir.exists()) {

			fileDir.mkdirs();

		}

		String filename = UUID.randomUUID().toString(); // 重命名文件

		return saveDirPath + filename + "." + extensionName;

	}

	// 使用I/O流输出 json格式的数据

	public void printInfo(HttpServletResponse response, String err, String newFileName) throws IOException {

		response.setContentType("text/plain");

		response.setCharacterEncoding("UTF-8");

		PrintWriter out = response.getWriter();

		out.println("{\"err\":\"" + err + "\",\"msg\":\"" + newFileName + "\"}");

		out.flush();

		out.close();

	}

}