maven配置iText的jar,主要不是所有私服都有iText的jar,maven仓库没有的,可以去https://mvnrepository.com/artifact/com.itextpdf/itextpdf/5.5.12 这里下载
1<!-- itextpdf --> 2 <dependency> 3 <groupId>com.itextpdf</groupId> 4 <artifactId>itextpdf</artifactId> 5 <version>5.5.12</version> 6 </dependency>
同样先写个工具类,这里是加文字水印和图片水印的
1import java.io.BufferedOutputStream; 2import java.io.File; 3import java.io.FileOutputStream; 4import java.io.IOException; 5import java.text.SimpleDateFormat; 6import java.util.Calendar; 7 8import com.itextpdf.text.BaseColor; 9import com.itextpdf.text.DocumentException; 10import com.itextpdf.text.Element; 11import com.itextpdf.text.Image; 12import com.itextpdf.text.pdf.BaseFont; 13import com.itextpdf.text.pdf.PdfContentByte; 14import com.itextpdf.text.pdf.PdfGState; 15import com.itextpdf.text.pdf.PdfReader; 16import com.itextpdf.text.pdf.PdfStamper; 17 18 19 20public class WaterMarkPDFUtils { 21 22 public static void main(String[] args)throws IOException, DocumentException { 23 // 要输出的pdf文件 24 BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(new File("C:/Users/admin/Desktop/target.pdf"))); 25 Calendar cal = Calendar.getInstance(); 26 SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss"); 27 // 将pdf文件先加水印然后输出 28 setWatermark(bos, "C:/Users/admin/Desktop/t1.pdf", format.format(cal.getTime()) + " 下载使用人:" + "测试user", 16); 29 } 30 31 /** 32 * 33 * @param bos输出文件的位置 34 * @param filePath 35 * 原PDF位置 36 * @param waterMarkName 37 * 页脚添加水印 38 * @param permission 39 * 权限码 40 * @throws DocumentException 41 * @throws IOException 42 */ 43 public static void setWatermark(BufferedOutputStream bos, String filePath, String waterMarkName, int permission) 44 throws IOException, DocumentException { 45 PdfReader reader = new PdfReader(filePath); 46 PdfStamper stamper = new PdfStamper(reader, bos); 47 int total = reader.getNumberOfPages() + 1; 48 PdfContentByte waterMar; 49 50 PdfGState gs = new PdfGState(); 51 long startTime = System.currentTimeMillis(); 52 System.out.println("PDF加图片水印>> start"); 53 for (int i = 1; i < total; i++) { 54 //content = stamper.getOverContent(i);// 在内容上方加水印 55 waterMar = stamper.getUnderContent(1);//在内容下方加水印 56 // 设置图片透明度为0.2f 57 gs.setFillOpacity(0.2f); 58 // 设置笔触字体不透明度为0.4f 59 gs.setStrokeOpacity(0.4f); 60 // 开始水印处理 61 waterMar.beginText(); 62 // 设置透明度 63 waterMar.setGState(gs); 64 // 设置水印字体参数及大小 65 waterMar.setFontAndSize(BaseFont.createFont(BaseFont.HELVETICA, BaseFont.WINANSI, BaseFont.NOT_EMBEDDED), 60); 66 // 设置水印对齐方式 水印内容 X坐标 Y坐标 旋转角度 67 waterMar.showTextAligned(Element.ALIGN_CENTER, "公司内部文件,请注意保密!", 500, 430, 45); 68 // 设置水印颜色 69 waterMar.setColorFill(BaseColor.GRAY); 70 // 创建水印图片 71 Image image = Image.getInstance("C:/Users/admin/Desktop/icon.jpg"); 72 // 水印图片位置 73 image.setAbsolutePosition(380, 720); 74 // 边框固定 75 image.scaleToFit(200, 200); 76 // 设置旋转弧度 77 //image.setRotation(30);// 旋转 弧度 78 // 设置旋转角度 79 //image.setRotationDegrees(45);// 旋转 角度 80 // 设置等比缩放 81 image.scalePercent(90); 82 // 自定义大小 83 image.scaleAbsolute(200,100); 84 // 附件加上水印图片 85 waterMar.addImage(image); 86 // 完成水印添加 87 waterMar.endText(); 88 // stroke 89 waterMar.stroke(); 90 } 91 long endTime = System.currentTimeMillis(); 92 System.out.println("PDF加图片水印>> ok 所用时间:[{}]"+(endTime-startTime)+"s"); 93 stamper.close(); 94 reader.close(); 95 } 96 97} 98
PDF加上水印 
【拓展功能】 ok,这只是基本功能,然后要对其进行拓展
业务场景:要在上传的pdf文件自动加上二维码水印,用户可以扫描二维码获取对应数据
首先二维码里面其实也就是一些数据,比如一个链接,或者一堆文字等等,这里可以通过Google开源的zxing库来事项生成二维码图片,然后附加到图片,形成水印
maven配置zxing对应jar:
1<!-- 条形码、二维码生成 --> 2 <dependency> 3 <groupId>com.google.zxing</groupId> 4 <artifactId>core</artifactId> 5 <version>2.2</version> 6 </dependency> 7 <dependency> 8 <groupId>com.google.zxing</groupId> 9 <artifactId>javase</artifactId> 10 <version>2.2</version> 11 </dependency>
写个工具类用于生成二维码图片:
1import java.awt.image.BufferedImage; 2import java.io.IOException; 3import java.util.Hashtable; 4 5import com.google.zxing.BarcodeFormat; 6import com.google.zxing.EncodeHintType; 7import com.google.zxing.MultiFormatWriter; 8import com.google.zxing.WriterException; 9import com.google.zxing.client.j2se.MatrixToImageWriter; 10import com.google.zxing.common.BitMatrix; 11import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel; 12 13/** 14 * 二维码生成工具类 15 */ 16public class QrCodeUtils { 17 18 /** 19 * 生成二维码 20 * @author nicky.ma 21 * @date 2019年6月11日下午4:39:16 22 * @param contents 二维码的内容 23 * @param width 二维码图片宽度 24 * @param height 二维码图片高度 25 */ 26 public static BufferedImage createQrCodeBufferdImage(String contents, int width, int height){ 27 Hashtable hints= new Hashtable(); 28 hints.put(EncodeHintType.CHARACTER_SET, "utf-8"); 29 BufferedImage image = null; 30 try { 31 BitMatrix bitMatrix = new MultiFormatWriter().encode( 32 contents,BarcodeFormat.QR_CODE, width, height, hints); 33 image = toBufferedImage(bitMatrix); 34 } catch (WriterException e) { 35 e.printStackTrace(); 36 } 37 return image; 38 } 39 40 public static BufferedImage toBufferedImage(BitMatrix matrix) { 41 int width = matrix.getWidth(); 42 int height = matrix.getHeight(); 43 BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB); 44 for (int x = 0; x < width; x++) { 45 for (int y = 0; y < height; y++) { 46 image.setRGB(x, y, matrix.get(x, y) ? 0xFF000000 : 0xFFFFFFFF); 47 } 48 } 49 return image; 50 } 51} 52 53 54 55import java.awt.AlphaComposite; 56import java.awt.Color; 57import java.awt.Graphics2D; 58import java.awt.Image; 59import java.awt.RenderingHints; 60import java.awt.image.BufferedImage; 61import java.io.BufferedOutputStream; 62import java.io.File; 63import java.io.FileOutputStream; 64import java.io.IOException; 65import java.io.InputStream; 66import java.io.OutputStream; 67import java.text.SimpleDateFormat; 68import java.util.Date; 69 70import javax.imageio.ImageIO; 71import javax.swing.ImageIcon; 72 73import com.itextpdf.text.BadElementException; 74import com.itextpdf.text.DocumentException; 75import com.itextpdf.text.Rectangle; 76import com.itextpdf.text.pdf.PdfContentByte; 77import com.itextpdf.text.pdf.PdfGState; 78import com.itextpdf.text.pdf.PdfReader; 79import com.itextpdf.text.pdf.PdfStamper; 80import com.itextpdf.text.pdf.PdfStructTreeController.returnType; 81import com.itextpdf.text.pdf.parser.PdfImageObject.ImageBytesType; 82import com.stuff.stuffmanage.model.CommonStuffModel; 83/** 84 * 85 * <pre> 86 * 进行水印处理. <br> 87 * </pre> 88 * 89 * @author mazq 90 * @date 2019/06/11 91 */ 92public class WaterMarkUtils { 93 94 //Logger LOG = LoggerFactory.getLogger(WaterMarkUtils.class); 95 96 97 /** 98 * 生成二维码 99 * @author nicky.ma 100 * @date 2019年6月12日下午2:15:51 101 * @param commonStuffModel 102 * @return 103 */ 104 private static BufferedImage createQrCodeImg(CommonStuffModel commonStuffModel){ 105 StringBuffer strBuf = new StringBuffer(); 106 strBuf.append("材料入库时间:").append(new SimpleDateFormat("yyyy-MM-dd").format(new Date())).append("\n"); 107 strBuf.append("材料有效期:").append(commonStuffModel.getValidEndDateStr()).append("\n"); 108 strBuf.append("材料名称:").append(commonStuffModel.getStuffName()).append("\n"); 109 strBuf.append("材料目录:").append(commonStuffModel.getDirName()).append("\n"); 110 strBuf.append("材料版本:").append(commonStuffModel.getVersion()).append("\n"); 111 strBuf.append("出具单位:").append(commonStuffModel.getIssueUnit()).append("\n"); 112 // 生成二维码 113 BufferedImage bufferedImage = QrCodeUtils.createQrCodeBufferdImage(strBuf.toString(), 175, 175); 114 return bufferedImage; 115 } 116 117 /** 118 * PDF附件添加二维码 119 * @author nicky.ma 120 * @date 2019年6月11日下午3:42:15 121 * @param bos 输出文件的位置 122 * @param input 输入文件流 123 */ 124 public static void setQrCodeForPDF(BufferedOutputStream bos, InputStream input, 125 CommonStuffModel commonStuffModel){ 126 BufferedImage bufferedImage = createQrCodeImg(apprCommonStuffModel); 127 try { 128 //PDF附件加上二维码水印 129 setWatermarkForPDF(bos, input, bufferedImage); 130 } catch (IOException e) { 131 e.printStackTrace(); 132 } catch (DocumentException e) { 133 e.printStackTrace(); 134 } 135 } 136 137 /** 138 * 为PDF附件添加图片水印 139 * @author nicky.ma 140 * @date 2019/6/11 12:00:32 141 * @param bos 输出文件的位置 142 * @param input 输入文件流 143 * @param image 水印图片 144 */ 145 public static void setWatermarkForPDF(BufferedOutputStream bos, InputStream input, BufferedImage image) 146 throws IOException, DocumentException { 147 PdfReader reader = new PdfReader(input); 148 PdfStamper stamper = new PdfStamper(reader, bos); 149 int total = reader.getNumberOfPages() + 1; 150 PdfContentByte waterMar; 151 152 PdfGState gs = new PdfGState(); 153 long startTime = System.currentTimeMillis(); 154 System.out.println("PDF加图片水印 start"); 155 for (int i = 1; i < total; i++) { 156 //content = stamper.getOverContent(i);// 在内容上方加水印 157 waterMar = stamper.getUnderContent(1);//在内容下方加水印 158 // 设置图片透明度为0.2f 159 //gs.setFillOpacity(0.2f); 160 // 设置笔触字体不透明度为0.4f 161 //gs.setStrokeOpacity(0.4f); 162 // 开始水印处理 163 waterMar.beginText(); 164 // 设置透明度 165 waterMar.setGState(gs); 166 // 设置水印字体参数及大小 167 //waterMar.setFontAndSize(BaseFont.createFont(BaseFont.HELVETICA, BaseFont.WINANSI, BaseFont.NOT_EMBEDDED), 60); 168 // 设置水印对齐方式 水印内容 X坐标 Y坐标 旋转角度 169 //waterMar.showTextAligned(Element.ALIGN_CENTER, "公司内部文件,请注意保密!", 500, 430, 45); 170 // 设置水印颜色 171 //waterMar.setColorFill(BaseColor.GRAY); 172 // 创建水印图片 173 com.itextpdf.text.Image itextimage = getImage(image,99); 174 // 水印图片位置 175 itextimage.setAbsolutePosition(415, 745); 176 // 边框固定 177 itextimage.scaleToFit(200, 200); 178 // 设置旋转弧度 179 //image.setRotation(30);// 旋转 弧度 180 // 设置旋转角度 181 //image.setRotationDegrees(45);// 旋转 角度 182 // 设置等比缩放 183 //itextimage.scalePercent(90); 184 // 自定义大小 185 itextimage.scaleAbsolute(100,100); 186 // 附件加上水印图片 187 waterMar.addImage(itextimage); 188 // 完成水印添加 189 waterMar.endText(); 190 // stroke 191 waterMar.stroke(); 192 } 193 long endTime = System.currentTimeMillis(); 194 System.out.println("PDF加图片水印 ok 所用时间:"+(endTime-startTime)+"s"); 195 stamper.close(); 196 reader.close(); 197 } 198}
对于上传的文件,我们怎么知道类型?如果用Spring提供的MultipartFile,这里可以获取ContentType来判断,这里只提供思路
1 /**文件类型集合*/ 2 private static Map<String,String> FILE_TYPES =new HashMap<String,String>(); 3 4 5 static{ 6 FILE_TYPES.put("xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");//xlsx 7 FILE_TYPES.put("xls", "application/vnd.ms-excel");//xls 8 FILE_TYPES.put("docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document");//docx 9 FILE_TYPES.put("doc", "application/msword");//doc 10 FILE_TYPES.put("jpg", "image/jpeg");//jpg 11 FILE_TYPES.put("png", "image/png");//png 12 FILE_TYPES.put("gif", "image/gif");//gif 13 FILE_TYPES.put("bmp", "image/bmp");//bmp 14 FILE_TYPES.put("txt", "text/plain");//txt 15 FILE_TYPES.put("pdf", "application/pdf");//pdf 16 FILE_TYPES.put("zip", "application/x-zip-compressed");//zip 17 FILE_TYPES.put("rar", "application/octet-stream");//rar 18 } 19
有了工具类之后,我们需要获取文件上传的inputStream
1public void upload(MultipartFile myfiles,String url,String rootPath,CommonStuffModel commonStuffModel)throws Exception{ 2 3 if(!myfiles.isEmpty()){ 4 File localFile = new File(rootPath+url); 5 File parentFile = localFile.getParentFile(); 6 if(!parentFile.exists()){ 7 parentFile.mkdirs(); 8 } 9 10 String contentType = myfiles.getContentType(); 11 if (FILE_TYPES.get("pdf").equals(contentType)) {//上传了PDF附件 12 InputStream inputStream = myfiles.getInputStream(); 13 BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(localFile)); 14 WaterMarkUtils.setQrCodeForPDF(bos, inputStream, commonStuffModel); 15 }else{ 16 myfiles.transferTo(localFile); 17 } 18 } 19}
ok,然后对上传的PDF文件就可以加上二维码 