前端:
<el-button
type="primary"
icon="el-icon-download"
@click="downloadTemplate(‘药品清单-模板.xlsx‘)">下载模板</el-button>
在main.js中:
import fileDownload from ‘js-file-download‘
Vue.prototype.downloadTemplate = function (templateName) {
const fileEntity = {
fileName: templateName,
filePath: "D:\\prism\\svn\\drug-question\\drugques-pc\\src\\template\\"+templateName,
downloadChannel: ‘DIR‘
}
medicineListApi.ptsFileDownload(fileEntity).then(response => {
fileDownload(response.data, templateName)
}).catch(error => {
console.log(error)
})
}
在package.json中添加依赖版本
"dependencies": { "@aximario/json-tree": "^2.1.0", "axios": "^0.19.0", "core-js": "^2.6.5", "echarts": "^4.2.1", "element-ui": "^2.13.2", "js-file-download": "^0.4.4", },
如果是下载本地的文件,则应写详细的路径:D:\prism\svn\drug-question\drugques-pc\src\template
在medicineList.js中
const baseUrl = "/medicineList"
ptsFileDownload(query) { return request({ url: baseUrl +‘/fileDownload‘, method: ‘post‘, data: query, responseType: ‘arraybuffer‘ }) },
后台代码:
controller:
@RequestMapping(value = "/fileDownload", method = { RequestMethod.POST, RequestMethod.GET })
public String fileDownload(@RequestBody(required=true) PtsFileEntity fileEntity, HttpServletResponse response) throws Exception {
medicineListService.fileDownload(fileEntity, response);
return null;
}
service接口
public interface MedicineListService extends IService<DrugData> {void fileDownload(PtsFileEntity fileEntity, HttpServletResponse response) throws Exception; }
service实现类
@Override
public void fileDownload(PtsFileEntity fileEntity, HttpServletResponse response) throws Exception {
String fileName = fileEntity.getFileName();
if (null == fileName || ("").equals(fileName.trim())) {
logger.error("No FileName Found.");
throw new Exception("No FileName Found.");
}
String downloadChannel = fileEntity.getDownloadChannel();
if (null == downloadChannel || ("").equals(downloadChannel.trim())) {
logger.error("Need to identity download channel.");
throw new Exception("Need to identity download channel.");
}
if (CHANNEL_DIR.equalsIgnoreCase(downloadChannel)) {
this.downloadFromDir(fileEntity, response);
} else if (CHANNEL_URL.equalsIgnoreCase(downloadChannel)) {
this.downloadFromUrl(fileEntity, response);
}
}
/**
* 从URL地址下载
*
* @param fileEntity
* @param response
* @throws Exception
*/
private void downloadFromUrl(PtsFileEntity fileEntity, HttpServletResponse response) throws Exception {
String filePath = fileEntity.getFilePath();
String serverFilePath = fileEntity.getServerFilePath();
if ((null == filePath || ("").equals(filePath.trim())) && (null == serverFilePath || ("").equals(serverFilePath.trim()))) {
logger.error("No FilePath Found.");
throw new Exception("No FilePath Found.");
}
String realFilePath = (null == filePath || ("").equals(filePath.trim())) ? serverFilePath : filePath;
logger.info("Begin download file from Url: " + realFilePath);
try {
URL url = new URL(realFilePath);
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
//设置超时间为3秒
conn.setConnectTimeout(3 * 1000);
//防止屏蔽程序抓取而返回403错误
conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
//得到输入流
InputStream inputStream = conn.getInputStream();
//获取自己数组
byte[] srcBytes = readInputStream(inputStream);
this.download(fileEntity, response);
} catch (IOException e) {
e.printStackTrace();
throw new Exception(e);
}
}
/**
* 从输入流中获取字节数组
*
* @param inputStream
* @return
* @throws IOException
*/
public static byte[] readInputStream(InputStream inputStream) throws IOException {
byte[] buffer = new byte[1024];
int len = 0;
ByteArrayOutputStream bos = new ByteArrayOutputStream();
while ((len = inputStream.read(buffer)) != -1) {
bos.write(buffer, 0, len);
}
bos.close();
return bos.toByteArray();
}
/**
* 创建临时文件
*
* @param fileEntity
* @param srcBytes
*/
public void downloadFromDir(PtsFileEntity fileEntity, HttpServletResponse response) throws Exception {
String filePath = fileEntity.getFilePath();
String serverFilePath = fileEntity.getServerFilePath();
if ((null == filePath || ("").equals(filePath.trim())) && (null == serverFilePath || ("").equals(serverFilePath.trim()))) {
logger.error("No FilePath Found.");
throw new Exception("No FilePath Found.");
}
String realFilePath = (null == filePath || ("").equals(filePath.trim())) ? serverFilePath : filePath;
logger.info("Begin download file from Directory: " + realFilePath);
fileEntity.setRealFilePath(realFilePath);
try {
// 以流的形式下载文件。
InputStream fis;
fis = new BufferedInputStream(new FileInputStream(realFilePath));
byte[] srcBytes = new byte[fis.available()];
fis.read(srcBytes);
fis.close();
this.download(fileEntity, response);
} catch (IOException ex) {
ex.printStackTrace();
throw new Exception(ex);
}
}
/**
* 拼接response header 并已流的形式下载文件
*
* @param fileEntity
* @param response
* @throws Exception
*/
public void download(PtsFileEntity fileEntity, HttpServletResponse response) throws Exception {
String fileName = fileEntity.getFileName();
String realFilePath = fileEntity.getRealFilePath();
File file = new File(realFilePath);
try {
// 以流的形式下载文件。
InputStream fis;
fis = new BufferedInputStream(new FileInputStream(realFilePath));
byte[] srcBytes = new byte[fis.available()];
fis.read(srcBytes);
fis.close();
// 清空response
response.reset();
// 设置response的Header
response.setHeader("Content-type", "text/html;charset=UTF-8");
response.setCharacterEncoding("utf-8");//设置编码集,文件名不会发生中文乱码
response.setContentType("application/force-download");//
response.setHeader("content-type", "application/octet-stream");
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(fileName.getBytes(), "utf-8"));// 设置文件名
response.addHeader("Content-Length", "" + file.length());
response.setHeader("Access-Control-Allow-Origin", "*");
OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
toClient.write(srcBytes);
toClient.flush();
toClient.close();
} catch (IOException ex) {
throw new Exception(ex);
}
}
如果报错:The value of the ‘Access-Control-Allow-Origin‘ header in the response must not be the wildcard ‘*‘ when the request‘s credentials mode is ‘include‘
解决:将request.js中的withCredentials由true改为false
const service = axios.create({ baseURL: "http://localhost:8080/drugques", withCredentials: false, timeout: 150000 });