springboot为导出的pdf和excel加水印

相信很多小伙伴们在做导出pdf或者excel文件时会被要求在文件上加上水印，本篇博客就来讲讲如何为pdf和excel加水印。

导出pdf加水印

其实在导出pdf时加上水印并不难，因为itext提供了添加水印的方法，而且能设置水印的位置角度等等，直接来看一下代码

public void createPDF(String filename) throws IOException {

    Document document = new Document(PageSize.A4);

    try {

        PdfWriter writer = PdfWriter.getInstance(document, new FileOutputStream(filename));

        document.addTitle("example of PDF");

        document.open();

        PdfPTable table = createTable(writer);

        document.add(table);

        // 加入水印

        PdfContentByte waterMar = writer.getDirectContentUnder();

        // 开始设置水印

        waterMar.beginText();

        // 设置水印透明度

        PdfGState gs = new PdfGState();

        // 设置填充字体不透明度为0.2f

        gs.setFillOpacity(0.2f);

        // 设置水印字体参数及大小 (这里在上一篇博客中已经讲过了)

        BaseFont baseFont = BaseFont.createFont(JavaPdfHelloWorld.class.getResource("/simsun.ttf").getPath(), BaseFont.IDENTITY_H,BaseFont.NOT_EMBEDDED);

        waterMar.setFontAndSize(baseFont,);

        // 设置透明度

        waterMar.setGState(gs);

        // 设置水印对齐方式 水印内容 X坐标 Y坐标 旋转角度

        waterMar.showTextAligned(Element.ALIGN_RIGHT, "我是水印" , , , );

        //结束设置

        waterMar.endText();

        waterMar.stroke();

    } catch (FileNotFoundException e) {

        e.printStackTrace();

    } catch (DocumentException e) {

        e.printStackTrace();

    } finally {

        document.close();

    }

}

createTable（）方法的代码就不贴了，就是生成一个pdf表格，这在之前的博客中已经提过了。

导出excel加水印

接下来我们着重看一下excel文件怎么加水印，现在业界对excel文件读写大多采用poi这个东西，而poi没有提供添加水印的方法，所以要怎么办呢，我在网上搜索了好久也没找到能直接拿来用的办法，但是看到有人提供思路，先准备一份打了水印的模版Excel，然后加载该模版，再将内容输出到该模版中，以达到为Excel添加水印的目的。这个思路看起来的确可行，然后就开干了，先看代码

public class ExcelUtilTest{

  public void createExcel(HttpServletResponse response)throws Exception {

    //获取excel文件

    File finalXlsxFile = new File("src/main/resources/watermark.xlsx");

    //获取excel文件流

    FileInputStream inputStream = new FileInputStream(finalXlsxFile);

    //根据文件流创建XSSFWorkbook对象

    XSSFWorkbook wb = new XSSFWorkbook(inputStream);

    XSSFSheet sheet = wb.getSheetAt(0);

    sheet.setDefaultColumnWidth(16);//设置默认列宽

    sheet.setDefaultRowHeightInPoints(20);//设置默认行高

    for(int j = 0;j < 5;j++) {

    XSSFRow row = sheet.createRow(j);

    for(int i = 0;i<5;i++) {

    XSSFCell cell=row.createCell(i);

    cell.setCellValue("第"+j+"行第"+i+"列");

      }

    }

    //输出Excel文件

    OutputStream output= response.getOutputStream();

    response.reset();

    String  fileName = "水印测试文件";

    response.setHeader("Content-Disposition",

      "attachment; filename=" +new String(fileName.getBytes("UTF-8"), "iso-8859-1")+".xlsx");

    response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");

    wb.write(output);

    output.close();

  }

}

这个代码换一下文件地址可以直接拿来用，值得注意的是这是用springboot以附件形式导出的，不是用java程序生成的，也就是说我还有一个controller文件，只是没贴出来。

和我们直接生成excel文件代码有所不同的是下面两句代码，如果是直接生成excel文件这构造方法参数是空的，然后getSheetAt()方法改为createSheet()方法即可

XSSFWorkbook wb = new XSSFWorkbook(inputStream); 
XSSFSheet sheet = wb.getSheetAt(0);

至于如何设计excel文件上的水印这个网上教程很多，随便找一个来看就行，我就不赘述了，看一下效果如何

这是原始excel

这是生成后的excel

如果要将其打包部署的话，不能直接根据文件来获取inputStream，需要将代码改一改，将最上面两句代码改成springboot提供的获取类路径下文件流的方法，然后文件文件要放在类路径下。当然，这个在打包所有需要通过文件路径来获取文件的springboot项目情况都适用。

InputStream finalXlsxFile = new ClassPathResource("watermark.xlsx").getInputStream();

总结

相对pdf生成水印的方法，excel这个方法还是有很多不足之处的，比如生成水印的位置不能通过代码更改，如果有多个sheet不能全都加上去（除非事先知道有几个sheet），生成后的excel中的水印可以随意拖动位置（这个貌似可以改进）。总之这份代码是非常简陋的，只能解决最简单的为导出excel文件加水印的需求。