package cn.csdn.web.servlet;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.SQLException;
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 javax.sql.DataSource;
import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadBase.FileSizeLimitExceededException;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import cn.csdn.web.c3p0.DBManager_c3p0;
public class Upload2Servlet extends HttpServlet {
/**
* 
*/
private static final long serialVersionUID = 1L;
public void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
doPost(request,response);
}
public void doPost(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
request.setCharacterEncoding(\"UTF-8\");
try {
//实例化一个文件工厂 
DiskFileItemFactory factory=new DiskFileItemFactory();
factory.setRepository(new File(\"C:\\\\osp\"));
String paramName=null;
String paramValue=null;
//配置上传组件ServletFileUpload 
ServletFileUpload upload=new ServletFileUpload(factory);
upload.setHeaderEncoding(\"UTF-8\");
upload.setFileSizeMax(1024*1024);
//从request得到所有上传域的列表 
List<FileItem> list=upload.parseRequest(request);
for(FileItem item:list){
//如果是上传域的文件域 
if(item.isFormField()){
//表单普通输入项
paramName = item.getFieldName(); //上传于的Name
// String paramValue=item.getString();
// paramValue=new String(paramValue.getBytes(\"iso8859-1\"),\"UTF-8\");
paramValue=item.getString(\"UTF-8\");
System.out.println(paramName+\"=\"+paramValue);
}else{
//上传文件处理
String fileName = item.getName();
fileName=fileName.substring(fileName.lastIndexOf(\"\\\\\")+1); //截取扩展名 
System.out.println(\"name=\"+fileName);
if(!fileName.equals(\"\")){
// fileName=refactorFileName(fileName);
InputStream in=item.getInputStream();
File file = new File(\"c:\\\\\"+fileName);
FileOutputStream os=new FileOutputStream(file);
byte[] buf = new byte[1024];
int len=0;
while((len=in.read(buf))>0){
os.write(buf,0,len);
}
os.flush();
os.close();
in.close();
item.delete();
request.setAttribute(\"message\", \"文件上传成功\");
try {
DataSource ds = DBManager_c3p0.getDataSource();
QueryRunner runner = new QueryRunner(ds);
String sql = \"insert into user(name,file) values(?,?)\";
Object[] params = {paramValue,fileName};
runner.update(sql, params);
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
}
} catch(FileSizeLimitExceededException e1){
e1.printStackTrace();
request.setAttribute(\"message\", \"文件尺寸太大\");
}catch (FileUploadException e) {
// TODO Auto-generated catch block
e.printStackTrace();
request.setAttribute(\"message\", \"文件上传失败\");
}
request.getRequestDispatcher(\"/message.jsp\").forward(request, response);
}
// public String refactorFileName(String fileName){
// return UUID.randomUUID().toString()+\"_\"+fileName;
// }
}