javaWeb向服务器上传和下载文件

项目经常需要向服务器上传文件或者从服务器下载文件,因此将所需要的许多方法整理到了一个工具类中,包括从服务器下载流文件,上传文件到服务器(用Apache和spring不同的方法),希望能帮助到大家。

package com.phy.szms.utils;

import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadBase;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;



import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

public class UploadFileUtils {
	
	public static final String UPLOAD_SUCCESS = "upload success";
	
	public static final String UPLOAD_FAIL = "upload fail";
	
	public static final String UPLOAD_OVER_MAX_SIZE = "single file more than max size";
	
	public static BASE64Encoder encoder = new BASE64Encoder();
	public static BASE64Decoder decoder = new BASE64Decoder();
	
	public static String uploadByCommon(HttpServletRequest request) {
		//得到上传文件的保存目录,将上传的文件存放于WEB-INF目录下,不允许外界直接访问,保证上传文件的安全
		String savePath = request.getServletContext().getRealPath("WEB-INF/upload");
		//上传时生成的临时文件保存目录
		String tempPath = request.getServletContext().getRealPath("temp");
		File tempFile = new File(tempPath);
		if (!tempFile.exists()) {
			//创建临时目录
			tempFile.mkdir();
		}
		//消息提示
		String message = "";
		try {
			//使用Apache文件上传组件处理文件上传步骤
			//1、创建一个DiskFileItemFactory工厂
			DiskFileItemFactory factory = new DiskFileItemFactory();
			//设置工厂的缓冲区的大小,当上传的文件大小超过缓冲区的大小时,就会生成一个临时文件存放到指定的临时目录中
			factory.setSizeThreshold(1024*100);
			//设置上传时生成的临时文件的保存目录
			factory.setRepository(tempFile);
			//2、创建一个文件上传解析器
			ServletFileUpload upload = new ServletFileUpload(factory);
			//监听文件上传进度
			upload.setProgressListener(new ProgressListener() {
				@Override
				public void update(long pBytesRead, long pContentLength, int arg2) {
					System.out.println("文件大小为:" + pContentLength + ",当前已处理:" + pBytesRead);
				}
			});
			//解决上传文件名的中文乱码
			upload.setHeaderEncoding("UTF-8");
			//3、判断提交上来的数据是否是上传表单的数据
			if (!ServletFileUpload.isMultipartContent(request)) {
				//按照传统的方式获取数据
				return null;
			}
			//设置上传单个文件的大小的最大值,目前是设置为1024*1024字节,也就是1MB
			upload.setFileSizeMax(1024*1024);
			//设置上传文件总量的最大值,最大值=同时上传的多个文件的大小的最大值的和,目前设置为10MB
			upload.setSizeMax(1024*1024*10);
			//4、使用ServletFileUpload解析器解析上传数据,解析结果返回的事一个List<FileItem>集合,每一个FileItem对应一个Form表单的输入项
			List<FileItem> list = upload.parseRequest(request);
			for (FileItem item : list) {
				//如果fileItem中封装的是普通输入项的数据
				if (item.isFormField()) {
					String name = item.getFieldName();
					//解决普通输入项的数据的中文乱码问题
					String value = item.getString("UTF-8");
					//value = new String(value.getBytes("iso8859-1", "UTF-8");
					System.out.println(name + "=" + value);
				} else {
					//如果fileItem中封装的是上传文件
					//得到上传的文件名称
					String fileName = item.getName();
					System.out.println(fileName);
					if (fileName == null || fileName.trim().equals("")) {
						continue;
					}
					//注意:不同的浏览器提交的文件名是不一样的,有些浏览器提交上来的文件名是带有路径的,如:c:\a\b\1.txt,而有些只是单纯的文件名,如:1.txt
					//处理获取到的上传文件的文件名的路径部分,只保留文件名部分
					String fileExtName = fileName.substring(fileName.lastIndexOf("\\.") + 1);
					//如果需要限制文件上传的文件类型,那么可以通过文件的扩展名来判断上传的文件类型是否合法
					System.out.println("上传的文件的扩展名是:" + fileExtName);
					//获取item中的上传文件的输入流
					InputStream in = item.getInputStream();
					//得到文件保存的名称
					String saveFileName = makeFileName(fileName);
					//得到文件的保存目录
					String realSavePath = makePath(saveFileName, savePath);
					//创建一个文件输出流
					FileOutputStream out = new FileOutputStream(realSavePath + "\\" + saveFileName);
					//创建一个缓冲区
					byte[] buffer = new byte[1024];
					//判断输入流中的数据是否已经读完的标识
					int len = 0;
					//循环将输入流读入到缓冲区中,(len=in.read(buffer))>0就表示in里面还有数据
					while ((len = in.read(buffer)) > 0) {
						//使用FileOutputStream输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + fileName)当中
						out.write(buffer, 0, len);
					}
					//关闭输入流
					in.close();
					//关闭输出流
					out.close();
					//删除处理文件上传时生成的临时文件
					item.delete();
					message = UploadFileUtils.UPLOAD_SUCCESS;
				}
			}
		} catch (FileUploadBase.FileSizeLimitExceededException e) {
			e.printStackTrace();
			message = UploadFileUtils.UPLOAD_OVER_MAX_SIZE;
			return message;
		} catch (Exception e) {
			message = UploadFileUtils.UPLOAD_FAIL;
			e.printStackTrace(); 
			return message;
		}
		return message;
	}
	
	public static void download(HttpServletRequest request,HttpServletResponse response, String fileName) throws IOException{
		//得到下载的文件名
//		String fileName = request.getParameter("filename");
		//fileName = new String(fileName.getBytes("iso8859-1"),"UTF-8");
		//上传的文件都是保存在/WEB-INF/upload目录下的子目录中的
		String fileSaveRootPath = request.getServletContext().getRealPath("/WEB-INF/upload");
		//通过文件名找出文件所在目录
		String path = makePath(fileName,fileSaveRootPath);
		//得到要下载的文件
		File file = new File(path+"\\"+fileName);
		//如果文件不存在
		if(!file.exists()){
			request.setAttribute("message", "您下载的资源已被删除");
			return;
		}
		//处理文件名
		String realname = fileName.substring(fileName.indexOf("_")+1);
		
		//设置响应头,控制浏览器下载该文件
		response.setHeader("content-Disposition", "attachment;filename="+URLEncoder.encode(realname,"UTF-8"));
		//读取要下载的文件,保存到文件输入流
		FileInputStream in = new FileInputStream(path+"\\"+fileName);
		//创建输出流
		OutputStream out = response.getOutputStream();
		//创建缓冲区
		byte buffer[] = new byte[1024];
		int len = 0;
		//循环将输入流中的内容读取到缓冲区当中
		while((len = in.read(buffer))>0){
			//输出缓冲区的内容到浏览器,实现文件下载
			out.write(buffer,0,len);
		}
		//关闭文件输入流
		in.close();
		//关闭输出流
		out.close();
	}
	
	
	public static Map<String, Object> uploadBySpring(HttpServletRequest request, String[] fileExts) throws Exception {
		Map<String, Object> map = new HashMap<>();
		//得到上传文件的保存目录,将上传的文件存放于upload目录下
		String savePath = request.getServletContext().getRealPath("/WEB-INF/upload");
		//创建一个通用的多部分解析器
		CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(request.getSession().getServletContext());
		//判断request是否有文件上传,即多部分请求
		if (multipartResolver.isMultipart(request)) {
			//转换成多部分request
			MultipartHttpServletRequest multiRequest = (MultipartHttpServletRequest) request;
			Iterator<String> iter = multiRequest.getFileNames();
			while (iter.hasNext()) {
				//取得上传文件
				MultipartFile file = multiRequest.getFile(iter.next());
				if (file != null) {
					//取得当前上传文件的文件名称
					String fileName = file.getOriginalFilename();
					//获取文件后缀名
					String fileExtName = fileName.substring(fileName.lastIndexOf("."));
					if (null != fileExts) {
						for (int i = 0; i < fileExts.length; i++) {
							if (fileExts[i].equalsIgnoreCase(fileExtName)) {
								break;
							}
							if (i < fileExts.length -1 && !fileExts[i].equalsIgnoreCase(fileExtName)) {
								continue;
							} else if (!fileExts[i].equalsIgnoreCase(fileExtName)) {
								return null;
							}
						}
					}
					//如果名称不为”“,说明该文件存在,否则说明该文件不存在
					if (fileName.trim() != "") {
						//重命名上传后的文件名,得到文件保存的名称
						String saveFileName = makeFileName(file.getOriginalFilename());
						//得到文件的保存目录
						String realSavePath = makePath(saveFileName, savePath);
						File localFile = new File(realSavePath + "\\" + saveFileName);
						try {
							file.transferTo(localFile);
							map.put("originalFileName", fileName);
							map.put("realPath", realSavePath + "\\" + saveFileName);
						} catch (IllegalStateException | IOException e) {
							throw e;
						}
					}
				}
			}
		}
		return map;
	}
	
	/**
	 * 生成上传文件的文件名,文件名以:uuid + "-" + 文件的原始名称
	 * @param fileName
	 * @return
	 */
	private static String makeFileName(String fileName) {
		//为防止文件覆盖的现象发生,要为上传文件产生一个唯一的文件名
		return UUID.randomUUID().toString() + "_" + fileName;
	}
	
	/**
	 * 为防止一个目录下面出现太多文件,要使用hash算法打散存储
	 * @param fileName
	 * @param savePath
	 * @return
	 */
	private static String makePath(String fileName, String savePath) {
		//得到文件名的hashCode值,得到的就是fileName这个字符串对象在内存中的地址
		int hashCode = fileName.hashCode();
		int dir1 = hashCode & 0xf; //0-15
		int dir2 = (hashCode & 0xf0) >> 4; //0-15
		//构造新的保存目录
		String dir = savePath + "\\" + dir1 + "\\" + dir2;  //如:upload\2\3
		//File既可以代表文件也可以代表目录
		File file = new File(dir);
		//如果目录不存在
		if (!file.exists()) {
			//创建目录
			file.mkdirs();
		}
		return dir;
	}
	
	public static String uploadFile(HttpServletRequest request, String[] fileExts) throws Exception {
		//得到上传文件的保存目录,将上传的文件存放于images目录下
		String savePath = request.getServletContext().getRealPath("/images");
		String backPath = null;
		//创建一个通用的多部分解析器
		CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(request.getSession().getServletContext());
		//判断request是否有文件上传,即多部分请求
		if (multipartResolver.isMultipart(request)) {
			//转换成多部分request
			MultipartHttpServletRequest multiRequest = (MultipartHttpServletRequest) request;
			Iterator<String> iter = multiRequest.getFileNames();
			while (iter.hasNext()) {
				//取得上传文件
				MultipartFile file = multiRequest.getFile(iter.next());
				if (file != null) {
					//取得当前上传文件的文件名称
					String fileName = file.getOriginalFilename();
					//获取文件后缀名
					String fileExtName = fileName.substring(fileName.lastIndexOf("."));
					if (null != fileExts) {
						for (int i = 0; i < fileExts.length; i++) {
							if (fileExts[i].equalsIgnoreCase(fileExtName)) {
								break;
							}
							if (i < fileExts.length -1 && !fileExts[i].equalsIgnoreCase(fileExtName)) {
								continue;
							} else if (!fileExts[i].equalsIgnoreCase(fileExtName)) {
								return null;
							}
						}
					}
					//如果名称不为”“,说明该文件存在,否则说明该文件不存在
					if (fileName.trim() != "") {
						//重命名上传后的文件名,得到文件保存的名称
						String saveFileName = makeFileName(file.getOriginalFilename());
						//得到文件的保存目录
						String realSavePath = makePath(saveFileName, savePath);
						File localFile = new File(realSavePath + "\\" + saveFileName);
						try {
							file.transferTo(localFile);
							backPath = realSavePath + "\\" + saveFileName;
						} catch (IllegalStateException | IOException e) {
							throw e;
						}
					}
				}
			}
		}
		return backPath;
	}
	
	/**
	 * getImageBinary
	 * @描述: 获得图片二进制流
	 * @作者: liangzhg
	 * @创建时间: 2018-4-4下午02:27:43
	 * @param imagePath
	 * @return
	 */
	public static String getImageBinary(String imagePath) {
		
		File file = new File(imagePath);
		System.out.println(imagePath);
		BufferedImage bi;
		try {
			bi = ImageIO.read(file);
			ByteArrayOutputStream baos = new ByteArrayOutputStream();
			ImageIO.write(bi, "jpg", baos);//格式可不管。
			byte[] bytes = baos.toByteArray();
			return encoder.encodeBuffer(bytes).trim();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return null;	
	}
	
	/**
	 * base64StringToImage
	 * @描述:将二进制字符转化为图片
	 * @作者: liangzhg
	 * @创建时间: 2018-4-4下午04:13:49
	 * @param savePath
	 * @param base64String
	 */
	public static void base64StringToImage(String savePath,String base64String) {
		try {
			byte[] bytes = decoder.decodeBuffer(base64String);
			ByteArrayInputStream bais = new ByteArrayInputStream(bytes);
			BufferedImage bi = ImageIO.read(bais);
			File file = new File(savePath);
			ImageIO.write(bi, "jpg", file);
			
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
}

猜你喜欢

转载自blog.csdn.net/qq_39071530/article/details/80898566