快捷導(dǎo)航

Spring MVC的文件下載實(shí)例詳解

更新時(shí)間：2017年05月03日 14:29:25 投稿：lqh

這篇文章主要介紹了Spring MVC的文件下載實(shí)例詳解的相關(guān)資料,需要的朋友可以參考下

Spring MVC的文件下載實(shí)例詳解

讀取文件

要下載文件，首先是將文件內(nèi)容讀取進(jìn)來，使用字節(jié)數(shù)組存儲起來，這里使用spring里面的工具類實(shí)現(xiàn)

import org.springframework.util.FileCopyUtils;

  public byte[] downloadFile(String fileName) {
    byte[] res = new byte[0];
    try {
      File file = new File(BACKUP_FILE_PATH, fileName);
      if (file.exists() && !file.isDirectory()) {
        res = FileCopyUtils.copyToByteArray(file);
      }
    } catch (IOException e) {
      logger.error(e.getMessage());
    }
    return res;
  }

這個(gè)數(shù)組就是文件的內(nèi)容，后面將輸出到響應(yīng)，供瀏覽器下載

下載文件的響應(yīng)

下載文件的響應(yīng)頭和一般的響應(yīng)頭是有所區(qū)別的，而這里面還要根據(jù)用戶瀏覽器的不同區(qū)別對待

我把生成響應(yīng)的代碼封裝成了一個(gè)方法，這樣所有下載響應(yīng)都可以調(diào)用這個(gè)方法了，避免重復(fù)代碼到處寫

 protected ResponseEntity<byte[]> downloadResponse(byte[] body, String fileName) {
    HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder
        .getRequestAttributes()).getRequest();
    String header = request.getHeader("User-Agent").toUpperCase();
    HttpStatus status = HttpStatus.CREATED;
    try {
      if (header.contains("MSIE") || header.contains("TRIDENT") || header.contains("EDGE")) {
        fileName = URLEncoder.encode(fileName, "UTF-8");
        fileName = fileName.replace("+", "%20");  // IE下載文件名空格變+號問題
        status = HttpStatus.OK;
      } else {
        fileName = new String(fileName.getBytes("UTF-8"), "ISO8859-1");
      }
    } catch (UnsupportedEncodingException e) {}

    HttpHeaders headers = new HttpHeaders();
    headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
    headers.setContentDispositionFormData("attachment", fileName);
    headers.setContentLength(body.length);

    return new ResponseEntity<byte[]>(body, headers, status);
  }

這里需要注意，一般來說下載文件是使用201狀態(tài)碼的，但是IE瀏覽器不支持，還得我花了很大力氣才找出來是那個(gè)問題

其中對文件名的處理是為了防止中文以及空格導(dǎo)致文件名亂碼

控制器方法

在控制器的那里需要對返回值進(jìn)行處理

@RequestMapping(value = "/download-backup", method = RequestMethod.GET)
  @ResponseBody
  public ResponseEntity<byte[]> downloadBackupFile(@RequestParam String fileName) {
    byte[] body = backupService.downloadFile(fileName);
    return downloadResponse(body, fileName);
  }

感謝閱讀，希望能幫助到大家，謝謝大家對本站的支持！

您可能感興趣的文章: