POI匯出excel檔案加水印

千代兔兔子發表於2024-10-07

百分百能用,我用的POI版本是5.2.3,效果如下

import lombok.extern.slf4j.Slf4j;
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.xssf.usermodel.*;
import org.openxmlformats.schemas.spreadsheetml.x2006.main.CTSheetProtection;

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

/**
 * Excel表格新增水印
 */
@Slf4j
public class WaterMarkUtil {

    /**
     * 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(null, Font.PLAIN, 30);
        try {
            //指定字型位置 (宋體:simsun.ttf)
            String fontFilePath = "D:\\qxt\\feihuasongti.ttf";
            FileInputStream fontInputStream = new FileInputStream(fontFilePath);
            font = Font.createFont(Font.TRUETYPE_FONT, fontInputStream);
            font = font.deriveFont(30f);  // 設定字型大小為30
        } catch (Exception e) {
            log.error("指定使用漢字字型位置:", e);
            font = new Font("simsun", Font.PLAIN, 30);
        }
        //設定水印的密集程度,可以用調節圖片大小來控制
        int width = 350;
        int height = 250;

        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        // 背景透明 開始
        Graphics2D g = image.createGraphics();
        image = g.getDeviceConfiguration().createCompatibleImage(width, height, Transparency.TRANSLUCENT);
        // 背景透明 結束
        g = image.createGraphics();
        g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.5f));//設定水印透明度(0-1)之間
        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;
    }

    /**
     * 保護工作表以及設定保護選項
     */
    public static void protectWorksheet(XSSFWorkbook excel) {
        for (int i = 0; i < excel.getNumberOfSheets(); i++) {
            XSSFSheet sheet = excel.getSheetAt(i);
            // 保護工作表並設定密碼
            sheet.protectSheet("8246");
            // 獲取CTSheetProtection物件以進行更精細的控制
            CTSheetProtection protection = sheet.getCTWorksheet().getSheetProtection();
            // 設定保護選項(例如,允許使用者選擇未鎖定的單元格)
            protection.setSheet(true);//控制是否保護整個工作表。預設為false,設定為true`後,使用者不能修改工作表內容,除非解除了保護
            protection.setObjects(true);//如果為true,則禁止插入、刪除或移動圖表、圖片等物件
            protection.setFormatCells(true);//是否允許使用者格式化單元格
            protection.setFormatColumns(true);//是否允許使用者格式化列
            protection.setFormatRows(true);//是否允許使用者格式化行
            protection.setInsertColumns(true);//是否允許使用者插入列
            protection.setInsertRows(true);//是否允許使用者插入行
            protection.setDeleteColumns(true);//是否允許使用者刪除列
            protection.setDeleteRows(true);//是否允許使用者刪除行
            protection.setSelectLockedCells(true);//如果為false,則不允許選擇鎖定的單元格,即使整體工作表保護已被解除
            protection.setSelectUnlockedCells(true);//如果為false,則不允許選擇未鎖定的單元格,即使整體工作表保護已被解除
            protection.setSort(true);//如果設定為true,允許使用者對受保護的工作表進行排序操作
            protection.setAutoFilter(true);//如果設定為true,允許使用者在受保護的工作表上應用或編輯自動過濾器。
            protection.setPivotTables(true);//如果設定為true,允許使用者在受保護的工作表上建立、修改或刪除透視表
            protection.setInsertHyperlinks(true);//控制是否允許插入超連結
        }
    }

}
public static void main(String[] args) throws IOException {
        XSSFWorkbook workbook = new XSSFWorkbook();
        XSSFSheet sheet = workbook.createSheet("Sheet1");
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        String waterMarkText = "這是個水印\n" + dateFormat.format(new Date());
        WaterMarkUtil.insertWaterMarkTextToXlsx(workbook, waterMarkText);
        try (FileOutputStream fileOut = new FileOutputStream("watermark_example.xlsx")) {
            workbook.write(fileOut);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

相關文章