import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.math.BigDecimal;
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.fileupload.FileItem;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.FileUploadBase.FileSizeLimitExceededException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import com.itheima.util.IOUtils;
public class UploadServlet extends HttpServlet {
public void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
response.setContentType("text/html;charset=utf-8");
try{
//1.创建工厂
DiskFileItemFactory factory = new DiskFileItemFactory();
factory.setSizeThreshold(100*1024);
factory.setRepository(new File(this.getServletContext().getRealPath("WEB-INF/temp")));
//2.生产文件上传核心类
ServletFileUpload fileUpload = new ServletFileUpload(factory);
//--检查是否是正确的文件上传表单
if(!fileUpload.isMultipartContent(request)){
throw new RuntimeException("请用正确的表单进行上传!");
}
//--设置文件上传的大小限制
fileUpload.setFileSizeMax(1024*1024*100);//单个文件不大于10M
fileUpload.setSizeMax(1024*1024*100);//总大小不大于100M
//--设置编码集,解决上传文件名的乱码问题
fileUpload.setHeaderEncoding("utf-8");
//--设置文件上传监听
fileUpload.setProgressListener(new ProgressListener(){
Long beginTime = System.currentTimeMillis();
public void update(long bytesRead, long contentLength, int items) {
BigDecimal br = new BigDecimal(bytesRead).divide(new BigDecimal(1024),2,BigDecimal.ROUND_HALF_UP);
BigDecimal cl = new BigDecimal(contentLength).divide(new BigDecimal(1024),2,BigDecimal.ROUND_HALF_UP);
System.out.print("当前读取的是第"+items+"个上传项,总大小"+cl+"KB,已经读取"+br+"KB");
//剩余字节数
BigDecimal ll = cl.subtract(br);
System.out.print("剩余"+ll+"KB");
//上传百分比
BigDecimal per = br.multiply(new BigDecimal(100)).divide(cl,2,BigDecimal.ROUND_HALF_UP);
System.out.print("已经完成"+per+"%");
//上传用时
Long nowTime = System.currentTimeMillis();
Long useTime = (nowTime - beginTime)/1000;
System.out.print("已经用时"+useTime+"秒");
//上传速度
BigDecimal speed = new BigDecimal(0);
if(useTime!=0){
speed = br.divide(new BigDecimal(useTime),2,BigDecimal.ROUND_HALF_UP);
}
System.out.print("上传速度为"+speed+"KB/S");
//大致剩余时间
BigDecimal ltime = new BigDecimal(0);
if(!speed.equals(new BigDecimal(0))){
ltime = ll.divide(speed,0,BigDecimal.ROUND_HALF_UP);
}
System.out.print("大致剩余时间为"+ltime+"秒");
System.out.println();
}
});
//3.利用文件上传核心类解析request
List<FileItem> list = fileUpload.parseRequest(request);
//4.遍历所有的FileItem
for(FileItem item : list){
if(item.isFormField()){
//当前是一个普通的字段项
String name = item.getFieldName();
String value = item.getString("utf-8");
System.out.println(name+":"+value);
}else{
//当前是一个文件上传项
String filename = item.getName();
String uuidName = UUID.randomUUID().toString()+"_"+filename;
int hash = uuidName.hashCode();
String hashStr = Integer.toHexString(hash);
char [] hss = hashStr.toCharArray();
String path = this.getServletContext().getRealPath("WEB-INF/upload");
for(char c : hss){
path+="/"+c;
}
new File(path).mkdirs();
InputStream in = item.getInputStream();
OutputStream out = new FileOutputStream(new File(path,uuidName));
IOUtils.In2Out(in, out);
IOUtils.close(in, out);
//--删除临时文件
item.delete();
}
}
}catch (FileSizeLimitExceededException e) {
response.getWriter().write("单个文件不超过10M,总大小不超过100M!");
return;
}catch (Exception e) {
e.printStackTrace();
throw new RuntimeException(e);
}
}
public void doPost(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
doGet(request, response);
}
}
//收集文件信息,保存到数据库中,将上传文件的路径,保存到数据库中
//设置Map
Map map=new HashMap();
//获取到上传者的ip
map.put("ip", request.getRemoteAddr());
for(FileItem item:fileItems){
if(item.isFormField()){
String name=item.getFieldName();
String value=item.getString("utf-8");
map.put(name, value);
}else{//文件上传
//--uuidname防止文件名重复
String realname = item.getName();
String uuidname = UUID.randomUUID().toString()+"_"+realname;
map.put("realname", realname);
map.put("uuidname", uuidname);
}
}
map.put("savepath", savepath);
//2.向数据库中插入数据
Resource r = new Resource();
BeanUtils.populate(r, map);
String sql ="insert into netdisk values(null,?,?,?,?,null,?)";
QueryRunner runner = new QueryRunner(DbUtil.getDataSource());
runner.update(sql,r.getUuidname(),r.getRealname(),r.getSavepath(),r.getIp(),r.getDescription());