Java操作excel自动生成水印背景

操作基于spire的程序包，经代码及测试能够将导出的数据进行增加背景图片无问题，以下为代码：

代码看前先给个提醒吧，为啥要这么做呢是因为首先从操作上来说可以往excel中直接写入图片覆盖到数据层上面，但是这样并不严谨，肯定对用户操作有影响而且用户可以将水印图片删除掉，但是生成背景图片插入进去就不一样了，肯定不影响用户操作的前提下有了背景图，这时候除非去手动更换背景图不然不影响用户任何使用操作，看到过一堆写的乱七八糟的加水印的，其实就是往数据层上加了个图片，操作极为恶心根本没达到需求，还降低用户体验！

注：操作必须先生成水印图片并将水印图片添加至excel中，然后再将数据往excel中写入，否则在加入水印的时候会删除原excel文件并重新生成新的文件（原因是：因为懒没有去细看为啥会这样）。使用的包是 spire.xls.free-2.2.0.jar 做的测试POI包自己找一个吧。至于spire包网上搜的很多都收费，索性找到了弄个免费的链接吧，下载后自行解压使用。https://download.csdn.net/download/baidu_35051056/20401576

import com.spire.xls.ExcelVersion;
import com.spire.xls.Worksheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.Color;
import java.awt.font.FontRenderContext;
import java.awt.geom.Rectangle2D;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.*;
import java.util.List;

/**
 * 创建excel 并保存背景图片（水印）
 */
public class CreateExcelImg {
    private static final String XLS = "xls";
    private static final String XLSX = "xlsx";

    /**
     * 创建水印图片
     *
     * @param content
     * @return
     * @throws IOException
     */
    public static void createWaterMark(String content, String filePath) throws IOException {
        Integer width = 300;
        Integer height = 200;
        // 获取bufferedImage对象
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        String fontType = "宋体";
        Integer fontStyle = java.awt.Font.PLAIN;
        Integer fontSize = 50;
        java.awt.Font font = new java.awt.Font(fontType, fontStyle, fontSize);
        // 获取Graphics2d对象
        Graphics2D g2d = image.createGraphics();
        image = g2d.getDeviceConfiguration().createCompatibleImage(width, height, Transparency.TRANSLUCENT);
        g2d.dispose();
        g2d = image.createGraphics();
        g2d.setColor(new Color(0, 0, 0, 80)); //设置字体颜色和透明度
        // 设置字体
        g2d.setStroke(new BasicStroke(1));
        // 设置字体类型  加粗 大小
        g2d.setFont(font);
        //设置倾斜度
        g2d.rotate(Math.toRadians(-10), (double) image.getWidth() / 2, (double) image.getHeight() / 2);
        FontRenderContext context = g2d.getFontRenderContext();
        Rectangle2D bounds = font.getStringBounds(content, context);
        double x = (width - bounds.getWidth()) / 2;
        double y = (height - bounds.getHeight()) / 2;
        double ascent = -bounds.getY();
        double baseY = y + ascent;
        // 写入水印文字原定高度过小，所以累计写水印，增加高度
        g2d.drawString(content, (int) x, (int) baseY);
        // 设置透明度
        g2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER));
        // 释放对象
        g2d.dispose();
        ImageIO.write(image, "png", new File(filePath));
    }

    /**
     * 为Excel打上水印工具函数 请自行确保参数值，以保证水印图片之间不会覆盖。 在计算水印的位置的时候，并没有考虑到单元格合并的情况，请注意
     *
     * @param wb                Excel Workbook
     * @param sheet             需要打水印的Excel
     * @param waterRemarkPath   水印地址，classPath，目前只支持png格式的图片，
     *                          因为非png格式的图片打到Excel上后可能会有图片变红的问题，且不容易做出透明效果。
     *                          同时请注意传入的地址格式，应该为类似："\\excelTemplate\\test.png"
     * @param startXCol         水印起始列
     * @param startYRow         水印起始行
     * @param betweenXCol       水印横向之间间隔多少列
     * @param betweenYRow       水印纵向之间间隔多少行
     * @param XCount            横向共有水印多少个
     * @param YCount            纵向共有水印多少个
     * @param waterRemarkWidth  水印图片宽度为多少列
     * @param waterRemarkHeight 水印图片高度为多少行
     * @throws IOException
     */
    public static void putWaterRemarkToExcel(Workbook wb, Sheet sheet, String waterRemarkPath, int startXCol,
                                             int startYRow, int betweenXCol, int betweenYRow, int XCount, int YCount, int waterRemarkWidth,
                                             int waterRemarkHeight) throws IOException {

        // 校验传入的水印图片格式
        if (!waterRemarkPath.endsWith("png") && !waterRemarkPath.endsWith("PNG")) {
            throw new RuntimeException("向Excel上面打印水印，目前支持png格式的图片。");
        }

        // 加载图片
        ByteArrayOutputStream byteArrayOut = new ByteArrayOutputStream();
//        InputStream imageIn = new FileInputStream(waterRemarkPath);
        InputStream imageIn = Thread.currentThread().getContextClassLoader().getResourceAsStream(waterRemarkPath);
        if (null == imageIn || imageIn.available() < 1) {
            throw new RuntimeException("向Excel上面打印水印，读取水印图片失败(1)。");
        }
        BufferedImage bufferImg = ImageIO.read(imageIn);
        if (null == bufferImg) {
            throw new RuntimeException("向Excel上面打印水印，读取水印图片失败(2)。");
        }
        ImageIO.write(bufferImg, "png", byteArrayOut);

        // 开始打水印
        Drawing drawing = sheet.createDrawingPatriarch();

        // 按照共需打印多少行水印进行循环
        for (int yCount = 0; yCount < YCount; yCount++) {
            // 按照每行需要打印多少个水印进行循环
            for (int xCount = 0; xCount < XCount; xCount++) {
                // 创建水印图片位置
                int xIndexInteger = startXCol + (xCount * waterRemarkWidth) + (xCount * betweenXCol);
                int yIndexInteger = startYRow + (yCount * waterRemarkHeight) + (yCount * betweenYRow);
                /*
                 * 参数定义： 第一个参数是（x轴的开始节点）； 第二个参数是（是y轴的开始节点）； 第三个参数是（是x轴的结束节点）；
                 * 第四个参数是（是y轴的结束节点）； 第五个参数是（是从Excel的第几列开始插入图片，从0开始计数）；
                 * 第六个参数是（是从excel的第几行开始插入图片，从0开始计数）； 第七个参数是（图片宽度，共多少列）；
                 * 第8个参数是（图片高度，共多少行）；
                 */
                ClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, xIndexInteger,
                        yIndexInteger, xIndexInteger + waterRemarkWidth, yIndexInteger + waterRemarkHeight);

                Picture pic = drawing.createPicture(anchor,
                        wb.addPicture(byteArrayOut.toByteArray(), Workbook.PICTURE_TYPE_PNG));
                pic.resize();
            }
        }
    }


    /**
     * 写入Excel
     *
     * @param dataList      写入Excel 的数据
     * @param cloumn        总列数
     * @param finalXlsxPath 写入路径
     */
    private static void writeExcel(List<Map> dataList, int cloumn, String finalXlsxPath) {
        OutputStream out = null;
        try {
            File finalXlsxFile = new File(finalXlsxPath);
            Workbook workBook = getWorkbok(finalXlsxFile);
            // sheet 工作页
            Sheet sheet = workBook.getSheetAt(0);
            //获取Excel最后一行
            int rowNumber = sheet.getLastRowNum();
            //empty
            for (int i = 1; i <= rowNumber; i++) {
                Row row = sheet.getRow(i);
                sheet.removeRow(row);
            }
            for (int j = 0; j < dataList.size(); j++) {
                // 创建一行：从第二行开始，跳过属性列
                Row row = sheet.createRow(j);
                // 得到要插入的每一条记录
                Map dataMap = dataList.get(j);
                int k = 0;
                //轮询map中的所有key值并赋值
                for (Iterator iterator = dataMap.keySet().iterator(); iterator.hasNext(); ) {
                    //获取Set集合中每一个键值
                    String key = (String) iterator.next();
                    //根据键值取map中的键对应的值；
                    Object value = dataMap.get(key);
                    Cell cell = row.createCell(k);
                    cell.setCellValue(value.toString());
                    k++;
                }
            }
            // 创建文件输出流，准备输出电子表格：这个必须有，否则你在sheet上做的任何操作都不会有效
            out = new FileOutputStream(finalXlsxPath);
            workBook.write(out);
        } catch (Exception e) {
            System.out.println("error error" + e);
        } finally {
            try {
                if (out != null) {
                    out.flush();
                    out.close();
                    System.out.println("数据导出成功");
                }
            } catch (IOException e) {
                System.out.println("error error error" + e);
            }
        }
    }

    //判断Excel格式(在此有点鸡肋，当做功能拓展~_~)
    public static Workbook getWorkbok(File file) throws IOException {
        Workbook wb = null;
        FileInputStream in = new FileInputStream(file);
        if (file.getName().endsWith(XLS)) {     //Excel&nbsp;2003
            wb = new HSSFWorkbook(in);
        } else if (file.getName().endsWith(XLSX)) {    // Excel 2007/2010
            wb = new XSSFWorkbook(in);
        }
        return wb;
    }

    /**
     * 创建带用户名水印的excel并保存对应的从查询数据
     *
     * @param dataList 数据
     * @param size     列数量
     * @param userName 用户名
     * @param filePath 存放路径
     * @throws Exception
     */
    public static void createExcel(List<Map> dataList, int size, String userName, String filePath) throws Exception {
        String imgPath = filePath + "/" + userName + ".png";
        String excelFilePath = filePath + "/" + userName + ".xlsx";
        try {
            createWaterMark(userName, imgPath);
            com.spire.xls.Workbook workbook = new com.spire.xls.Workbook();
            Worksheet sheet1 = workbook.getWorksheets().get(0);
            BufferedImage image = ImageIO.read(new File(imgPath));
            sheet1.getPageSetup().setBackgoundImage(image);
            //保存文档
            workbook.saveToFile(excelFilePath, ExcelVersion.Version2007);
            workbook.dispose();
            File file = new File(imgPath);
            file.delete();
        } catch (Exception e) {
            System.out.println("error error error" + e);
        }
        writeExcel(dataList, size, excelFilePath);
    }

    public static void main(String[] args) throws Exception {
        List<Map> list = new ArrayList<>();
        for (int i = 0; i < 100000; i++) {
            Map map = new HashMap();
            map.put("id", 1 + "" + i);
            map.put("name", "1231" + "i");
            list.add(map);
        }
        CreateExcelImg.createExcel(list, 2, "test", "D:\\tt");
    }
}