java实现导出excel添加水印或excel转pdf并添加水印
1.通过Map对象存入excel模板中获得excel表，并添加水印
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
public static void exportExcelWatermark(TemplateExportParams templatePath, Map<String, Object> map, String fileName, String markText, HttpServletResponse response){
XSSFWorkbook workbook = (XSSFWorkbook) ExcelExportUtil.exportExcel(templatePath, map);
ExcelMark.Watermark watermark = new ExcelMark.Watermark();
//设置水印文字
watermark.setText(markText);
BufferedImage image = ExcelMark.createWatermarkImage(watermark);
// 导出到字节流byte
ByteArrayOutputStream os = new ByteArrayOutputStream();
ImageIO.write(image, "png", os);
int pictureIdx = workbook.addPicture(os.toByteArray(), Workbook.PICTURE_TYPE_PNG);
POIXMLDocumentPart poixmlDocumentPart = workbook.getAllPictures().get(pictureIdx);
XSSFSheet sheet = workbook.getSheetAt(0);
PackagePartName ppn = poixmlDocumentPart.getPackagePart().getPartName();
String relType = XSSFRelation.IMAGES.getRelation();
//add relation from sheet to the picture data
PackageRelationship pr = sheet.getPackagePart().addRelationship(ppn, TargetMode.INTERNAL, relType, null);
//set background picture to sheet
sheet.getCTWorksheet().addNewPicture().setId(pr.getId());
downLoadExcel(fileName, response, workbook);
}
/**
* excel下载
*
* @param fileName 下载时的文件名称
* @param response
* @param workbook excel数据
*/
private static void downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) throws IOException {
try {
response.setCharacterEncoding("UTF-8");
response.setHeader("content-Type", "application/vnd.ms-excel");
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName + ".xlsx", "UTF-8"));
workbook.write(response.getOutputStream());
} catch (Exception e) {
throw new IOException(e.getMessage());
}
}
/**
*添加水印方法
*/
@Data
public static class Watermark {
private Boolean enable;
private String text;
private String dateFormat;
private String color;
}
public static BufferedImage createWatermarkImage(Watermark watermark) {
if (watermark == null) {
watermark = new ExcelMark.Watermark();
watermark.setEnable(true);
// watermark.setText("userName");
watermark.setText("水印水印水印");
watermark.setColor("#C5CBCF");
watermark.setDateFormat("yyyy-MM-dd HH:mm");
} else {
if (StringUtils.isEmpty(watermark.getDateFormat())) {
watermark.setDateFormat("yyyy-MM-dd HH:mm");
} else if (watermark.getDateFormat().length() == 16) {
watermark.setDateFormat("yyyy-MM-dd HH:mm");
} else if (watermark.getDateFormat().length() == 10) {
watermark.setDateFormat("yyyy-MM-dd");
}
if (StringUtils.isEmpty(watermark.getText())) {
watermark.setText("水印水印水印");
}
if (StringUtils.isEmpty(watermark.getColor())) {
watermark.setColor("#C5CBCF");
}
}
String[] textArray = watermark.getText().split("\n");
Font font = new Font("microsoft-yahei", Font.PLAIN, 20);
Integer width = 500;
Integer height = 200;
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
// 背景透明 开始
Graphics2D g = image.createGraphics();
image = g.getDeviceConfiguration().createCompatibleImage(width, height, Transparency.TRANSLUCENT);
g.dispose();
// 背景透明 结束
g = image.createGraphics();
g.setColor(new Color(Integer.parseInt(watermark.getColor().substring(1), 16)));// 设定画笔颜色
g.setFont(font);// 设置画笔字体
g.shear(0.1, -0.26);// 设定倾斜度
// 设置字体平滑
g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
int y = 100;
for (int i = 0; i < textArray.length; i++) {
g.drawString(textArray[i], 0, y);// 画出字符串
y = y + font.getSize();
}
g.drawString(DateUtils.dateTimeNow(watermark.getDateFormat()), 0, y);// 画出字符串
g.dispose();// 释放画笔
return image;
}
2.将excel转pdf并添加水印（先给excel添加水印会导致导出为pdf后消失）
在pom.xml中引入
<dependency>
<groupId>e-iceblue</groupId>
<artifactId>spire.xls.free</artifactId>
<version>2.2.0</version>
</dependency>
<dependency>
<groupId>com.itextpdf</groupId>
<artifactId>itextpdf</artifactId>
<version>5.4.3</version>
</dependency>
<dependency>
<groupId>com.itextpdf</groupId>
<artifactId>itext-asian</artifactId>
<version>5.2.0</version>
</dependency>
public static void exportExcelWatermark(TemplateExportParams templatePath, Map<String, Object> map, String fileName, String markText, HttpServletResponse response) throws Exception {
XSSFWorkbook workbook = (XSSFWorkbook) ExcelExportUtil.exportExcel(templatePath, map);
File tempFile = File.createTempFile(fileName, ".xlsx");
try (FileOutputStream out = new FileOutputStream(tempFile)) {
workbook.write(out);
}
//加载Excel工作表
com.spire.xls.Workbook wb = new com.spire.xls.Workbook();
wb.loadFromFile(tempFile.getAbsolutePath());
Worksheet excelSheet = wb.getWorksheets().get(0);
File tempPdfFile = File.createTempFile(fileName, ".pdf");
wb.getConverterSetting().setSheetFitToPage(true);
excelSheet.saveToPdf(tempPdfFile.getAbsolutePath());
// 添加水印
setPdfWaterMarForFileTest(tempPdfFile.getAbsolutePath(),fileName, markText, response);
// 步骤 7: 清理临时文件
tempPdfFile.delete();
}
public static void setPdfWaterMarForFileTest(String inputFile, String fileName, String waterText, HttpServletResponse response) throws Exception {
// 创建PdfReader对象读取原始PDF文件
PdfReader reader = new PdfReader(inputFile);
// 使用 ByteArrayOutputStream 来避免创建临时文件
try (ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream()) {
PdfStamper stamper = new PdfStamper(reader, byteArrayOutputStream);
PdfContentByte waterMar;
//试了很多字体，这个是itextpdf自带的支持中文的字体
BaseFont base = BaseFont.createFont("STSongStd-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
PdfGState gs = new PdfGState();
gs.setFillOpacity(0.3f);
gs.setStrokeOpacity(0.3f);
int n = reader.getNumberOfPages();
for (int i = 1; i <= n; i++) {
waterMar = stamper.getOverContent(i);
waterMar.setGState(gs);
waterMar.setFontAndSize(base, 12);
com.itextpdf.text.Rectangle pageSizeWithRotation = reader.getPageSizeWithRotation(i);
float height = pageSizeWithRotation.getHeight();
float width = pageSizeWithRotation.getWidth();
String[] marTextArray = stringToStringArray(waterText, 15);
waterMar.beginText();
//打印的行数和列数可自行修改
//135为横向偏移量，可以自行修改
for (int y = 0; y < 8; y++) {
for (int x = 0; x < 7; x++) {
for (int t = 0; t < marTextArray.length; t++) {
float xFloat = x * 190 + 135 + t * 10;
float yFloat = y * 100 + 87 - t * 10;
waterMar.showTextAligned(Element.ALIGN_CENTER, marTextArray[t], xFloat, yFloat, 30);
}
}
}
waterMar.endText();
}
stamper.close();
reader.close();
// 设置响应头以便下载 PDF
response.reset();
response.setContentType("application/pdf");
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName+".pdf", "UTF-8"));
// 将生成的 PDF 写入响应输出流
try (OutputStream out = response.getOutputStream()) {
byteArrayOutputStream.writeTo(out);
out.flush();
}
} catch (Exception e) {
throw new RuntimeException("pdf生成水印失败"); // 可以根据需求进行异常处理
}
}
public static String[] stringToStringArray(String text, int length) {
//检查参数是否合法
if (StringUtils.isEmpty(text)) {
return null;
}
if (length <= 0) {
return null;
}
// 首尾空格去除
text = text.trim();
//获取整个字符串可以被切割成字符子串的个数
int n = (text.length() + length - 1) / length;
String[] splitArr = new String[n];
for (int i = 0; i < n; i++) {
if (i < (n - 1)) {
splitArr[i] = text.substring(i * length, (i + 1) * length);
} else {
splitArr[i] = text.substring(i * length);
}
}
return splitArr;
}
