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();
}
}