java 给pdf添加水印

引用

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

<dependency>
    <groupId>com.itextpdf</groupId>
    <artifactId>itext-asian</artifactId>
    <version>5.2.0</version>
</dependency>

<dependency>
    <groupId>org.dom4j</groupId>
    <artifactId>dom4j</artifactId>
    <version>2.1.3</version>
</dependency>

工具类

	/**
     * 斜角排列、全屏多个重复的花式文字水印
     *
     * @param bytes             需要加水印的PDF读取输入流
     * @param output            输出生成PDF的输出流
     * @param waterMarkString   水印字符
     * @param xAmout            x轴步长
     * @param yAmout            y轴步长
     * @param opacity           水印透明度
     * @param rotation          水印文字旋转角度，一般为45度角
     * @param waterMarkFontSize 水印字体大小
     * @param color             水印字体颜色
     */
    public static void stringWaterMark(byte[] bytes, FileOutputStream output, String waterMarkString, int xAmout, int yAmout, float opacity, float rotation, int waterMarkFontSize, BaseColor color) {
        try {

            PdfReader reader = new PdfReader(bytes);
            PdfReader.unethicalreading = true;
            PdfStamper stamper = new PdfStamper(reader, output);

            // 添加中文字体
            BaseFont baseFont = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);

            int total = reader.getNumberOfPages() + 1;

            PdfContentByte over;
            // 给每一页加水印
            for (int i = 1; i < total; i++) {
                Rectangle pageRect = stamper.getReader().getPageSizeWithRotation(i);
                // 计算水印每个单位步长X,Y
//                float x = pageRect.getWidth() / xAmout;
//                float y = pageRect.getHeight() / yAmout;
				// 间隔长度
                xAmout += (waterMarkString.length() * 10);
                //计算水印个数
                float x = pageRect.getWidth() / xAmout;
                float y = pageRect.getHeight() / yAmout;

                over = stamper.getOverContent(i);
                PdfGState gs = new PdfGState();
                // 设置透明度为
                gs.setFillOpacity(opacity);

                over.setGState(gs);
                over.saveState();

                over.beginText();
                over.setColorFill(color);
                over.setFontAndSize(baseFont, waterMarkFontSize);

                for (int n = 0; n < x + 1; n++) {
                    for (int m = 0; m < y + 1; m++) {
                        BigDecimal xAmoutBig = new BigDecimal(xAmout);
                        BigDecimal yAmoutBig = new BigDecimal(yAmout);
                        BigDecimal mBig = new BigDecimal(m);
                        BigDecimal nBig = new BigDecimal(n);
                        Float x_afferent = xAmoutBig.multiply(nBig).floatValue();
                        Float y_afferent = yAmoutBig.multiply(mBig).floatValue();
                        over.showTextAligned(Element.ALIGN_CENTER, waterMarkString, x_afferent, y_afferent, rotation);
                    }
                }

                over.endText();
            }
            stamper.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

	/**
     * @param response
     * @param fileName        文件存放名称
     * @param uploadPath         文件地址
     * @return void
     * @Description //TODO 下载文件
     **/
    public static void downFile(HttpServletResponse response, String fileName, String uploadPath) {
        //下載文件
        String filePath = uploadPath + File.separator + fileName;
        //判断文件是否存在
        File file = new File(filePath);
        if (!file.exists()) {
            throw new RuntimeException("文件不存在");
        }
        FileInputStream fis = null;
        OutputStream myout = null;
        BufferedInputStream buff = null;
        try {
            if (file.exists()) {
                response.setHeader("Content-Disposition",
                        "attachment;filename=" + new String((fileName).getBytes("UTF-8"), "ISO8859-1"));
                response.setContentLength((int) file.length());
                //暴露Header
                response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
                // 定义输出类型
                response.setContentType("application/octet-stream");
                fis = new FileInputStream(file);
                buff = new BufferedInputStream(fis);
                // 相当于我们的缓存
                byte[] b = new byte[1024];
                // 该值用于计算当前实际下载了多少字节
                long k = 0;
                // 从response对象中得到输出流,准备下载
                myout = response.getOutputStream();
                // 开始循环下载
                while (k < file.length()) {
                    int j = buff.read(b, 0, 1024);
                    k += j;
                    myout.write(b, 0, j);
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (fis != null) {
                try {
                    fis.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (myout != null) {
                try {
                    myout.flush();
                    myout.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (buff != null) {
                try {
                    buff.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * @param filePath 文件地址
     * @return boolean
     * @Description //TODO 删除单个文件
     * @Param
     **/
    public static boolean deleteFile(String filePath) {
        File file = new File(filePath);
        // 如果文件路径所对应的文件存在，并且是一个文件，则直接删除
        if (file.exists() && file.isFile()) {
            if (file.delete()) {
                LOGGER.info("删除单个文件" + filePath + "成功！");
                return true;
            } else {
                LOGGER.error("删除单个文件" + filePath + "失败！");
                return false;
            }
        } else {
            LOGGER.error("删除单个文件失败：" + filePath + "不存在！");
            return false;
        }
    }

调用类

	public void queryView(HttpServletResponse response, String viewNo, String pageNo) {
		// 输入地址
		String url = "http://....";
		// 输出文件夹
		String outViewUrl = "D:/";
        String fileName = "测试.pdf";
        String outUrl = outViewUrl + File.separator + fileName;

        InputStream inputStream = null;
        FileOutputStream fileOutputStream = null;
        try {
            File file = new File(outUrl);
            Boolean fileCreate = true;
            if (!file.exists()) {
                Files.createDirectories(Path.of(file.getParent()));
                // 在服务器上创建file
                fileCreate = file.createNewFile();
            }
            // 创建成功
            if (fileCreate) {
                fileOutputStream = new FileOutputStream(file);
                URL urlConn = new URL(url);
                HttpURLConnection httpURLConnection = (HttpURLConnection) urlConn.openConnection();
                httpURLConnection.setRequestMethod("GET");
                // 响应
                Integer code = httpURLConnection.getResponseCode();
                if (code == 200) {
                    inputStream = httpURLConnection.getInputStream();
                    // 输入流转成bytes
                    byte[] bytes = inputStream.readAllBytes();
                    HttpRequest.stringWaterMark(bytes, fileOutputStream, waterMarkString, 100, 100, 0.2f, 45, 20, new BaseColor(165, 165, 165));
                    HttpRequest.downFile(response, fileName, outViewUrl);
                }
                // 处理完删除临时文件
                HttpRequest.deleteFile(outUrl);
            }
        } catch (Exception e){
            e.printStackTrace();
        } finally {
            if (fileOutputStream != null){
                try {
                    fileOutputStream.close();
                } catch (Exception e){
                    e.printStackTrace();
                }
            }
            if (inputStream != null){
                try{
                    inputStream.close();
                } catch (Exception e){
                    e.printStackTrace();
                }
            }
        }
	}