ZQ/laws-sinotruk-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 不带水印下载接口

Browse Source
This commit is contained in:
mzaxd
2024-01-02 10:07:26 +08:00
parent b77af10c65
commit d4e6b778af
2 changed files with 45 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files
laws-module-system/src/main/java/com/jero/modules/system/controller/CommonController.java
+45 -30
View File
@@ -12,6 +12,7 @@ import com.jero.common.util.*;
import com.jero.modules.oss.entity.OSSFile; import com.jero.modules.oss.entity.OSSFile;
import com.jero.modules.oss.service.IOSSFileService; import com.jero.modules.oss.service.IOSSFileService;
import com.jero.modules.system.util.SysWaterMarkUtil; import com.jero.modules.system.util.SysWaterMarkUtil;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
@@ -50,7 +51,7 @@ public class CommonController {
private ISysBaseAPI sysBaseAPI; private ISysBaseAPI sysBaseAPI;
@Resource @Resource
private IOSSFileService ossFileService; private IOSSFileService ossFileService;
@Resource @Resource
private SysWaterMarkUtil sysWaterMarkUtil; private SysWaterMarkUtil sysWaterMarkUtil;
@@ -61,29 +62,30 @@ public class CommonController {
/** /**
* 本地：local minio：minio 阿里：alioss * 本地：local minio：minio 阿里：alioss
*/ */
@Value(value="${jero.uploadType}") @Value(value = "${jero.uploadType}")
private String uploadType; private String uploadType;
/** /**
* 文件后缀黑名单 * 文件后缀黑名单
*/ */
@Value(value="${jero.fileSuffixLimits}") @Value(value = "${jero.fileSuffixLimits}")
private String[] fileSuffixLimits; private String[] fileSuffixLimits;
private static final String UTF8 ="UTF-8"; private static final String UTF8 = "UTF-8";
private static final String FILE_VIEW_ERROR = "预览文件失败"; private static final String FILE_VIEW_ERROR = "预览文件失败";
/** /**
* @Author 政辉
* @return * @return
* @Author 政辉
*/ */
@GetMapping("/403") @GetMapping("/403")
public Result<String> noauth() { public Result<String> noauth() {
return Result.error("没有权限，请联系管理员授权"); return Result.error("没有权限，请联系管理员授权");
} }
/** /**
* 文件上传统一方法 * 文件上传统一方法
*
* @param request * @param request
* @param response * @param response
* @return * @return
@@ -102,27 +104,27 @@ public class CommonController {
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
// 获取上传文件对象 // 获取上传文件对象
MultipartFile file = multipartRequest.getFile("file"); MultipartFile file = multipartRequest.getFile("file");
if(file==null) { if (file == null) {
return result; return result;
} }
// 文件类型是否处于黑名单 // 文件类型是否处于黑名单
if(CommonUtils.limitFileSuffix(file.getOriginalFilename(),fileSuffixLimits)){ if (CommonUtils.limitFileSuffix(file.getOriginalFilename(), fileSuffixLimits)) {
result.setMessage("该文件类型不允许上传"); result.setMessage("该文件类型不允许上传");
result.setSuccess(false); result.setSuccess(false);
result.setCode(0); result.setCode(0);
return result; return result;
} }
if(oConvertUtils.isEmpty(bizPath)){ if (oConvertUtils.isEmpty(bizPath)) {
bizPath = ""; bizPath = "";
} }
if(CommonConstant.UPLOAD_TYPE_LOCAL.equals(uploadType)){ if (CommonConstant.UPLOAD_TYPE_LOCAL.equals(uploadType)) {
// 本地上传 // 本地上传
savePath = CommonUtils.uploadLocal(file,bizPath,uploadpath); savePath = CommonUtils.uploadLocal(file, bizPath, uploadpath);
}else{ } else {
// minio上传 // minio上传
savePath = CommonUtils.upload(file, bizPath, uploadType); savePath = CommonUtils.upload(file, bizPath, uploadType);
} }
if(oConvertUtils.isNotEmpty(savePath)){ if (oConvertUtils.isNotEmpty(savePath)) {
//上传成功 进行数据库存储 //上传成功 进行数据库存储
OSSFile ossFile = new OSSFile(); OSSFile ossFile = new OSSFile();
// 文件名 // 文件名
@@ -134,7 +136,7 @@ public class CommonController {
result.setMessage(savePath); result.setMessage(savePath);
result.setResult(ossFile); result.setResult(ossFile);
result.setSuccess(true); result.setSuccess(true);
}else { } else {
result.setMessage("上传失败！"); result.setMessage("上传失败！");
result.setSuccess(false); result.setSuccess(false);
} }
@@ -144,24 +146,24 @@ public class CommonController {
/** /**
* 下载文件 * 下载文件
* *
* @param id 传入文件id * @param id 传入文件id
* @param request * @param request
* @param response * @param response
*/ */
@GetMapping(value = "/download/{id}") @GetMapping(value = "/download/{id}")
public void download(@PathVariable String id,HttpServletRequest request, HttpServletResponse response) { public void download(@PathVariable String id, HttpServletRequest request, HttpServletResponse response) {
downloadAndView(id, response); downloadAndViewWithWaterMark(id, response);
} }
/** /**
* 预览图片 * 预览图片
* *
* @param id 传入文件id * @param id 传入文件id
* @param request * @param request
* @param response * @param response
*/ */
@GetMapping(value = "/view/{id}") @GetMapping(value = "/view/{id}")
public void view(@PathVariable String id,HttpServletRequest request, HttpServletResponse response) { public void view(@PathVariable String id, HttpServletRequest request, HttpServletResponse response) {
downloadAndViewWithWaterMark(id, response); downloadAndViewWithWaterMark(id, response);
} }
@@ -229,6 +231,18 @@ public class CommonController {
} }
} }
/**
* 下载不带水印的文件
*
* @param id 传入文件id
* @param response
*/
@ApiOperation(value = "下载不带水印的文件", notes = "下载不带水印的文件")
@GetMapping(value = "/downloadWithoutWaterMark/{id}")
public void downloadWithoutWaterMark(@PathVariable String id, HttpServletResponse response) {
downloadAndView(id, response);
}
private void downloadAndViewWithWaterMark(@PathVariable String id, HttpServletResponse response) { private void downloadAndViewWithWaterMark(@PathVariable String id, HttpServletResponse response) {
// 查询数据表数据是否存在 // 查询数据表数据是否存在
LambdaQueryWrapper<OSSFile> queryWrapper = new LambdaQueryWrapper<>(); LambdaQueryWrapper<OSSFile> queryWrapper = new LambdaQueryWrapper<>();
@@ -277,14 +291,14 @@ public class CommonController {
) { ) {
byte[] buf = new byte[1024]; byte[] buf = new byte[1024];
int len; int len;
if(inputStream==null) { if (inputStream == null) {
return; return;
} }
while ((len = inputStream.read(buf)) > 0) { while ((len = inputStream.read(buf)) > 0) {
outputStream.write(buf, 0, len); outputStream.write(buf, 0, len);
} }
response.flushBuffer(); response.flushBuffer();
} catch (Exception e){ } catch (Exception e) {
log.error(e.getMessage()); log.error(e.getMessage());
response.setStatus(404); response.setStatus(404);
e.printStackTrace(); e.printStackTrace();
@@ -299,9 +313,9 @@ public class CommonController {
} }
/** /**
* @功能：pdf预览Iframe
* @param modelAndView * @param modelAndView
* @return * @return
* @功能：pdf预览Iframe
*/ */
@RequestMapping("/pdf/pdfPreviewIframe") @RequestMapping("/pdf/pdfPreviewIframe")
public ModelAndView pdfPreviewIframe(ModelAndView modelAndView) { public ModelAndView pdfPreviewIframe(ModelAndView modelAndView) {
@@ -310,8 +324,9 @@ public class CommonController {
} }
/** /**
* 把指定URL后的字符串全部截断当成参数 * 把指定URL后的字符串全部截断当成参数
* 这么做是为了防止URL中包含中文或者特殊字符（/等）时，匹配不了的问题 * 这么做是为了防止URL中包含中文或者特殊字符（/等）时，匹配不了的问题
*
* @param request * @param request
* @return * @return
*/ */
@@ -344,7 +359,7 @@ public class CommonController {
headers.set("X-Access-Token", token); headers.set("X-Access-Token", token);
// 发送请求 // 发送请求
String httpURL = URLDecoder.decode(url, UTF8); String httpURL = URLDecoder.decode(url, UTF8);
ResponseEntity<String> response = RestUtil.request(httpURL, method, headers , variables, params, String.class); ResponseEntity<String> response = RestUtil.request(httpURL, method, headers, variables, params, String.class);
// 封装返回结果 // 封装返回结果
Result<Object> result = new Result<>(); Result<Object> result = new Result<>();
int statusCode = response.getStatusCodeValue(); int statusCode = response.getStatusCodeValue();
@@ -391,18 +406,18 @@ public class CommonController {
public void view(HttpServletRequest request, HttpServletResponse response) { public void view(HttpServletRequest request, HttpServletResponse response) {
// ISO-8859-1 ==> UTF-8 进行编码转换 // ISO-8859-1 ==> UTF-8 进行编码转换
String imgPath = extractPathFromPattern(request); String imgPath = extractPathFromPattern(request);
if(oConvertUtils.isEmpty(imgPath) || "null".equals(imgPath)){ if (oConvertUtils.isEmpty(imgPath) || "null".equals(imgPath)) {
return; return;
} }
imgPath = imgPath.replace("..", "").replace("../",""); imgPath = imgPath.replace("..", "").replace("../", "");
if (imgPath.endsWith(",")) { if (imgPath.endsWith(",")) {
imgPath = imgPath.substring(0, imgPath.length() - 1); imgPath = imgPath.substring(0, imgPath.length() - 1);
} }
String filePath = uploadpath + File.separator + imgPath; String filePath = uploadpath + File.separator + imgPath;
File file = new File(filePath); File file = new File(filePath);
if(!file.exists()){ if (!file.exists()) {
response.setStatus(404); response.setStatus(404);
throw new JeroBootException("文件["+imgPath+"]不存在.."); throw new JeroBootException("文件[" + imgPath + "]不存在..");
} }
response.setContentType("application/force-download");// 设置强制下载不打开 response.setContentType("application/force-download");// 设置强制下载不打开
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1)); response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
laws-modules/src/main/java/com/jero/modules/laws/common/controller/LawsCommonController.java
-71
View File
@@ -86,75 +86,4 @@ public class LawsCommonController {
return Result.OK(lawsCommonService.getSplitsByStandardNumber(standardNumber)); return Result.OK(lawsCommonService.getSplitsByStandardNumber(standardNumber));
} }
/**
* 下载带水印的PDF文件
*
* @param id 传入文件id
* @param response
*/
@ApiOperation(value="PDF带水印下载", notes="PDF带水印下载")
@GetMapping(value = "/downloadWithWaterMark/{id}")
public void downloadWithWaterMark(@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("文件不存在..");
}
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 = waterMarkUtil.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();
}
}
}
} }