本文为大家分享了Spring4的下载组件，供大家参考，具体内容如下
package com.hnust.common.controller; import org.apache.commons.io.FileUtils; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpStatus; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.RestController; import java.io.File; import java.io.IOException; import java.io.UnsupportedEncodingException; import java.net.URLEncoder; /** * Created by Heweipo on 2016/5/27. * <p> * 下载通用控制器 */ @RestController public class DownloadController extends BaseController { /** * 下载文件通用方法 * * @param file 文件对象 * @return 文件字节流 */ public ResponseEntity<byte[]> export(File file) { return export(file.getName(), file); } /** * 下载文件通用方法 * * @param fileName 文件名称 * @param file 文件对象 * @return 文件字节流 */ public ResponseEntity<byte[]> export(String fileName, File file) { HttpHeaders headers = new HttpHeaders(); headers.setContentType(MediaType.APPLICATION_OCTET_STREAM); headers.setContentDispositionFormData("attachment", encodeFileName(fileName)); ResponseEntity<byte[]> rs = null; try { // 这里不能使用 HttpStatus.CREATED 201 是因为 IE Edge 无法识别，但是Firefox chrome 没问题 rs = new ResponseEntity<>(FileUtils.readFileToByteArray(file), headers, HttpStatus.OK); } catch (IOException e) { //throw new CommonException(ResponseStatusEnum.FILE_ERROR, e); } return rs; } /** * 下载文件的名称，这个是在浏览器那里显示的名称 * * @param fileName 文件名称 * @return 加码的文件名称 * <p> * IE * Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; rv:11.0) like Gecko * <p> * Edge * Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/46.0.2486.0 Safari/537.36 Edge/13.10586 * <p> * Firefox * Mozilla/5.0 (Windows NT 10.0; WOW64; rv:46.0) Gecko/20100101 Firefox/46.0 * <p> * Chrome * Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/47.0.2526.106 Safari/537.36 */ private String encodeFileName(String fileName) { String name = fileName; try { String agent = request.getHeader("USER-AGENT").toLowerCase(); if (null != agent && (agent.contains("msie") || agent.contains("edge"))) { // IE edge name = URLEncoder.encode(fileName, "UTF-8"); } else if (agent.contains("safari") || agent.contains("chrome") || agent.contains("firefox")) { // safari chrome firefox name = new String(fileName.getBytes("UTF-8"), "iso-8859-1"); } else { // IE10 IE11 name = URLEncoder.encode(fileName, "UTF-8"); } // 把加号还原为空格(IE Edge 有问题) name = name.replace("+", "%20"); } catch (UnsupportedEncodingException e) { //throw new CommonException(ResponseStatusEnum.FAILURE, e); } return name; } }
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持查字典教程网。
【分享Spring的下载组件】相关文章：