1. 添加pom依赖pom.xml
<dependency>
<groupId>com.itextpdf</groupId>
<artifactId>itextpdf</artifactId>
<version>5.5.13</version>
</dependency>
<dependency>
<groupId>com.itextpdf</groupId>
<artifactId>itext-asian</artifactId>
<version>5.2.0</version>
</dependency>
itextpdf
可以兼容中文
2. 创建工具类WatermarkUtils.java
public class WaterMarkUtils {
public static OutputStream addWaterMark(byte[] bytes, OutputStream outputStream, String text, int textWidth, int textHeight,
String imgFile, int imgWidth, int imgHeight) throws IOException, DocumentException {
// 待加水印的文件
PdfReader reader = new PdfReader(bytes);
// 加完水印的文件
PdfStamper stamper = new PdfStamper(reader, outputStream);
// 设置字体
BaseFont font = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.EMBEDDED); // 华文宋体
//BaseFont font = BaseFont.createFont("C:\\Windows\\Fonts\\simhei.ttf", BaseFont.IDENTITY_H, BaseFont.EMBEDDED); // 标准黑体
// 水印透明度
PdfGState gs = new PdfGState();
gs.setFillOpacity(0.5f);
gs.setStrokeOpacity(0.5f);
// PDF总页数
int total = reader.getNumberOfPages() + 1;
// 循环对每页插入水印
PdfContentByte content;
for (int i = 1; i < total; i++) {
// 水印在之前文本之上
content = stamper.getOverContent(i);
// 透明度
content.setGState(gs);
// 图片水印
if (imgFile != null) {
Image image = null;
if (imgFile != null) {
image = Image.getInstance(imgFile);
image.setAbsolutePosition(imgWidth, imgHeight);
// 设置图片的显示大小
image.scaleToFit(100, 125);
}
content.addImage(image);
}
// 文字水印
if (text != null) {
content.beginText();
// 设置颜色 默认为蓝色
content.setColorFill(new BaseColor(204, 204, 204));
//content.setColorFill(Color.BLUE);
// 设置字体及字号
content.setFontAndSize(font, 25);
// 设置起始位置
content.setTextMatrix(textWidth, textHeight);
// 字符间距
content.setCharacterSpacing(2);
// 中间水印
content.showTextAligned(Element.ALIGN_LEFT, text, textWidth, textHeight, 45);
// 底部水印
// for (int k = 0; k < text.length(); k++) {
// // 距离底边的距离
// content.setTextRise(10);
//
// // 将char转成字符串
// content.showText(String.valueOf(text.charAt(k)));
// }
content.endText();
}
}
stamper.close();
reader.close();
return outputStream;
}
}
3. 从URL获取文件二进制数据HttpRequestUtils.java
public class HttpRequestUtils {
/**
* 发送get请求 返回byte
* @param url 路径
* @return
*/
public static byte[] httpGetFile(String url){
//get请求返回结果
byte[] byteResult = null;
try {
CloseableHttpClient client = HttpClients.createDefault();
//发送get请求
HttpGet request = new HttpGet(url);
CloseableHttpResponse response = client.execute(request);
try {
/**请求发送成功,并得到响应**/
if (response.getStatusLine().getStatusCode() == 200) {
byteResult = EntityUtils.toByteArray(response.getEntity());
} else {
logger.error("get请求提交失败:" + url);
}
} finally {
response.close();
}
} catch (IOException e) {
logger.error("get请求提交失败:" + url, e);
}
return byteResult;
}
}
4. 给文件加水印实例
@RestController
@ResponseBody
@RequestMapping(value = "/api/demo")
public class DemoController {
@GetMapping("/addWaterPdf")
public ResponseEntity<byte[]> addWaterPdf(String url) throws Exception{
// url中获取文件流
byte[] buffer = HttpRequestUtils.httpGetFile(url);
if (Objects.isNull(buffer) || buffer.length == 0) {
throw new RuntimeException("文件不存在");
}
ByteArrayOutputStream baos = new ByteArrayOutputStream();// 输出流实例
String waterText = "这是一个水印";
WatermarkUtils.addWaterMark(buffer, baos, waterText,20, 30, null, 400, 500);
HttpHeaders headers = new HttpHeaders();
// header只支持ASCII,需要把中文通过ISO-8859-1重新编码
String fileName = new String("水印文件.pdf".getBytes("utf-8"), "iso-8859-1");
headers.setContentDispositionFormData("attachment", fileName);
headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
return new ResponseEntity<byte[]>(baos.toByteArray(), headers, HttpStatus.OK);
}
}