package com.peng.web;
import java.io.BufferedInputStream;
import
java.io.BufferedOutputStream;
import java.io.File;
import
java.io.FileInputStream;
import java.io.FileOutputStream;
import
java.io.IOException;
import java.io.InputStream;
import
java.util.List;
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.codec.digest.DigestUtils;
import
org.apache.commons.fileupload.FileItem;
import
org.apache.commons.fileupload.FileUploadException;
import
org.apache.commons.fileupload.disk.DiskFileItemFactory;
import
org.apache.commons.fileupload.servlet.ServletFileUpload;
public class uploadServlet extends HttpServlet {
private static final
long serialVersionUID = 1L;
@SuppressWarnings("unchecked")
public void doPost(HttpServletRequest
request, HttpServletResponse response)
throws ServletException,
IOException {
//上传路径
File file =new
File("E:/upload");
//上传临时路径
File tFile = new
File("E:/tempupload");
//判断文件夹是否存在，如果不存在就创建
if (!file.exists())
{
file.mkdir();
}
//判断临时文件夹是否存在，如果不存在就创建
if (!tFile.exists())
{
tFile.mkdir();
}
//判断form中enctype="multipart/form-data"有没有写
if
(ServletFileUpload.isMultipartContent(request)) {
DiskFileItemFactory
factory = new
DiskFileItemFactory();
//设置缓冲区大小
factory.setSizeThreshold(10*1024);
//设置临时文件夹
factory.setRepository(tFile);
ServletFileUpload
upload = new
ServletFileUpload(factory);
//设置文件上传的大小
upload.setSizeMax(10*1024*1024);
try
{
//获取表单元素传过来的值
List<FileItem> fileItems =
upload.parseRequest(request);
for(FileItem item : fileItems){
if
(item.isFormField())
{
//普通的表单元素
item.getFieldName();//获取表单的name
item.getString();//获取表单的值
}else{
//文件元素，进行io操作
item.getName();//上传文件的文件名
item.getFieldName();//获取文件的name属性
item.getContentType();//获取文件的类型
item.getSize();//获取文件的大小
InputStream
inputStream =
item.getInputStream();
//根据MD5的值判断文件系统中是否存在被上传的文件
String path =
haveFile(inputStream);
if(path != null)
{
System.out.println("文件已经存在：" + path);
} else
{
BufferedInputStream bis = new
BufferedInputStream(item.getInputStream());
String newFileName =
getFileNewName(item.getName());
BufferedOutputStream bos = new
BufferedOutputStream(new FileOutputStream(new
File(file,newFileName)));
byte[] buffer = new
byte[1024];
int len = -1;
while((len = bis.read(buffer))
!= -1) {
bos.write(buffer, 0,
len);
}
bos.flush();
bos.close();
bis.close();
}
}
}
}
catch (FileUploadException e)
{
e.printStackTrace();
}
}
}
//防止名字重复
private
String getFileNewName(String name) {
String extName =
name.substring(name.lastIndexOf("."));
UUID uuid =
UUID.randomUUID();
String newName = uuid.toString() + extName;
return
newName;
}
//查看是否有相同的文件
public String haveFile(InputStream
inputStream) {
String path = null;
try {
String md5 =
DigestUtils.md5Hex(inputStream);
File file = new
File("E:/upload");
File[] files = file.listFiles();
for(File subFile
: files) {
InputStream ins = new FileInputStream(subFile);
String
subFileMd5 = DigestUtils.md5Hex(ins);
if(md5.equals(subFileMd5))
{
path = subFile.getPath();
}
}
} catch
(IOException e) {
e.printStackTrace();
}
return
path;
}
}