import com.alibaba.excel.write.handler.SheetWriteHandler;
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;
import lombok.RequiredArgsConstructor;
import lombok.SneakyThrows;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFRelation;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.font.FontRenderContext;
import java.awt.geom.Rectangle2D;
import java.awt.image.BufferedImage;
import java.io.BufferedInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
@RequiredArgsConstructor
public class WaterMarkHandler implements SheetWriteHandler {
private final String WATER_MARK;
public static ByteArrayOutputStream createWaterMark(String content) throws Exception {
int width = 300;
int height = 200;
int fontStyle = Font.PLAIN;
int fontSize = 12;
// 添加中文字体
Resource resource = new ClassPathResource("font/msyh.ttf");
InputStream fi = resource.getInputStream();
BufferedInputStream fb = new BufferedInputStream(fi);
Font font = Font.createFont(fontStyle, fb);
font = font.deriveFont(fontStyle, fontSize);
// 获取bufferedImage对象
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
// 获取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, 20));
// 设置字体
g2d.setStroke(new BasicStroke(1));
// 设置字体类型 加粗 大小
g2d.setFont(font);
//设置倾斜度
g2d.rotate(-0.5, (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();
ByteArrayOutputStream os = new ByteArrayOutputStream();
ImageIO.write(image, "png", os);
return os;
}
public static void putWaterRemarkToExcel(XSSFSheet sheet, byte[] bytes) {
//add relation from sheet to the picture data
XSSFWorkbook workbook = sheet.getWorkbook();
int pictureIdx = workbook.addPicture(bytes, Workbook.PICTURE_TYPE_PNG);
String rID = sheet.addRelation(null, XSSFRelation.IMAGES, workbook.getAllPictures().get(pictureIdx))
.getRelationship().getId();
//set background picture to sheet
sheet.getCTWorksheet().addNewPicture().setId(rID);
}
@Override
public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
}
@SneakyThrows
@Override
public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
try (ByteArrayOutputStream waterMark = createWaterMark(WATER_MARK)) {
XSSFSheet sheet = (XSSFSheet) writeSheetHolder.getSheet();
putWaterRemarkToExcel(sheet, waterMark.toByteArray());
}
}
}
配置SheetWriteHandler
ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream())
.inMemory(true)
// 添加水印处理器
.registerWriteHandler(new WaterMarkHandler("名字"+ DateTime.now().toString(" yyyy-MM-dd HH:mm") + " 内部资料禁止外传"))
.build();
//这里 需要指定写用哪个class去写
List list;
WriteSheet writeSheet = EasyExcel.writerSheet(0, "sheet1").head(XXXDTO.class).build();
excelWriter.write(list, writeSheet);
writeSheet = EasyExcel.writerSheet(1, "sheet2").head(XXXDTO.class).build();
excelWriter.write(list, writeSheet);
writeSheet = EasyExcel.writerSheet(2, "sheet3").head(XXXDTO.class).build();
excelWriter.write(list, writeSheet);
//关闭流
excelWriter.finish();
核心是配置:
ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream())
.inMemory(true)
// 添加水印处理器
.registerWriteHandler(new WaterMarkHandler("名字"+ DateTime.now().toString(" yyyy-MM-dd HH:mm") + " 内部资料禁止外传"))
.build();
解决中文乱码
java.awt.FontFormatException: bad pfa font
- 将字体文件放到resources/font/目录下面
- 加载字体资源文件
src/main/resources true ***.xml ***.ttf src/main/resources false **/*.ttf
我使用这种方式失败了,但是还是在这里记录下:
org.apache.maven.plugins maven-resources-plugin 2.5 ttf xlsx xls zip cer pfx py



