最近有个项目的功能需要上传和下载文件。
废话不多说直接看代码(有注释哦)
Controller层 直接调取Service层接口实现上传下载功能
@Controller
@RequestMapping("/check")
public class CheckFileController {
private final static Logger logger = LoggerFactory.getLogger(CheckFileController.class);
@Autowired
private CheckFileService checkFileService;
/**
* 上传正向源文件
*
* @param request
* @return
*/
@RequestMapping("/uploadFile.do")
@ResponseBody
public String uploadFile(HttpServletRequest request) {
JSONObject jsonObject = new JSONObject();
jsonObject.put("result", "true");
try {
jsonObject = checkFileService.uploadCheckFile(request);
} catch (Exception e) {
jsonObject.put("result", "false");
jsonObject.put("message", e.getMessage());
}
return jsonObject.toString();
}
/**
* 查看下载
* @param request
* @param response
*/
@RequestMapping("/downLoadFile.do")
public void downLoadFile(HttpServletRequest request, HttpServletResponse response) {
try {
checkFileService.downLoadFile(request,response);
}catch (Exception e){
logger.error(e.getMessage());
}
}
}
Service层 Service接口和实现类
@Service("CheckFileService")
public class CheckFileServiceImpl implements CheckFileService {
private final static Logger logger = LoggerFactory.getLogger(CheckFileServiceImpl.class);
//上传文件的目录地址 在配置文件中配置
//这个目录地址在服务器上时 最好选择在apache-tomcat你的项目下 这样不会出现目录权限的问题
private static String uploadFilePath = PropertyReader.getContextProperty("uploadFilePath");
//下载文件的目录地址 在配置文件中配置
//这个目录地址在服务器上时 最好选择在apache-tomcat你的项目下 这样不会出现目录权限的问题
private static String downLoadFilePath = PropertyReader.getContextProperty("downLoadFilePath");
@Resource
DateFileDao dateFileDao;
@Override
public JSONObject uploadCheckFile(HttpServletRequest request) throws Exception {
JSONObject jsonObject = new JSONObject();
String date = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
String fileDate = date.split("-")[0] + date.split("-")[1];
//得到上传文件的保存目录,将上传的文件存放于WEB-INF目录下,不允许外界直接访问,保证上传文件的安全
//上传时生成的临时文件保存目录
String tempPath = uploadFilePath + fileDate + "/1002/" + "tmp/";
File file = new File(tempPath);
if (!file.exists() && !file.isDirectory()) {
System.out.println("目录或文件不存在!");
file.mkdir();
}
//消息提示
String message = "";
try {
//使用Apache文件上传组件处理文件上传步骤:
//1、创建一个DiskFileItemFactory工厂
DiskFileItemFactory diskFileItemFactory = new DiskFileItemFactory();
//设置工厂的缓冲区的大小,当上传的文件大小超过缓冲区的大小时,就会生成一个临时文件存放到指定的临时目录当中。
diskFileItemFactory.setSizeThreshold(1024 * 100);
//设置上传时生成的临时文件的保存目录
diskFileItemFactory.setRepository(file);
//2、创建一个文件上传解析器
ServletFileUpload fileUpload = new ServletFileUpload(diskFileItemFactory);
//解决上传文件名的中文乱码
fileUpload.setHeaderEncoding("UTF-8");
//监听文件上传进度
fileUpload.setProgressListener(new ProgressListener() {
public void update(long pBytesRead, long pContentLength, int arg2) {
System.out.println("文件大小为:" + pContentLength + ",当前已处理:" + pBytesRead);
}
});
//3、判断提交上来的数据是否是上传表单的数据
if (!fileUpload.isMultipartContent(request)) {
//按照传统方式获取数据
throw new Exception("不是表单提交的数据");
}
//设置上传单个文件的大小的最大值,目前是设置为1024*1024字节,也就是1MB
fileUpload.setFileSizeMax(1024 * 1024 * 10);
//设置上传文件总量的最大值,最大值=同时上传的多个文件的大小的最大值的和,目前设置为10MB
fileUpload.setSizeMax(1024 * 1024 * 50);
//4、使用ServletFileUpload解析器解析上传数据,解析结果返回的是一个List<FileItem>集合,
//每一个FileItem对应一个Form表单的输入项
List<FileItem> list = fileUpload.parseRequest(request);
for (FileItem item : list) {
//如果fileitem中封装的是普通输入项的数据
if (item.isFormField()) {
String name = item.getFieldName();
//解决普通输入项的数据的中文乱码问题
String value = item.getString("UTF-8");
String value1 = new String(name.getBytes("iso8859-1"), "UTF-8");
System.out.println(name + " " + value);
System.out.println(name + " " + value1);
} else {
//如果fileitem中封装的是上传文件,得到上传的文件名称,
String fileName = item.getName();
System.out.println(fileName);
if (fileName == null || fileName.trim().equals("")) {
continue;
}
//注意:不同的浏览器提交的文件名是不一样的,有些浏览器提交上来的文件名是带有路径的
//如: c:\a\b\1.txt,而有些只是单纯的文件名,如:1.txt
//处理获取到的上传文件的文件名的路径部分,只保留文件名部分
fileName = fileName.substring(fileName.lastIndexOf(File.separator) + 1);
//得到上传文件的扩展名
String fileExtName = fileName.substring(fileName.lastIndexOf(".") + 1);
if ("zip".equals(fileExtName) || "rar".equals(fileExtName) || "tar".equals(fileExtName) || "jar".equals(fileExtName)) {
throw new Exception("上传文件的类型不符合!!!");
}
//如果需要限制上传的文件类型,那么可以通过文件的扩展名来判断上传的文件类型是否合法
System.out.println("上传文件的扩展名为:"+fileExtName);
InputStream is = item.getInputStream();
//得到文件保存的名称
//得到文件保存的路径
System.out.println("保存路径为:"+uploadFilePath);
//创建一个文件输出流
FileOutputStream fos = new FileOutputStream(uploadFilePath + fileDate + "/1002/" + fileName);
//创建一个缓冲区
byte buffer[] = new byte[1024];
//判断输入流中的数据是否已经读完的标识
int length = 0;
//循环将输入流读入到缓冲区当中,(len=in.read(buffer))>0就表示in里面还有数据
while ((length = is.read(buffer)) > 0) {
//使用FileOutputStream输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + filename)当中
fos.write(buffer, 0, length);
}
//关闭输入流
is.close();
//关闭输出流
fos.close();
//删除处理文件上传时生成的临时文件
item.delete();
message = "文件上传成功";
}
}
} catch (FileUploadBase.FileSizeLimitExceededException e) {
jsonObject.put("result", "false");
jsonObject.put("message", e.getMessage());
logger.error(e.toString());
throw new Exception("单个文件超出最大值");
} catch (FileUploadBase.SizeLimitExceededException e) {
jsonObject.put("result", "false");
jsonObject.put("message", e.getMessage());
logger.error(e.toString());
throw new Exception("上传文件的总的大小超出限制的最大值!!!");
} catch (FileUploadException e) {
jsonObject.put("result", "false");
jsonObject.put("message", e.getMessage());
logger.error(e.toString());
throw new Exception("文件上传失败");
}
jsonObject.put("result", "true");
return jsonObject;
}
@Override
public void downLoadFile(HttpServletRequest request, HttpServletResponse response) throws Exception {
//得到要下载的文件名
String fileName = request.getParameter("name");
fileName = new String(fileName.getBytes("iso8859-1"),"UTF-8");
String date = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
String fileDate = date.split("-")[0] + date.split("-")[1];
String fileCode = "1002";
//上传的文件都是保存在/WEB-INF/upload目录下的子目录当中
String fileSaveRootPath=downLoadFilePath + fileDate + "/" + fileCode;
//处理文件名
String realName = fileName.substring(fileName.indexOf("_")+1);
//通过文件名找出文件的所在目录
//得到要下载的文件
File file = new File(fileSaveRootPath + "/" + fileName);
//如果文件不存在
if(!file.exists()){
request.setAttribute("message", "您要下载的资源已被删除!!");
request.getRequestDispatcher("/jsp/login/message.jsp").forward(request, response);
return;
}
//设置响应头,控制浏览器下载该文件
response.setHeader("content-disposition", "attachment;filename=" + URLEncoder.encode(realName, "UTF-8"));
//读取要下载的文件,保存到文件输入流
FileInputStream in = new FileInputStream(fileSaveRootPath + "/" + fileName);
//创建输出流
OutputStream os = response.getOutputStream();
//设置缓存区
byte[] bytes = new byte[1024];
int len = 0;
while(-1 != (len = in.read(bytes))){
os.write(bytes,0,len);
}
os.flush();
//关闭输出流
os.close();
//关闭输入流
in.close();
}
}
这样就是实现上传下载的功能了,每天进步一点点。