springboot爲導出的pdf和excel加水印

相信不少小夥伴們在作導出pdf或者excel文件時會被要求在文件上加上水印，本篇博客就來說講如何爲pdf和excel加水印。java

導出pdf加水印

其實在導出pdf時加上水印並不難，由於itext提供了添加水印的方法，並且能設置水印的位置角度等等，直接來看一下代碼spring

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,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表格，這在以前的博客中已經提過了。springboot

導出excel加水印

接下來咱們着重看一下excel文件怎麼加水印，如今業界對excel文件讀寫大多采用poi這個東西，而poi沒有提供添加水印的方法，因此要怎麼辦呢，我在網上搜索了很久也沒找到能直接拿來用的辦法，可是看到有人提供思路，先準備一份打了水印的模版Excel，而後加載該模版，再將內容輸出到該模版中，以達到爲Excel添加水印的目的。這個思路看起來的確可行，而後就開幹了，先看代碼app

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文件上的水印這個網上教程不少，隨便找一個來看就行，我就不贅述了，看一下效果如何spa

這是原始excel設計

這是生成後的excel3d

若是要將其打包部署的話，不能直接根據文件來獲取inputStream，須要將代碼改一改，將最上面兩句代碼改爲springboot提供的獲取類路徑下文件流的方法，而後文件文件要放在類路徑下。固然，這個在打包全部須要經過文件路徑來獲取文件的springboot項目狀況都適用。excel

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

總結

相對pdf生成水印的方法，excel這個方法仍是有不少不足之處的，好比生成水印的位置不能經過代碼更改，若是有多個sheet不能全都加上去（除非事先知道有幾個sheet），生成後的excel中的水印能夠隨意拖動位置（這個貌似能夠改進）。總之這份代碼是很是簡陋的，只能解決最簡單的爲導出excel文件加水印的需求。