最近有個項目的功能需要上傳和下載文件。
廢話不多說直接看代碼(有註釋哦)
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();
}
}
這樣就是實現上傳下載的功能了,每天進步一點點。