项目经常需要向服务器上传文件或者从服务器下载文件,因此将所需要的许多方法整理到了一个工具类中,包括从服务器下载流文件,上传文件到服务器(用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();
}
}
}