java/ruoyi-vue-pro
1
0
Fork 0
mirror of https://github.com/YunaiV/ruoyi-vue-pro.git synced 2025-10-31 10:37:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

根据艿艿的提示修改

Browse Source
This commit is contained in:
timfruit
2021-03-13 22:40:16 +08:00
parent 7089ef7651
commit 562f4cb953
1 changed files with 10 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/main/java/cn/iocoder/dashboard/modules/infra/controller/doc/InfDbDocController.java
View File

@ -48,9 +48,7 @@ public class InfDbDocController {
@GetMapping("/export-html") @GetMapping("/export-html")
@ApiOperation("导出html格式的数据文档") @ApiOperation("导出html格式的数据文档")
@ApiImplicitParams({ @ApiImplicitParam(name = "deleteFile", value = "是否删除在服务器本地生成的数据库文档", example = "true", dataTypeClass = Boolean.class)
@ApiImplicitParam(name = "deleteFile", value = "是否删除在服务器本地生成的数据库文档", example = "true", dataTypeClass = Boolean.class),
})
public void exportHtml(@RequestParam(defaultValue = "true") Boolean deleteFile, public void exportHtml(@RequestParam(defaultValue = "true") Boolean deleteFile,
HttpServletResponse response) throws IOException { HttpServletResponse response) throws IOException {
doExportFile(EngineFileType.HTML, deleteFile, response); doExportFile(EngineFileType.HTML, deleteFile, response);
@ -58,9 +56,7 @@ public class InfDbDocController {
@GetMapping("/export-word") @GetMapping("/export-word")
@ApiOperation("导出word格式的数据文档") @ApiOperation("导出word格式的数据文档")
@ApiImplicitParams({ @ApiImplicitParam(name = "deleteFile", value = "是否删除在服务器本地生成的数据库文档", example = "true", dataTypeClass = Boolean.class)
@ApiImplicitParam(name = "deleteFile", value = "是否删除在服务器本地生成的数据库文档", example = "true", dataTypeClass = Boolean.class),
})
public void exportWord(@RequestParam(defaultValue = "true") Boolean deleteFile, public void exportWord(@RequestParam(defaultValue = "true") Boolean deleteFile,
HttpServletResponse response) throws IOException { HttpServletResponse response) throws IOException {
doExportFile(EngineFileType.WORD, deleteFile, response); doExportFile(EngineFileType.WORD, deleteFile, response);
@ -68,9 +64,7 @@ public class InfDbDocController {
@GetMapping("/export-markdown") @GetMapping("/export-markdown")
@ApiOperation("导出markdown格式的数据文档") @ApiOperation("导出markdown格式的数据文档")
@ApiImplicitParams({ @ApiImplicitParam(name = "deleteFile", value = "是否删除在服务器本地生成的数据库文档", example = "true", dataTypeClass = Boolean.class)
@ApiImplicitParam(name = "deleteFile", value = "是否删除在服务器本地生成的数据库文档", example = "true", dataTypeClass = Boolean.class),
})
public void exportMarkdown(@RequestParam(defaultValue = "true") Boolean deleteFile, public void exportMarkdown(@RequestParam(defaultValue = "true") Boolean deleteFile,
HttpServletResponse response) throws IOException { HttpServletResponse response) throws IOException {
doExportFile(EngineFileType.MD, deleteFile, response); doExportFile(EngineFileType.MD, deleteFile, response);
@ -81,12 +75,13 @@ public class InfDbDocController {
String docFileName = DOC_FILE_NAME + "_" + IdUtil.fastSimpleUUID(); String docFileName = DOC_FILE_NAME + "_" + IdUtil.fastSimpleUUID();
String filePath = doExportFile(fileOutputType, docFileName); String filePath = doExportFile(fileOutputType, docFileName);
String downloadFileName = DOC_FILE_NAME + fileOutputType.getFileSuffix(); //下载后的文件名 String downloadFileName = DOC_FILE_NAME + fileOutputType.getFileSuffix(); //下载后的文件名
// 读取，返回 try {
//IoUtil.readBytes 直接读取FileInputStream 不会关闭流，有bug,所以用BufferedInputStream包装一下, 关闭流后才能删除文件 // 读取，返回
byte[] content = IoUtil.readBytes(new BufferedInputStream(new FileInputStream(filePath))); //这里不用hutool工具类，它的中文文件名编码有问题,导致在浏览器下载时有问题
//这里不用hutool工具类，它的中文文件名编码有问题,导致在浏览器下载时有问题 ServletUtils.writeAttachment(response, downloadFileName, FileUtil.readBytes(filePath));
ServletUtils.writeAttachment(response, downloadFileName, content); }finally {
handleDeleteFile(deleteFile, filePath); handleDeleteFile(deleteFile, filePath);
}
} }
/** /**