• springboot生成PDF,并且添加水印

    1.  
    2. 	/**
    3. 	 *  导出调查问卷
    4. 	 */
    5. 	@ApiLog("导出调查问卷")
    6. 	@PostMapping("/print/{id}")
    7. 	@ApiOperationSupport(order = 23)
    8. 	@ApiOperation(value = "导出报告", notes = "导出报告")
    9. 	public void print(@PathVariable Long id, HttpServletResponse response) {
    10. 		BackendGradeEntity record = backendGradeService.getById(id);
    11.  
    12. 		//如果record为空，直接返回错误信息
    13. 		if(record == null){
    14. 			throw new RuntimeException("未找到该记录");
    15. 		}
    16.  
    17. 		Map values = new HashMap<>();
    18. 		Field[] fields = BackendGradeEntity.class.getDeclaredFields();
    19. 		//通过反射拿到对象的属性名并且赋值给map
    20. 		for (Field field : fields) {
    21. 			field.setAccessible(true);
    22. 			try {
    23. 				Object value = field.get(record);
    24. 				values.put(field.getName(), value);
    25. 			} catch (IllegalAccessException e) {
    26. 				// 处理访问异常
    27. 				e.printStackTrace();
    28. 			}
    29. 		}
    30.  
    31. 		//通过用户id查询用户信息
    32. 		BackendUserinformationEntity user = backendUserinformationService.getById(record.getUserId());
    33. 		//获取用户性别
    34. 		Integer sex = null;
    35. 		if (user!= null){
    36. 			sex = user.getSex();
    37. 		}
    38.  
    39. 		//添加用户性别
    40. 		values.put("sex", sex==1?"男":"女");
    41.  
    42. 		String totalTime = values.get("totalTime").toString();
    43. 		double timeInSeconds = Double.parseDouble(totalTime) * 60; // 将分钟转换为秒
    44. 		int minutes = (int) timeInSeconds / 60;
    45. 		int seconds = (int) timeInSeconds % 60;
    46. 		String formattedTime = minutes + "分钟" + seconds + "秒";
    47. 		values.put("totalTime", formattedTime);
    48.  
    49. 		//增加打印日期为当前日期
    50. 		values.put("printDate", DateUtil.format(LocalDate.now(), "yyyy年MM月dd日"));
    51. 		//修改map中的startTime 和 endTime 格式由2023-09-19T15:34:25  为 2023-09-19 15:34:25
    52. 		String startTime = values.get("startTime").toString().replace("T", " ");
    53. 		String endTime = values.get("endTime").toString().replace("T", " ");
    54. 		values.put("startTime", startTime);
    55. 		values.put("endTime", endTime);
    56.  
    57. 		//将 正确率 错误率 未答题率 乘100再填回，因为数据存的是小数
    58. 		values.put("errorRate", Double.parseDouble(values.get("errorRate").toString())*100);
    59. 		values.put("accuracy", Double.parseDouble(values.get("accuracy").toString())*100);
    60. 		values.put("unansweredRate", Double.parseDouble(values.get("unansweredRate").toString())*100);
    61.  
    62.  
    63.  
    64. 		String fileName = null;
    65. 		String tplName = null;
    66. 		if(true){
    67. 			fileName = "报告";
    68. 			tplName = "intuitionReport.ftl";
    69. 		}
    70. 		fileName = fileName + "-" + (values.get("userName")==null?UUID.randomUUID():values.get("userName"));
    71. 		String file = printer.print(values, tplName, fileName);
    72.  
    73. 		//下载文件
    74. 		InputStream inStream = null;
    75. 		try {
    76. 			inStream = new FileInputStream(file);
    77. 			response.reset();
    78. 			response.setContentType("application/pdf");
    79. 			response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(new File(file).getName(), "UTF-8"));
    80. 			response.setCharacterEncoding("UTF-8");
    81. 			IoUtil.copy(inStream, response.getOutputStream());
    82. 		} catch (Exception e) {
    83. 			e.printStackTrace();
    84. 		} finally {
    85. 			if (inStream != null) {
    86. 				try { inStream.close(); } catch (IOException e) { e.printStackTrace(); }
    87. 			}
    88. 		}
    89. 	}
    90.  
    91.  
    92. 	/**
    93. 	 *  批量导出报告
    94. 	 */
    95. 	@ApiLog("批量导出报告")
    96. 	@PostMapping("/print/batch")
    97. 	@ApiOperationSupport(order = 23)
    98. 	@ApiOperation(value = "批量导出", notes = "批量导出报告")
    99. 	public void printBatch(@RequestParam String ids, HttpServletResponse response) {
    100. 		List records = backendGradeService.listByIds(Func.toLongList(ids));
    101.  
    102. 		String uuid = UUID.randomUUID().toString();
    103.  
    104. 		int size = records.size();
    105. 		DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyyMMddHHmmss");
    106. 		String zipName = LocalDateTime.now().format(formatter)+"-"+size+"records.zip";
    107.  
    108. 		Path zipPath = Paths.get(pathProperties.getPdf(), zipName);
    109. 		File zipFile = zipPath.toFile();
    110. 		if(zipFile.exists()){
    111. 			zipFile.delete();//如果文件存在则先删除旧的文件
    112. 		}
    113.  
    114. 		List files = null;
    115. 		try{
    116. 			if(!zipFile.getParentFile().exists()){
    117. 				Files.createDirectories(zipPath.getParent());
    118. 			}
    119.  
    120. 			files = records.stream()
    121. //				.filter(j -> j.getStatus() == 4)
    122. 				.map(record -> {
    123.  
    124. 					Map values = new HashMap<>();
    125. 					Field[] fields = BackendGradeEntity.class.getDeclaredFields();
    126. 					//通过反射拿到对象的属性名并且赋值给map
    127. 					for (Field field : fields) {
    128. 						field.setAccessible(true);
    129. 						try {
    130. 							Object value = field.get(record);
    131. 							values.put(field.getName(), value);
    132. 						} catch (IllegalAccessException e) {
    133. 							// 处理访问异常
    134. 							e.printStackTrace();
    135. 						}
    136. 					}
    137.  
    138.  
    139. 					//通过用户id查询用户信息
    140. 					BackendUserinformationEntity user = backendUserinformationService.getById(record.getUserId());
    141. 					//获取用户性别
    142. 					Integer sex = null;
    143. 					if (user!= null){
    144. 						sex = user.getSex();
    145. 					}
    146.  
    147. 					//添加用户性别
    148. 					values.put("sex", sex==1?"男":"女");
    149.  
    150. 					//处理总时长
    151. 					String totalTime = values.get("totalTime").toString();
    152. 					double timeInSeconds = Double.parseDouble(totalTime) * 60; // 将分钟转换为秒
    153. 					int minutes = (int) timeInSeconds / 60;
    154. 					int seconds = (int) timeInSeconds % 60;
    155. 					String formattedTime = minutes + "分钟" + seconds + "秒";
    156. 					values.put("totalTime", formattedTime);
    157.  
    158. 					//增加打印日期为当前日期
    159. 					values.put("printDate", DateUtil.format(LocalDate.now(), "yyyy年MM月dd日"));
    160. 					//修改map中的startTime 和 endTime 格式由2023-09-19T15:34:25  为 2023-09-19 15:34:25
    161. 					String startTime = values.get("startTime").toString().replace("T", " ");
    162. 					String endTime = values.get("endTime").toString().replace("T", " ");
    163. 					values.put("startTime", startTime);
    164. 					values.put("endTime", endTime);
    165.  
    166.  
    167. 					//将 正确率 错误率 未答题率 乘100再填回，因为数据存的是小数
    168. 					values.put("errorRate", Double.parseDouble(values.get("errorRate").toString())*100);
    169. 					values.put("accuracy", Double.parseDouble(values.get("accuracy").toString())*100);
    170. 					values.put("unansweredRate", Double.parseDouble(values.get("unansweredRate").toString())*100);
    171.  
    172.  
    173. 					String fileName = null;
    174. 					String tplName = null;
    175. 					if(true){
    176. 						fileName = "报告";
    177. 						tplName = "intuitionReport.ftl";
    178. 					}
    179. 					fileName = fileName + "-" + (values.get("userName")==null?UUID.randomUUID():values.get("userName")+UUID.randomUUID().toString());
    180. 					String f = printer.print(values, tplName, uuid+"/"+fileName);
    181. 					return new File(f);
    182. 				}).collect(Collectors.toList());
    183.  
    184.  
    185. //			Path tempDir = Files.createTempDirectory("temp");
    186. //			List copiedFiles = new ArrayList<>();
    187. //			for (File file : files) {
    188. //				Path source = Paths.get(file.getPath());
    189. //				Path destination = tempDir.resolve(file.getName());
    190. //				Files.copy(source, destination);
    191. //				copiedFiles.add(destination.toFile());
    192. //			}
    193. //			// 在这里调用添加水印的方法
    194. //			PDFWatermarkExample.addWatermarkExample(copiedFiles);
    195. //
    196. //			files = copiedFiles;
    197. //			try {
    198. //				// 删除临时文件夹及其所有文件
    199. //				FileUtils.deleteDirectory(tempDir.toFile());
    200. //			} catch (IOException e) {
    201. //				// 处理删除错误
    202. //				e.printStackTrace();
    203. //			}
    204.  
    205. 			ZipTool.zipFile(files, zipPath.toFile().getAbsolutePath());
    206. 		}catch(Exception e){
    207. 			e.printStackTrace();
    208. 		}finally {
    209. 			if(files != null){
    210. 				for(File f : files){
    211. 					if(f.exists()) f.delete();
    212. 				}
    213. 			}
    214. 			Path dirPath = Paths.get(pathProperties.getPdf(), uuid);
    215. 			if(dirPath.toFile().exists()){
    216. 				dirPath.toFile().delete();
    217. 			}
    218. 		}
    219.  
    220. 		//下载文件
    221. 		InputStream inStream = null;
    222. 		try {
    223. 			if(!zipFile.exists()){
    224. 				return ;
    225. 			}
    226. 			inStream = new FileInputStream(zipFile);
    227. 			response.reset();
    228. 			response.setContentType("application/zip");
    229. 			response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(zipFile.getName(), "UTF-8"));
    230. 			response.setCharacterEncoding("UTF-8");
    231. 			IoUtil.copy(inStream, response.getOutputStream());
    232. 		} catch (Exception e) {
    233. 			e.printStackTrace();
    234. 		} finally {
    235. 			if (inStream != null) {
    236. 				try { inStream.close(); } catch (IOException e) { e.printStackTrace(); }
    237. 			}
    238. 		}
    239.  
    240. 		//这里删除临时文件夹内的压缩包，因为存着也没什么用浪费空间
    241. 		//通过zipPath获取绝对路径
    242. 		String absolutePath = zipPath.toFile().getAbsolutePath();
    243. 		//删除absolutepath文件夹，以及所有文件
    244. 		deleteDirectoryRecursively(new File(absolutePath));
    245.  
    246.  
    247. 	}

    下面是加水印

    1. package org.springblade.common.tool;
    2. import com.itextpdf.text.*;
    3. import com.itextpdf.text.pdf.*;
    4.  
    5. import java.io.File;
    6. import java.io.FileOutputStream;
    7. import java.util.List;
    8.  
    9. public class PDFWatermarkExample {
    10. 	public static void addWatermark(String inputFile, String outputFile, String watermarkText) {
    11. 		try {
    12. 			PdfReader reader = new PdfReader(inputFile);
    13. 			PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(outputFile));
    14. 			int numberOfPages = reader.getNumberOfPages();
    15.  
    16. 			for (int i = 1; i <= numberOfPages; i++) {
    17. 				PdfContentByte content = stamper.getUnderContent(i);
    18. 				PdfGState gs = new PdfGState();
    19. 				gs.setFillOpacity(0.5f); // 设置水印透明度
    20. 				content.setGState(gs);
    21.  
    22. 				ColumnText.showTextAligned(
    23. 					content,
    24. 					Element.ALIGN_CENTER,
    25. 					new Phrase(watermarkText, new Font(Font.FontFamily.HELVETICA, 40)),
    26. 					reader.getPageSizeWithRotation(i).getWidth() / 2,
    27. 					reader.getPageSizeWithRotation(i).getHeight() / 2,
    28. 					45
    29. 				);
    30. 			}
    31.  
    32. 			stamper.close();
    33. 			reader.close();
    34. 		} catch (Exception e) {
    35. 			e.printStackTrace();
    36. 		}
    37. 	}
    38.  
    39. 	public static void addWatermarkMulti(String inputFile, String outputFile, String watermarkText) {
    40. 		try {
    41. 			PdfReader reader = new PdfReader(inputFile);
    42. 			PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(outputFile));
    43. 			int numberOfPages = reader.getNumberOfPages();
    44.  
    45. 			for (int i = 1; i <= numberOfPages; i++) {
    46. 				PdfContentByte content = stamper.getOverContent(i);
    47. 				PdfGState gs = new PdfGState();
    48. 				gs.setFillOpacity(0.05f); // 设置水印透明度
    49. 				content.setGState(gs);
    50.  
    51. 				Rectangle pageSize = reader.getPageSizeWithRotation(i);
    52. 				float pageWidth = pageSize.getWidth();
    53. 				float pageHeight = pageSize.getHeight();
    54.  
    55. 				// 设置水印间隔
    56. 				float xInterval = 200; // X轴间隔
    57. 				float yInterval = 50; // Y轴间隔
    58.  
    59. 				// 计算水印个数
    60. 				int xCount = (int) Math.ceil(pageWidth / xInterval);
    61. 				int yCount = (int) Math.ceil(pageHeight / yInterval);
    62.  
    63. 				// 平铺水印
    64. 				for (int x = 0; x < xCount; x++) {
    65. 					for (int y = 0; y < yCount; y++) {
    66. 						float xPosition = x * xInterval;
    67. 						float yPosition = y * yInterval;
    68.  
    69. 						ColumnText.showTextAligned(
    70. 							content,
    71. 							Element.ALIGN_CENTER,
    72. 							new Phrase(watermarkText, new Font(Font.FontFamily.HELVETICA, 40)),
    73. 							xPosition,
    74. 							yPosition,
    75. 							0
    76. 						);
    77. 					}
    78. 				}
    79. 			}
    80.  
    81. 			stamper.close();
    82. 			reader.close();
    83. 		} catch (Exception e) {
    84. 			e.printStackTrace();
    85. 		}
    86. 	}
    87.  
    88. 	public static void addWatermarkExample(List files) {
    89. 		// 在这里编写添加水印的代码逻辑，使用上面提到的添加水印的示例代码
    90. 		for (File file : files) {
    91. 			addWatermark(file.getPath(), file.getPath(), "Watermark Text");
    92. 		}
    93. 	}
    94.  
    95. 	public static void main(String[] args) {
    96. 		String inputFile = "C:\\Users\\admin\\Downloads\\123.pdf";
    97. 		String outputFile = "C:\\Users\\admin\\Downloads\\789.pdf";
    98. 		String watermarkText = "zhijue.com";
    99.  
    100. 		addWatermarkMulti(inputFile, outputFile, watermarkText);
    101. 	}
    102. }

  • 相关阅读:
    目标检测——YOLOv2算法解读
    迪拜推出国家元宇宙战略
    WRFV3.8.1编译报错，无法显示exe文件
    艾美捷FcyRl (CD64)，FCGR1A，生物活性活动分析结果
    Android 面经总结分享（相当走心）
    双目视觉实战---三维重建基础与极几何
    【吴恩达笔记】卷积神经网络
    不会代码的时候，如何使用Jmeter完成接口测试？
    不要在问了！工作六年总结的Java面试题与经验
    VBA技术资料MF50：VBA_在Excel中突出显示前3个值
  • 原文地址：https://blog.csdn.net/weixin_42759398/article/details/134442956