feat: 预览接口支持token校验用户是否有企标权限
This commit is contained in:
-81
@@ -145,19 +145,6 @@ public class CommonController {
|
|||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* 下载文件
|
|
||||||
*
|
|
||||||
* @param id 传入文件id
|
|
||||||
* @param request
|
|
||||||
* @param response
|
|
||||||
*/
|
|
||||||
@GetMapping(value = "/download/{id}")
|
|
||||||
public void download(@PathVariable String id, HttpServletRequest request, HttpServletResponse response) {
|
|
||||||
downloadAndViewWithWaterMark(id, response);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
private void downloadAndView(@PathVariable String id, HttpServletResponse response) {
|
private void downloadAndView(@PathVariable String id, HttpServletResponse response) {
|
||||||
// 查询数据表数据是否存在
|
// 查询数据表数据是否存在
|
||||||
@@ -235,74 +222,6 @@ public class CommonController {
|
|||||||
downloadAndView(id, response);
|
downloadAndView(id, response);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void downloadAndViewWithWaterMark(@PathVariable String id, HttpServletResponse response) {
|
|
||||||
// 查询数据表数据是否存在
|
|
||||||
LambdaQueryWrapper<OSSFile> queryWrapper = new LambdaQueryWrapper<>();
|
|
||||||
queryWrapper.eq(OSSFile::getId, id);
|
|
||||||
OSSFile ossFile = ossFileService.getOne(queryWrapper);
|
|
||||||
if (null == ossFile) {
|
|
||||||
throw new JeroBootException("文件不存在..");
|
|
||||||
}
|
|
||||||
// 如果是企标的文件则需要判断当前登录人是否有查看权限,如果没有则需要返回没有权限的文件
|
|
||||||
if (StrUtil.isBlank(ossFile.getStandardId())) {
|
|
||||||
String userId = UserUtils.getUserId();
|
|
||||||
|
|
||||||
}
|
|
||||||
String fileUrl = ossFile.getUrl();
|
|
||||||
// minio 下载
|
|
||||||
// 通过MinioUtil查询时 只需要桶后面的路径
|
|
||||||
String minioUrl = MinioUtil.getMinioUrl();
|
|
||||||
// Linux/unix 系统下文件路径分隔符为"/" 获取minio与存储桶的路径
|
|
||||||
minioUrl = minioUrl + MinioUtil.getBucketName() + "/";
|
|
||||||
String url = fileUrl.replace(minioUrl, "");
|
|
||||||
// 文件名称
|
|
||||||
String fileName = ossFile.getFileName();
|
|
||||||
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
|
|
||||||
// 设置强制下载不打开
|
|
||||||
response.setContentType("application/force-download");
|
|
||||||
// 然后在您的controller方法中:
|
|
||||||
if ("pdf".equalsIgnoreCase(FileUtil.extName(fileName))) {
|
|
||||||
try (InputStream originalInputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url);
|
|
||||||
InputStream watermarkedInputStream = sysWaterMarkUtil.addWatermarkToPdf(originalInputStream);
|
|
||||||
OutputStream outputStream = response.getOutputStream()) {
|
|
||||||
|
|
||||||
if (originalInputStream == null) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
byte[] buf = new byte[1024];
|
|
||||||
int len;
|
|
||||||
while ((len = watermarkedInputStream.read(buf)) > 0) {
|
|
||||||
outputStream.write(buf, 0, len);
|
|
||||||
}
|
|
||||||
response.flushBuffer();
|
|
||||||
|
|
||||||
} catch (Exception e) {
|
|
||||||
log.error(e.getMessage());
|
|
||||||
response.setStatus(404);
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
try (InputStream inputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url);
|
|
||||||
OutputStream outputStream = response.getOutputStream()
|
|
||||||
) {
|
|
||||||
byte[] buf = new byte[1024];
|
|
||||||
int len;
|
|
||||||
if (inputStream == null) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
while ((len = inputStream.read(buf)) > 0) {
|
|
||||||
outputStream.write(buf, 0, len);
|
|
||||||
}
|
|
||||||
response.flushBuffer();
|
|
||||||
} catch (Exception e) {
|
|
||||||
log.error(e.getMessage());
|
|
||||||
response.setStatus(404);
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void setResponse(HttpServletResponse response, String fileName) {
|
private void setResponse(HttpServletResponse response, String fileName) {
|
||||||
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
|
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
|
||||||
// 设置强制下载不打开
|
// 设置强制下载不打开
|
||||||
|
|||||||
@@ -47,6 +47,19 @@ public class SysCommonController {
|
|||||||
@Resource
|
@Resource
|
||||||
private EnterpriseStandardAuthUserService authUserService;
|
private EnterpriseStandardAuthUserService authUserService;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 下载文件
|
||||||
|
*
|
||||||
|
* @param id 传入文件id
|
||||||
|
* @param request
|
||||||
|
* @param response
|
||||||
|
*/
|
||||||
|
@GetMapping(value = "/download/{id}")
|
||||||
|
public void download(@PathVariable String id, HttpServletRequest request, HttpServletResponse response) {
|
||||||
|
downloadAndViewWithWaterMark(id, response);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 预览图片
|
* 预览图片
|
||||||
*
|
*
|
||||||
|
|||||||
Reference in New Issue
Block a user