iText系列之PDF文件添加二维码水印教程

Love The Way You Lie 2021-09-20 01:42 653阅读 0赞

继上一篇博客：图片添加二维码水印教程，https://smilenicky.blog.csdn.net/article/details/91653588， 本博客介绍一下，基于iText开源库做PDF文件添加文字水印和图片水印，并基于此基础，事项PDF文件添加二维码水印图片效果

## 一、PDF文件添加水印 ##

maven配置iText的jar，主要不是所有私服都有iText的jar，maven仓库没有的，可以去https://mvnrepository.com/artifact/com.itextpdf/itextpdf/5.5.12 这里下载

<dependency>
    		    <groupId>com.itextpdf</groupId>
    		    <artifactId>itextpdf</artifactId>
    		    <version>5.5.12</version>
    		</dependency>

同样先写个工具类,这里是加文字水印和图片水印的

import java.io.BufferedOutputStream;
    import java.io.File;
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.text.SimpleDateFormat;
    import java.util.Calendar;
    
    import com.itextpdf.text.BaseColor;
    import com.itextpdf.text.DocumentException;
    import com.itextpdf.text.Element;
    import com.itextpdf.text.Image;
    import com.itextpdf.text.pdf.BaseFont;
    import com.itextpdf.text.pdf.PdfContentByte;
    import com.itextpdf.text.pdf.PdfGState;
    import com.itextpdf.text.pdf.PdfReader;
    import com.itextpdf.text.pdf.PdfStamper;
    
    
    
    public class WaterMarkPDFUtils {
    	
    	public static void main(String[] args)throws IOException, DocumentException {
            // 要输出的pdf文件
            BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(new File("C:/Users/admin/Desktop/target.pdf")));
            Calendar cal = Calendar.getInstance();
            SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");
            // 将pdf文件先加水印然后输出
            setWatermark(bos, "C:/Users/admin/Desktop/t1.pdf", format.format(cal.getTime()) + "  下载使用人：" + "测试user", 16);
        }
    
        /**
         * 
         * @param bos输出文件的位置
         * @param filePath
         *            原PDF位置
         * @param waterMarkName
         *            页脚添加水印
         * @param permission
         *            权限码
         * @throws DocumentException
         * @throws IOException
         */
        public static void setWatermark(BufferedOutputStream bos, String filePath, String waterMarkName, int permission)
                throws  IOException, DocumentException {
        	PdfReader reader = new PdfReader(filePath);
            PdfStamper stamper = new PdfStamper(reader, bos);
            int total = reader.getNumberOfPages() + 1;
            PdfContentByte waterMar;
            
            PdfGState gs = new PdfGState();
            long startTime = System.currentTimeMillis();
            System.out.println("PDF加图片水印>> start");
            for (int i = 1; i < total; i++) {
                //content = stamper.getOverContent(i);// 在内容上方加水印
            	waterMar = stamper.getUnderContent(1);//在内容下方加水印
            	// 设置图片透明度为0.2f
            	gs.setFillOpacity(0.2f);
                // 设置笔触字体不透明度为0.4f
        		gs.setStrokeOpacity(0.4f);
        		// 开始水印处理
            	waterMar.beginText();
        		// 设置透明度
        		waterMar.setGState(gs);
                // 设置水印字体参数及大小
             	waterMar.setFontAndSize(BaseFont.createFont(BaseFont.HELVETICA, BaseFont.WINANSI, BaseFont.NOT_EMBEDDED), 60);
             	// 设置水印对齐方式 水印内容 X坐标 Y坐标 旋转角度
             	waterMar.showTextAligned(Element.ALIGN_CENTER, "公司内部文件，请注意保密！",  500, 430, 45);
             	// 设置水印颜色
             	waterMar.setColorFill(BaseColor.GRAY);
        		// 创建水印图片
                Image image = Image.getInstance("C:/Users/admin/Desktop/icon.jpg");
                // 水印图片位置
                image.setAbsolutePosition(380, 720); 
                // 边框固定
                image.scaleToFit(200, 200);
                // 设置旋转弧度
                //image.setRotation(30);// 旋转 弧度
             	// 设置旋转角度
             	//image.setRotationDegrees(45);// 旋转 角度
             	// 设置等比缩放
             	image.scalePercent(90);
             	// 自定义大小
             	image.scaleAbsolute(200,100);
             	// 附件加上水印图片
             	waterMar.addImage(image);
             	// 完成水印添加
             	waterMar.endText();
             	// stroke
             	waterMar.stroke();
            }
            long endTime = System.currentTimeMillis();
            System.out.println("PDF加图片水印>> ok 所用时间:[{}]"+(endTime-startTime)+"s");
            stamper.close();
            reader.close();
        }
       
    }

PDF加上水印  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9zbWlsZW5pY2t5LmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70]

## 二、PDF添加二维码水印 ##

【拓展功能】  
ok，这只是基本功能，然后要对其进行拓展

业务场景：要在上传的pdf文件自动加上二维码水印，用户可以扫描二维码获取对应数据

首先二维码里面其实也就是一些数据，比如一个链接，或者一堆文字等等，这里可以通过Google开源的zxing库来事项生成二维码图片，然后附加到图片，形成水印

maven配置zxing对应jar：

<dependency>
    			<groupId>com.google.zxing</groupId>
    			<artifactId>core</artifactId>
    			<version>2.2</version>
    		</dependency>
    		<dependency>
    			<groupId>com.google.zxing</groupId>
    			<artifactId>javase</artifactId>
    			<version>2.2</version>
    		</dependency>

写个工具类用于生成二维码图片：

import java.awt.image.BufferedImage;
    import java.io.IOException;
    import java.util.Hashtable;
    
    import com.google.zxing.BarcodeFormat;
    import com.google.zxing.EncodeHintType;
    import com.google.zxing.MultiFormatWriter;
    import com.google.zxing.WriterException;
    import com.google.zxing.client.j2se.MatrixToImageWriter;
    import com.google.zxing.common.BitMatrix;
    import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
    
    /**
     * 二维码生成工具类
     */
    public class QrCodeUtils {
    	
    	/**
    	 * 生成二维码
    	 * @author nicky.ma
    	 * @date   2019年6月11日下午4:39:16
    	 * @param contents 二维码的内容
         * @param width 二维码图片宽度
         * @param height 二维码图片高度
    	 */
    	public static BufferedImage createQrCodeBufferdImage(String contents, int width, int height){
    		Hashtable hints= new Hashtable();   
            hints.put(EncodeHintType.CHARACTER_SET, "utf-8");  
            BufferedImage image = null;
            try {
                BitMatrix bitMatrix = new MultiFormatWriter().encode(
                        contents,BarcodeFormat.QR_CODE, width, height, hints);
                image = toBufferedImage(bitMatrix);
            } catch (WriterException e) {
                e.printStackTrace();
            } 
            return image;
    	}
    
    	public static BufferedImage toBufferedImage(BitMatrix matrix) {
            int width = matrix.getWidth();
            int height = matrix.getHeight();
            BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
            for (int x = 0; x < width; x++) {
              for (int y = 0; y < height; y++) {
                image.setRGB(x, y, matrix.get(x, y) ? 0xFF000000 : 0xFFFFFFFF);
              }
            }
            return image;
        }
    }

import java.awt.AlphaComposite;
    import java.awt.Color;
    import java.awt.Graphics2D;
    import java.awt.Image;
    import java.awt.RenderingHints;
    import java.awt.image.BufferedImage;
    import java.io.BufferedOutputStream;
    import java.io.File;
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.io.InputStream;
    import java.io.OutputStream;
    import java.text.SimpleDateFormat;
    import java.util.Date;
    
    import javax.imageio.ImageIO;
    import javax.swing.ImageIcon;
    
    import com.itextpdf.text.BadElementException;
    import com.itextpdf.text.DocumentException;
    import com.itextpdf.text.Rectangle;
    import com.itextpdf.text.pdf.PdfContentByte;
    import com.itextpdf.text.pdf.PdfGState;
    import com.itextpdf.text.pdf.PdfReader;
    import com.itextpdf.text.pdf.PdfStamper;
    import com.itextpdf.text.pdf.PdfStructTreeController.returnType;
    import com.itextpdf.text.pdf.parser.PdfImageObject.ImageBytesType;
    import com.stuff.stuffmanage.model.CommonStuffModel;
    /**
     * 
     * <pre>
     *  进行水印处理. <br>
     * </pre>
     *
     * @author nicky
     * @date 2019/06/11
     */
    public class WaterMarkUtils {
    	
    	//Logger LOG = LoggerFactory.getLogger(WaterMarkUtils.class);
    	
    	
    	/**
    	 *  生成二维码
    	 * @author nicky
    	 * @date   2019年6月12日下午2:15:51
    	 * @param commonStuffModel
    	 * @return
    	 */
    	private static BufferedImage createQrCodeImg(CommonStuffModel commonStuffModel){
    		StringBuffer strBuf = new StringBuffer();
    		strBuf.append("材料入库时间：").append(new SimpleDateFormat("yyyy-MM-dd").format(new Date())).append("\n");
    		strBuf.append("材料有效期：").append(commonStuffModel.getValidEndDateStr()).append("\n");
    		strBuf.append("材料名称：").append(commonStuffModel.getStuffName()).append("\n");
    		strBuf.append("材料目录：").append(commonStuffModel.getDirName()).append("\n");
    		strBuf.append("材料版本：").append(commonStuffModel.getVersion()).append("\n");
    		strBuf.append("出具单位：").append(commonStuffModel.getIssueUnit()).append("\n");
    		// 生成二维码
    		BufferedImage bufferedImage = QrCodeUtils.createQrCodeBufferdImage(strBuf.toString(), 175, 175);
    		return bufferedImage;
    	}
    	
    	/**
    	 * PDF附件添加二维码
    	 * @author nicky
    	 * @date   2019年6月11日下午3:42:15
    	 * @param bos 输出文件的位置
         * @param input 输入文件流
    	 */
    	public static void setQrCodeForPDF(BufferedOutputStream bos, InputStream input, 
    			CommonStuffModel commonStuffModel){
    		BufferedImage bufferedImage = createQrCodeImg(apprCommonStuffModel);
    		try {
    			//PDF附件加上二维码水印
    			setWatermarkForPDF(bos, input, bufferedImage);
    		} catch (IOException e) {
    			e.printStackTrace();
    		} catch (DocumentException e) {
    			e.printStackTrace();
    		}
    	}
    	
    	/**
    	 * 为PDF附件添加图片水印
    	 * @author nicky
    	 * @date   2019/6/11 12:00:32
    	 * @param bos 输出文件的位置
         * @param input 输入文件流
         * @param image 水印图片
    	 */
    	public static void setWatermarkForPDF(BufferedOutputStream bos, InputStream input, BufferedImage image)
                throws  IOException, DocumentException {
            PdfReader reader = new PdfReader(input);
            PdfStamper stamper = new PdfStamper(reader, bos);
            int total = reader.getNumberOfPages() + 1;
            PdfContentByte waterMar;
            
            PdfGState gs = new PdfGState();
            long startTime = System.currentTimeMillis();
            System.out.println("PDF加图片水印 start");
            for (int i = 1; i < total; i++) {
                //content = stamper.getOverContent(i);// 在内容上方加水印
            	waterMar = stamper.getUnderContent(1);//在内容下方加水印
            	// 设置图片透明度为0.2f
            	//gs.setFillOpacity(0.2f);
                // 设置笔触字体不透明度为0.4f
        		//gs.setStrokeOpacity(0.4f);
        		// 开始水印处理
            	waterMar.beginText();
        		// 设置透明度
        		waterMar.setGState(gs);
                // 设置水印字体参数及大小
             	//waterMar.setFontAndSize(BaseFont.createFont(BaseFont.HELVETICA, BaseFont.WINANSI, BaseFont.NOT_EMBEDDED), 60);
             	// 设置水印对齐方式 水印内容 X坐标 Y坐标 旋转角度
             	//waterMar.showTextAligned(Element.ALIGN_CENTER, "公司内部文件，请注意保密！",  500, 430, 45);
             	// 设置水印颜色
             	//waterMar.setColorFill(BaseColor.GRAY);
        		// 创建水印图片
                com.itextpdf.text.Image itextimage = getImage(image,99);
                // 水印图片位置
                itextimage.setAbsolutePosition(415, 745); 
                // 边框固定
                itextimage.scaleToFit(200, 200);
                // 设置旋转弧度
                //image.setRotation(30);// 旋转 弧度
             	// 设置旋转角度
             	//image.setRotationDegrees(45);// 旋转 角度
             	// 设置等比缩放
                //itextimage.scalePercent(90);
             	// 自定义大小
                itextimage.scaleAbsolute(100,100);
             	// 附件加上水印图片
             	waterMar.addImage(itextimage);
             	// 完成水印添加
             	waterMar.endText();
             	// stroke
             	waterMar.stroke();
            }
            long endTime = System.currentTimeMillis();
            System.out.println("PDF加图片水印 ok 所用时间:"+(endTime-startTime)+"s");
            stamper.close();
            reader.close();
        }
    
    	 public static com.itextpdf.text.Image getImage(BufferedImage bufferedImage,float percent){
        	com.itextpdf.text.Image itextImage=null;
    		try {
    			//控制二维码图片背景透明
    			if(1==1){
    				itextImage=com.itextpdf.text.Image.getInstance(bufferedImage,null);
    			} else {
    				itextImage=com.itextpdf.text.Image.getInstance(bufferedImage,new Color(255,255,255));
    			}
    		} catch (BadElementException e) {
    			e.printStackTrace();
    		} catch (IOException e) {
    			e.printStackTrace();
    		}
    		itextImage.setWidthPercentage(0.1f);
    		itextImage.setSpacingAfter(0.1f);
    		itextImage.scalePercent(percent);
    		itextImage.setBorder(Rectangle.NO_BORDER);
    		return itextImage;
    	}
    }

对于上传的文件，我们怎么知道类型？如果用Spring提供的MultipartFile，这里可以获取ContentType来判断，这里只提供思路

/**文件类型集合*/
    	private static Map<String,String> FILE_TYPES =new HashMap<String,String>();
    
    
    	static{
    		FILE_TYPES.put("xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");//xlsx
    		FILE_TYPES.put("xls", "application/vnd.ms-excel");//xls
    		FILE_TYPES.put("docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document");//docx
    		FILE_TYPES.put("doc", "application/msword");//doc
    		FILE_TYPES.put("jpg", "image/jpeg");//jpg
    		FILE_TYPES.put("png", "image/png");//png
    		FILE_TYPES.put("gif", "image/gif");//gif
    		FILE_TYPES.put("bmp", "image/bmp");//bmp
    		FILE_TYPES.put("txt", "text/plain");//txt
    		FILE_TYPES.put("pdf", "application/pdf");//pdf
    		FILE_TYPES.put("zip", "application/x-zip-compressed");//zip
    		FILE_TYPES.put("rar", "application/octet-stream");//rar
    	}

有了工具类之后，我们需要获取文件上传的inputStream

public void upload(MultipartFile myfiles,String url,String rootPath,CommonStuffModel commonStuffModel)throws Exception{
    	
            if(!myfiles.isEmpty()){   
                File localFile = new File(rootPath+url);  
                File parentFile = localFile.getParentFile();
                if(!parentFile.exists()){
                	parentFile.mkdirs();
    			}
               
                String contentType = myfiles.getContentType();
             if (FILE_TYPES.get("pdf").equals(contentType)) {//上传了PDF附件
    					InputStream inputStream = myfiles.getInputStream();
    					BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(localFile));
    	    			WaterMarkUtils.setQrCodeForPDF(bos, inputStream, commonStuffModel);
    	            }else{
    	            	myfiles.transferTo(localFile);
    	            }
    		}
    }

ok，然后对上传的PDF文件就可以加上二维码  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9zbWlsZW5pY2t5LmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70 1]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9zbWlsZW5pY2t5LmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70]: /images/20210812/bb58133b3daf49bcb2cd020b3f327ebf.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9zbWlsZW5pY2t5LmJsb2cuY3Nkbi5uZXQ_size_16_color_FFFFFF_t_70 1]: /images/20210812/129a49007aba4c869757183b57860c6b.png