在以前,處理文件上傳是一個很痛苦的事情,大都借助于開源的上傳組件,諸如commons fileupload等。現在好了,很方便,便捷到比那些組件都方便至極。以前的HTML端上傳表單不用改變什么,還是一樣的multipart/form-data MIME類型。
讓Servlet支持上傳,需要做兩件事情
- 需要添加MultipartConfig注解
- 從request對象中獲取Part文件對象
但在具體實踐中,還是有一些細節處理,諸如設置上傳文件的最大值,上傳文件的保存路徑。
需要熟悉MultipartConfig注解,標注在@WebServlet之上,具有以下屬性:
屬性名 | 類型 | 是否可選 | 描述 |
---|
fileSizeThreshold | int | 是 | 當數據量大于該值時,內容將被寫入文件。 |
location | String | 是 | 存放生成的文件地址。 |
maxFileSize | long | 是 | 允許上傳的文件最大值。默認值為 -1,表示沒有限制。 |
maxRequestSize | long | 是 | 針對該 multipart/form-data 請求的最大數量,默認值為 -1,表示沒有限制。 |
一些實踐建議:
- 若是上傳一個文件,僅僅需要設置maxFileSize熟悉即可。
- 上傳多個文件,可能需要設置maxRequestSize屬性,設定一次上傳數據的最大量。
- 上傳過程中無論是單個文件超過maxFileSize值,或者上傳總的數據量大于maxRequestSize值都會拋出IllegalStateException異常;
- location屬性,既是保存路徑(在寫入的時候,可以忽略路徑設定),又是上傳過程中臨時文件的保存路徑,一旦執行Part.write方法之后,臨時文件將被自動清除。
- 但Servlet 3.0規范同時也說明,不提供獲取上傳文件名的方法,盡管我們可以通過part.getHeader("content-disposition")方法間接獲取得到。
- 如何讀取MultipartConfig注解屬性值,API沒有提供直接讀取的方法,只能手動獲取。
來一個示例吧,上傳前臺頁面:
后臺處理Servlet:
/**
* 上傳文件測試 location為臨時文件保存路徑
*
* @author yongboy
* @date 2011-1-13
* @version 1.0
*/
@MultipartConfig(location = "/home/yongboy/tmp/", maxFileSize = 1024 * 1024 * 10)
@WebServlet("/upload")
public class UploadFileAction extends HttpServlet {
private static final long serialVersionUID = 92166165626L;
private static final Log log = LogFactory.getLog(UploadFileAction.class);
// 得到注解信息
private static final MultipartConfig config;
static {
config = UploadFileAction.class.getAnnotation(MultipartConfig.class);
}
protected void doGet(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException {
request.getRequestDispatcher("/upload.jsp").forward(request, response);
}
protected void doPost(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException {
// 為避免獲取文件名稱時出現亂碼
request.setCharacterEncoding("UTF-8");
Part part = null;
try {
// <input name="file" size="50" type="file" />
part = request.getPart("file");
} catch (IllegalStateException ise) {
// 上傳文件超過注解所標注的maxRequestSize或maxFileSize值
if (config.maxRequestSize() == -1L) {
log.info("the Part in the request is larger than maxFileSize");
} else if (config.maxFileSize() == -1L) {
log.info("the request body is larger than maxRequestSize");
} else {
log.info("the request body is larger than maxRequestSize, or any Part in the request is larger than maxFileSize");
}
forwardErrorPage(request, response, "上傳文件過大,請檢查輸入是否有誤!");
return;
} catch (IOException ieo) {
// 在接收數據時出現問題
log.error("I/O error occurred during the retrieval of the requested Part");
} catch (Exception e) {
log.error(e.toString());
e.printStackTrace();
}
if (part == null) {
forwardErrorPage(request, response, "上傳文件出現異常,請檢查輸入是否有誤!");
return;
}
// 得到文件的原始名稱,eg :測試文檔.pdf
String fileName = UploadUtils.getFileName(part);
log.info("contentType : " + part.getContentType());
log.info("fileName : " + fileName);
log.info("fileSize : " + part.getSize());
log.info("header names : ");
for (String headerName : part.getHeaderNames()) {
log.info(headerName + " : " + part.getHeader(headerName));
}
String saveName = System.currentTimeMillis() + "."
+ FilenameUtils.getExtension(fileName);
log.info("save the file with new name : " + saveName);
// 因在注解中指定了路徑,這里可以指定要寫入的文件名
// 在未執行write方法之前,將會在注解指定location路徑下生成一臨時文件
part.write(saveName);
request.setAttribute("fileName", fileName);
request.getRequestDispatcher("/uploadResult.jsp").forward(request,
response);
}
private void forwardErrorPage(HttpServletRequest request,
HttpServletResponse response, String errMsg)
throws ServletException, IOException {
request.setAttribute("errMsg", errMsg);
request.getRequestDispatcher("/upload.jsp").forward(request, response);
}
}
獲取文件名的函數,很簡單:
/**
* 如何得到上傳的文件名, API沒有提供直接的方法,只能從content-disposition屬性中獲取
*
* @param part
* @return
*/
protected static String getFileName(Part part) {
if (part == null)
return null;
String fileName = part.getHeader("content-disposition");
if (StringUtils.isBlank(fileName)) {
return null;
}
return StringUtils.substringBetween(fileName, "filename=\"", "\"");
}
文件上傳成功之后,以及日志輸出的截圖如下:
截圖中可以看到Part包含content-disposition屬性,可以很容易從值中抽取出文件名。臨時生成的上傳文件大都以 .tmp為后綴,大致如下:
讓上傳出現錯誤,就可以在保存路徑下看到大致如上的臨時文件。
一次上傳多個文件的后臺servlet示范:
/**
* 多文件上傳支持
* @author yongboy
* @date 2011-1-14
* @version 1.0
*/
@MultipartConfig(
location = "/home/yongboy/tmp/",
maxFileSize = 1024L * 1024L, // 每一個文件的最大值
maxRequestSize = 1024L * 1024L * 10L // 一次上傳最大值,若每次只能上傳一個文件,則設置maxRequestSize意義不大
)
@WebServlet("/uploadFiles")
public class UploadFilesAction extends HttpServlet {
private static final long serialVersionUID = 2304820820384L;
private static final Log log = LogFactory.getLog(UploadFilesAction.class);
protected void doGet(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException {
request.getRequestDispatcher("/uploads.jsp").forward(request, response);
}
protected void doPost(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException {
request.setCharacterEncoding("UTF-8");
Collection parts = null;
try {
parts = request.getParts();
} catch (IllegalStateException ise) {
// 可能某個文件大于指定文件容量maxFileSize,或者提交數據大于maxRequestSize
log.info("maybe the request body is larger than maxRequestSize, or any Part in the request is larger than maxFileSize");
} catch (IOException ioe) {
// 在獲取某個文件時遇到拉IO異常錯誤
log.error("an I/O error occurred during the retrieval of the Part components of this request");
} catch (Exception e) {
log.error("the request body is larger than maxRequestSize, or any Part in the request is larger than maxFileSize");
e.printStackTrace();
}
if (parts == null || parts.isEmpty()) {
doError(request, response, "上傳文件為空!");
return;
}
// 前端具有幾個file組件,這里會對應幾個Part對象
List fileNames = new ArrayList();
for (Part part : parts) {
if (part == null) {
continue;
}
// 這里直接以源文件名保存
String fileName = UploadUtils.getFileName(part);
if (StringUtils.isBlank(fileName)) {
continue;
}
part.write(fileName);
fileNames.add(fileName);
}
request.setAttribute("fileNames", fileNames);
request.getRequestDispatcher("/uploadsResult.jsp").forward(request,
response);
}
private void doError(HttpServletRequest request,
HttpServletResponse response, String errMsg)
throws ServletException, IOException {
request.setAttribute("errMsg", errMsg);
this.doGet(request, response);
}
}
批量上傳很簡單,但也有風險,任一個文件若超過maxFileSize值,意味著整個上傳都會失敗。若不限大小,那就不存在以上憂慮了。
總之,在Servlet 3.0 中,無論是上傳一個文件,或者多個批量上傳都是非常簡單,但要處理好異常,避免出錯。