文件的上传与下载

文件的上传操作：

package cn.itcast.web.servlet;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.io.IOUtils;

import cn.itcast.domain.ResourceBean;
import cn.itcast.service.ResourceService;
import cn.itcast.utils.FileUploadUtils;

import com.sun.org.apache.commons.beanutils.BeanUtils;

public class UploadServlet extends HttpServlet {

	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		response.setContentType("text/html;charset=utf-8");
		Map<String, String[]> map = new HashMap<String, String[]>();
		// 1.创建 DiskFileItemFactory
		File file = new File(this.getServletContext().getRealPath("/upload"));// 获取upload目录部署到tomcat后的绝对磁盘路径
		DiskFileItemFactory factory = new DiskFileItemFactory(1024 * 100, file); // 使用默认的.

		// 2.创建ServletFileUpload
		ServletFileUpload upload = new ServletFileUpload(factory);
		boolean flag = upload.isMultipartContent(request); // 用于判断是否是上传操作.
		if (flag) {
			// 解决上传文件名称中文乱码
			upload.setHeaderEncoding("utf-8");

			// 设置上传文件大小
			// upload.setSizeMax(1024 * 1024 * 10);// 总大小为10m

			try {
				List<FileItem> items = upload.parseRequest(request);// 解决request,得到所有的上传项FileItem

				// 3.得到所有上传项
				for (FileItem item : items) {
					if (!item.isFormField()) {
						// 上传组件

						String name = item.getName(); // 上传文件名称
						// 得到上传文件真实名称

						String filename = FileUploadUtils.getRealName(name);
						System.out.println(filename+"filename");
						map.put("realname", new String[] { filename });
						// 得到随机名称
						String uuidname = FileUploadUtils
								.getUUIDFileName(filename);
						map.put("uuidname", new String[] { uuidname });
						// 得到随机目录
						String randomDirectory = FileUploadUtils
								.getRandomDirectory(filename);
						System.out.println("randomDir"+randomDirectory);
						// 注意:随机目录可能不存在，需要创建.
						String parentPath = this.getServletContext()
								.getRealPath("/upload");// 本地磁盘路径
						File rd = new File(parentPath, randomDirectory);
						System.out.println("parent"+parentPath);
						if (!rd.exists()) {
							rd.mkdirs();
						}
						map.put("savepath", new String[] { parentPath
								+ randomDirectory });

						IOUtils.copy(item.getInputStream(),
								new FileOutputStream(new File(rd, uuidname)));

						// 删除临时文件
						item.delete();
					} else {
						map.put(item.getFieldName(),
								new String[] { (item.getString("utf-8")) });
					}
				}

				ResourceBean rb = new ResourceBean();
				BeanUtils.populate(rb, map);

				ResourceService service = new ResourceService();

				service.upload(rb);
				response.sendRedirect(request.getContextPath()+"/index.jsp");

			} catch (FileUploadException e) {
				// e.printStackTrace();
				response.getWriter().write(e.getMessage());
				return;
			} catch (IllegalAccessException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (InvocationTargetException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (SQLException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		} else {
			response.getWriter().write("不是上传操作");
			return;
		}
	}

	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		doGet(request, response);
	}

}

 

 

文件的下载操作： 

package cn.itcast.web.servlet;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.sql.SQLException;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import cn.itcast.domain.ResourceBean;
import cn.itcast.service.ResourceService;

public class DownloadServlet extends HttpServlet {

	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		String id = request.getParameter("id");
		ResourceService service = new ResourceService();
		try {
			ResourceBean rb = service.findById(id);
			File file = new File(rb.getSavepath() , rb.getUuidname());
			
			String filename = rb.getRealname();
			//中文名称乱码--资源无法查找
			filename = new String(filename.getBytes("iso8859-1"), "utf-8");
			
			if (file.exists()) {
				// Download
				//文件格式，根据不同格式进行响应
				String mimeType = this.getServletContext().getMimeType(
						rb.getRealname());
				response.setContentType(mimeType);
				//设置头信息，全部都响应为下载
				response.setHeader("content-disposition",
						"attachment;filename=" + filename);
				//中文名称在浏览器端下载时显示乱码
				filename = URLEncoder.encode(filename, "utf-8");
			
				//用io流进行读写，下载文件
				FileInputStream fis = new FileInputStream(file);
				OutputStream os = response.getOutputStream();
				int len = -1;
				byte[] b = new byte[1024];
				while ((len = fis.read(b)) != -1) {
					os.write(b, 0, len);
					os.flush();
				}
				os.close();
				fis.close();
			} else {
				throw new RuntimeException("resource is not exist.");
			}
		} catch (SQLException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		doGet(request, response);
	}

}