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

feat: 带文件导出PDF带水印

Browse Source
This commit is contained in:
mzaxd
2023-09-28 16:35:32 +08:00
parent 96b947915a
commit 8b90d80017
6 changed files with 63 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
laws-modules/src/main/java/com/jero/modules/laws/common/controller/LawsCommonController.java
+1 -1
View File
@@ -100,7 +100,7 @@ public class LawsCommonController {
// 设置强制下载不打开
response.setContentType("application/force-download");
// 然后在您的controller方法中：
if ("pdf".equals(FileUtil.extName(fileName))) {
if ("pdf".equalsIgnoreCase(FileUtil.extName(fileName))) {
try (InputStream originalInputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url);
InputStream watermarkedInputStream = waterMarkUtil.addWatermarkToPdf(originalInputStream);
OutputStream outputStream = response.getOutputStream()) {
laws-modules/src/main/java/com/jero/modules/laws/common/service/ILawsCommonService.java
+9
View File
@@ -3,6 +3,7 @@ package com.jero.modules.laws.common.service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.jero.modules.laws.common.vo.ManyStandardSelectionBox;
import com.jero.modules.laws.common.vo.ManyStandardSelectionBoxVO;
import com.jero.modules.oss.entity.OSSFile;
import com.jero.modules.tag.entity.LawsTag;
import org.springframework.web.multipart.MultipartFile;
@@ -154,4 +155,12 @@ public interface ILawsCommonService {
* @Description: 增加浏览记录
**/
void addLookHistory(String id, String type, String module);
/**
* 获取文件内容
* @param file
* @return
* @throws Exception
*/
byte[] getMinioFileContent(OSSFile file) throws Exception;
}
laws-modules/src/main/java/com/jero/modules/laws/common/service/impl/LawsCommonServiceImpl.java
+48 -1
View File
@@ -12,6 +12,7 @@ import com.jero.common.constant.enums.YesOrNoEnum;
import com.jero.common.exception.JeroBootException;
import com.jero.common.system.vo.LoginUser;
import com.jero.common.util.MessageUtils;
import com.jero.common.util.MinioUtil;
import com.jero.modules.laws.common.util.ImportUtil;
import com.jero.modules.laws.common.constant.FieldCommon;
import com.jero.modules.laws.common.constant.ResultCommon;
@@ -23,6 +24,7 @@ import com.jero.modules.laws.enterprise.entity.EnterpriseStandardInspectContent;
import com.jero.modules.laws.enterprise.service.EnterpriseStandardAuthUserService;
import com.jero.modules.laws.enterprise.service.EnterpriseStandardInspectContentService;
import com.jero.modules.laws.enterprise.util.EnterpriseStandardUtil;
import com.jero.modules.laws.enterprise.util.WaterMarkUtil;
import com.jero.modules.laws.standard.entity.LawsNodeRelation;
import com.jero.modules.laws.standard.entity.LawsTreeNode;
import com.jero.modules.laws.standard.service.*;
@@ -120,6 +122,8 @@ public class LawsCommonServiceImpl implements ILawsCommonService {
private ILawsCommonService commonService;
@Resource
private EnterpriseStandardAuthUserService esAuthUserService;
@Resource
private WaterMarkUtil waterMarkUtil;
/**
* @Author: liao
@@ -1190,7 +1194,7 @@ public class LawsCommonServiceImpl implements ILawsCommonService {
ZipEntry fileEntry = new ZipEntry(file.getFileRelativePath());
zipOut.putNextEntry(fileEntry);
// 将文件取出存入流中
bytes = ossFileService.getMinioFileContent(file);
bytes = this.getMinioFileContent(file);
zipOut.write(bytes, 0, bytes.length);
zipOut.closeEntry();
}
@@ -1887,4 +1891,47 @@ public class LawsCommonServiceImpl implements ILawsCommonService {
}
}
@Override
public byte[] getMinioFileContent(OSSFile file) throws Exception {
String fileUrl = file.getUrl();
String fileName = file.getFileName();
String minioUrl = MinioUtil.getMinioUrl();
minioUrl = minioUrl + MinioUtil.getBucketName() + "/";
String url = fileUrl.replace(minioUrl, "");
try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
byte[] buf = new byte[1024];
int len;
if ("pdf".equalsIgnoreCase(FileUtil.extName(fileName))) {
// 如果是PDF文件, 添加水印
try (InputStream originalInputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url);
InputStream watermarkedInputStream = waterMarkUtil.addWatermarkToPdf(originalInputStream)) {
if (originalInputStream == null) {
return null;
}
while ((len = watermarkedInputStream.read(buf)) > 0) {
outputStream.write(buf, 0, len);
}
} catch (Exception e) {
log.error(e.getMessage());
throw e;
}
} else {
// 如果不是PDF文件
try (InputStream inputStream = MinioUtil.getMinioFile(MinioUtil.getBucketName(), url)) {
if (inputStream == null) {
return null;
}
while ((len = inputStream.read(buf)) > 0) {
outputStream.write(buf, 0, len);
}
} catch (Exception e) {
log.error(e.getMessage());
throw e;
}
}
return outputStream.toByteArray();
}
}
}
laws-modules/src/main/java/com/jero/modules/laws/enterprise/service/impl/EnterpriseStandardDeclareServiceImpl.java
+5 -1
View File
@@ -7,6 +7,7 @@ import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.jero.common.exception.JeroBootException;
import com.jero.common.util.MessageUtils;
import com.jero.modules.laws.common.constant.ResultCommon;
import com.jero.modules.laws.common.service.ILawsCommonService;
import com.jero.modules.laws.enterprise.entity.EnterpriseStandardDeclare;
import com.jero.modules.laws.enterprise.entity.dto.ESQueryCommonDTO;
import com.jero.modules.laws.enterprise.entity.dto.EnterpriseStandardDeclareExPortDto;
@@ -49,6 +50,9 @@ public class EnterpriseStandardDeclareServiceImpl extends ServiceImpl<Enterprise
@Resource
private IOSSFileService ossFileService;
@Resource
private ILawsCommonService lawsCommonService;
@Override
public IPage<EnterpriseStandardDeclareVo> queryPageList(ESQueryCommonDTO esDeclare, Integer pageNo, Integer pageSize) {
IPage<EnterpriseStandardDeclareVo> page = new Page<>(pageNo, pageSize);
@@ -133,7 +137,7 @@ public class EnterpriseStandardDeclareServiceImpl extends ServiceImpl<Enterprise
zipOut.putNextEntry(fileEntry);
// 将文件取出存入流中
bytes = ossFileService.getMinioFileContent(ossFile);
bytes = lawsCommonService.getMinioFileContent(ossFile);
zipOut.write(bytes, 0, bytes.length);
zipOut.closeEntry();
}