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

导出pdf加水印

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

public voidcreatePDF(String filename) throws IOException {

Document document= newDocument(PageSize.A4);try{

PdfWriter writer= PdfWriter.getInstance(document, newFileOutputStream(filename));

document.addTitle("example of PDF");

document.open();

PdfPTable table=createTable(writer);

document.add(table);//加入水印

PdfContentByte waterMar =writer.getDirectContentUnder();//开始设置水印

waterMar.beginText();//设置水印透明度

PdfGState gs = newPdfGState();//设置填充字体不透明度为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,30);//设置透明度

waterMar.setGState(gs);//设置水印对齐方式 水印内容 X坐标 Y坐标 旋转角度

waterMar.showTextAligned(Element.ALIGN_RIGHT, "我是水印" , 350, 730, 45);//结束设置

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 classExcelUtilTest{public void createExcel(HttpServletResponse response)throwsException {//获取excel文件

File finalXlsxFile = new File("src/main/resources/watermark.xlsx");//获取excel文件流

FileInputStream inputStream = newFileInputStream(finalXlsxFile);//根据文件流创建XSSFWorkbook对象

XSSFWorkbook wb = newXSSFWorkbook(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 = newXSSFWorkbook(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文件加水印的需求。