POI导出 Excel 加水印支持 SXSSFWorkbook 和 XSSFWorkbook 模式

本文介绍了一个支持SXSSFWorkbook和XSSFWorkbook模式的Excel水印添加工具类,通过反射解决了SXSSFWorkbook模式下无法直接添加水印的问题,并提供了详细的实现步骤及示例。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >


前言

最近要完成 Excel 导出添加水印,但是查了一圈一般导出只支持 XSSFWorkbook ,没有支持 SXSSFWorkbook 的,接下来我要完成导出 Excel SXSSFWorkbook 和 XSSFWorkbook 模式添加水印
效果图
在这里插入图片描述


一、导入maven依赖

		<!--POI 相关依赖 start -->
		<dependency>
			<groupId>org.apache.poi</groupId>
			<artifactId>poi</artifactId>
			<version>3.16</version>
		</dependency>
		<dependency>
			<groupId>org.apache.poi</groupId>
			<artifactId>poi-ooxml</artifactId>
			<version>3.16</version>
		</dependency>
		<dependency>
			<groupId>org.apache.poi</groupId>
			<artifactId>poi-ooxml-schemas</artifactId>
			<version>3.16</version>
		</dependency>
		<dependency>
			<groupId>org.apache.poi</groupId>
			<artifactId>ooxml-schemas</artifactId>
			<version>1.4</version>
		</dependency>
		<!--POI 相关依赖 end -->
		
		<!--hutool 工具类 -->
		<dependency>
			<groupId>cn.hutool</groupId>
			<artifactId>hutool-all</artifactId>
			<version>4.5.11</version>
		</dependency>

hutool 工具类主要是用到里面的反射工具类,大家自己写反射就可以不用这个工具包。

二、工具类

代码如下:

package 你的包;


import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.ReflectUtil;
import org.apache.poi.openxml4j.opc.PackagePartName;
import org.apache.poi.openxml4j.opc.PackageRelationship;
import org.apache.poi.openxml4j.opc.TargetMode;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.streaming.SXSSFSheet;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.*;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Date;

/**
 * Excel 添加水印。支持 SXSSFWorkbook 和 XSSFWorkbook 模式
 *
 * @author Kim Jong-un
 * @since 2022-01-17
 */
public class WaterMarkUtil {

    /**
     * Excel 导出添加水印
     *
     * @param workbook ExcelWorkbook
     */
    public static void insertWaterMarkTextToXlsx(Workbook workbook) throws IOException {
        String sysName = "朝鲜合众国运营平台"; //系统名称
        String userName = "金正恩"; //导出人
        String date = DateUtil.format(new Date(), DatePattern.NORM_DATE_PATTERN); //yyyy-MM-dd
        String waterMarkText = sysName + "\n" + userName + "  " + date;

        if (workbook instanceof SXSSFWorkbook) {
            insertWaterMarkTextToXlsx((SXSSFWorkbook) workbook, waterMarkText);
        } else if (workbook instanceof XSSFWorkbook) {
            insertWaterMarkTextToXlsx((XSSFWorkbook) workbook, waterMarkText);
        }
        //throw new RemoteException("HSSFWorkbook 模式不支持 Excel 水印");
    }


    /**
     * 给 Excel 添加水印
     *
     * @param workbook      SXSSFWorkbook
     * @param waterMarkText 水印文字内容
     */
    public static void insertWaterMarkTextToXlsx(SXSSFWorkbook workbook, String waterMarkText) throws IOException {
        BufferedImage image = createWatermarkImage(waterMarkText);
        ByteArrayOutputStream imageOs = new ByteArrayOutputStream();
        ImageIO.write(image, "png", imageOs);
        int pictureIdx = workbook.addPicture(imageOs.toByteArray(), XSSFWorkbook.PICTURE_TYPE_PNG);
        XSSFPictureData pictureData = (XSSFPictureData) workbook.getAllPictures().get(pictureIdx);
        for (int i = 0; i < workbook.getNumberOfSheets(); i++) {//获取每个Sheet表
            SXSSFSheet sheet = workbook.getSheetAt(i);
            //这里由于 SXSSFSheet 没有 getCTWorksheet() 方法,通过反射取出 _sh 属性
            XSSFSheet shReflect = (XSSFSheet) ReflectUtil.getFieldValue(sheet, "_sh");
            PackagePartName ppn = pictureData.getPackagePart().getPartName();
            String relType = XSSFRelation.IMAGES.getRelation();
            PackageRelationship pr = shReflect.getPackagePart().addRelationship(ppn, TargetMode.INTERNAL, relType, null);
            shReflect.getCTWorksheet().addNewPicture().setId(pr.getId());
        }
    }


    /**
     * 给 Excel 添加水印
     *
     * @param workbook      XSSFWorkbook
     * @param waterMarkText 水印文字内容
     */
    public static void insertWaterMarkTextToXlsx(XSSFWorkbook workbook, String waterMarkText) throws IOException {
        BufferedImage image = createWatermarkImage(waterMarkText);
        ByteArrayOutputStream imageOs = new ByteArrayOutputStream();
        ImageIO.write(image, "png", imageOs);
        int pictureIdx = workbook.addPicture(imageOs.toByteArray(), XSSFWorkbook.PICTURE_TYPE_PNG);
        XSSFPictureData pictureData = workbook.getAllPictures().get(pictureIdx);
        for (int i = 0; i < workbook.getNumberOfSheets(); i++) {//获取每个Sheet表
            XSSFSheet sheet = workbook.getSheetAt(i);
            PackagePartName ppn = pictureData.getPackagePart().getPartName();
            String relType = XSSFRelation.IMAGES.getRelation();
            PackageRelationship pr = sheet.getPackagePart().addRelationship(ppn, TargetMode.INTERNAL, relType, null);
            sheet.getCTWorksheet().addNewPicture().setId(pr.getId());
        }
    }

    /**
     * 创建水印图片
     *
     * @param waterMark 水印文字
     */
    public static BufferedImage createWatermarkImage(String waterMark) {
        String[] textArray = waterMark.split("\n");
        Font font = new Font("microsoft-yahei", Font.PLAIN, 32);
        int width = 500;
        int height = 400;

        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        // 背景透明 开始
        Graphics2D g = image.createGraphics();
        image = g.getDeviceConfiguration().createCompatibleImage(width, height, Transparency.TRANSLUCENT);
        g.dispose();
        // 背景透明 结束
        g = image.createGraphics();
        g.setColor(new Color(Color.lightGray.getRGB()));// 设定画笔颜色
        g.setFont(font);// 设置画笔字体
        //   g.shear(0.1, -0.26);// 设定倾斜度

//        设置字体平滑
        g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);

        //文字从中心开始输入,算出文字宽度,左移动一半的宽度,即居中
        FontMetrics fontMetrics = g.getFontMetrics(font);

        // 水印位置
        int x = width / 2;
        int y = height / 2;
        // 设置水印旋转
        g.rotate(Math.toRadians(-40), x, y);
        for (String s : textArray) {
            // 文字宽度
            int textWidth = fontMetrics.stringWidth(s);
            g.drawString(s, x - (textWidth / 2), y);// 画出字符串
            y = y + font.getSize();
        }

        g.dispose();// 释放画笔
        return image;
    }

    /**
     * 设置打印的参数
     *
     * @param wb XSSFWorkbook
     */
    public static void setPrintParams(XSSFWorkbook wb) {
        XSSFSheet sheet = wb.getSheetAt(0);
        XSSFPrintSetup printSetup = sheet.getPrintSetup();
        // 打印方向,true:横向,false:纵向(默认
        printSetup.setLandscape(true);
        //设置A4纸
        printSetup.setPaperSize(XSSFPrintSetup.A4_PAPERSIZE);
        // 将整个工作表打印在一页(缩放),如果行数很多的话,可能会出问题
        // sheet.setAutobreaks(true);
        //将所有的列调整为一页,行数多的话,自动分页
        printSetup.setScale((short) 70);//缩放的百分比,自行调整
        sheet.setAutobreaks(false);
    }
}

三、如何使用

代码如下(示例1):

	XSSFWorkbook workbook = new XSSFWorkbook();
	//...你的逻辑...
	WaterMarkUtil.insertWaterMarkTextToXlsx(workbook); //调用这段代码后水印就添加成功了

代码如下(示例2):

	SXSSFWorkbook workbook = new SXSSFWorkbook();
	//...你的逻辑...
	WaterMarkUtil.insertWaterMarkTextToXlsx(workbook); //调用这段代码后水印就添加成功了

四、总结

  1. 注意这个工具类不支持 HSSFWorkbook 导出的方式
  2. SXSSFWorkbook 模式导出为什么要用反射?
    因为 SXSSFWorkbook 通过
    SXSSFSheet sheet = workbook.getSheetAt(i);
    获取SXSSFSheetSXSSFSheet中没有XSSFSheet的方法,
    而且SXSSFSheet中的XSSFSheet是被final修饰的final XSSFSheet _sh;
    所以可以用个取到反射从SXSSFSheet中取到XSSFSheet,这样就解决了没有XSSFSheet中类中方法的问题了在这里插入图片描述

    五、参考

    https://blog.youkuaiyun.com/dream_broken/article/details/115489269
    https://www.cnblogs.com/skyislimit/articles/10514719.html
    https://blog.youkuaiyun.com/qq_37258792/article/details/111302978
Java中,我们可以使用Apache POI库来操作Excel文件并添加水。以下是一个简单的步骤概述: 1. **添加依赖**: 首先,需要将Apache POI及其衍生库(如XSSF,代表Excel 2007及更高版本)添加到你的项目中。如果你使用Maven,可以在pom.xml文件中加入以下依赖: ```xml <dependency> <groupId>org.apache.poi</groupId> <artifactId>poi-ooxml</artifactId> <version>5.0.0</version> </dependency> ``` 2. **创建Watermark对象**: 使用`XSSFRenderingHint`来设置水样式。例如,你可以创建一个`XSSFRichTextString`对象作为水内容,并设置透明度: ```java XSSFRichTextString watermark = new XSSFRichTextString("水文字"); XSSFRenderingHint hint = new XSSFRenderingHint(); hint.setWatermark(watermark); hint.setFillPattern(WatermarkFillPattern.BACKGROUND); ``` 3. **添加水到工作表**: 在`XSSFWorkbook`对象上创建一个新的`XSSFSheet`,然后在适当的单元格位置设置水: ```java XSSFSheet sheet = workbook.createSheet("Sheet1"); CellRangeAddress range = new CellRangeAddress(0, 0, 0, 0); // 设置水范围 sheet.addNewDrawingCellRange(range); Drawing<?> drawing = sheet.getDrawingPatriarch(); ClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, 0, 0, 0, 0); Picture pic = drawing.createPicture(anchor, ImageDataFactory.createFromStream(new FileInputStream("watermark.png"))); pic.setRenderHint(hint); ``` 这里假设你有一个名为"watermark.png"的图片文件作为水。 4. **保存Excel文件**: 最后,通过`Workbook.write()`方法将含有水的工作簿写入到磁盘上的Excel文件。 ```java workbook.write(outputFile); workbook.close(); ```
评论 14
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值