1 package ***; 2 3 import java.io.FileInputStream; 4 import java.io.FileNotFoundException; 5 import java.io.FileOutputStream; 6 import java.util.ArrayList; 7 import java.util.HashMap; 8 import java.util.List; 9 import java.util.Map; 10 import java.util.Set; 11 12 import org.apache.poi.util.Units; 13 import org.apache.poi.xwpf.usermodel.XWPFDocument; 14 import org.apache.poi.xwpf.usermodel.XWPFParagraph; 15 import org.apache.poi.xwpf.usermodel.XWPFRun; 16 import org.apache.poi.xwpf.usermodel.XWPFTable; 17 import org.apache.poi.xwpf.usermodel.XWPFTableCell; 18 import org.apache.poi.xwpf.usermodel.XWPFTableRow; 19 20 /** 21 * 文件数据替换 22 * @author 23 * 24 */ 25 26 public class ReplaceWord { 27 28 public static String path = "模板文件的路径/test.docx"; 29 30 public static void main(String[] args) throws Exception { 31 Map<String, Object> data = new HashMap<>(); 32 Map<String, Object> pic = new HashMap<>(); 33 34 pic.put("${carpicture}", "Files/sign.png"); 35 pic.put("${licensecopy}", "Files/sign.png"); 36 data.put("${seal}", "1"); 37 pic.put("${sing01}", "Files/sign.png"); 38 pic.put("${sing02}", "Files/sign.png"); 39 System.out.println(pic.containsKey("${carpicture}")); 40 41 // 列表(List)是对象的有序集合 42 List<List<String[]>> tabledataList = new ArrayList<>(); 43 getWord(data, tabledataList, pic); 44 } 45 46 public static void getWord(Map<String, Object> data, List<List<String[]>> tabledataList, Map<String, Object> picmap) 47 throws Exception { 48 try (FileInputStream is = new FileInputStream(path); XWPFDocument document = new XWPFDocument(is)) { 49 // 替换掉表格之外的文本(仅限文本) 50 changeText(document, data); 51 52 // 替换表格内的文本对象 53 changeTableText(document, data); 54 55 // 替换图片 56 changePic(document, picmap); 57 58 // 替换表格内的图片对象 59 changeTablePic(document, picmap); 60 61 long time = System.currentTimeMillis();// 获取系统时间 62 System.out.println(time); // 打印时间 63 // 使用try和catch关键字捕获异常 64 try (FileOutputStream out = new FileOutputStream("生成的文件路径/文件名" + ".docx")) { 65 document.write(out); 66 } 67 } catch (FileNotFoundException e) { 68 e.printStackTrace(); 69 } 70 } 71 72 /** 73 * 替换段落文本 74 * @param document docx解析对象 75 * @param textMap 需要替换的信息集合 76 * 77 */ 78 public static void changeText(XWPFDocument document, Map<String, Object> textMap) { 79 // 获取段落集合 80 // 返回包含页眉或页脚文本的段落 81 List<XWPFParagraph> paragraphs = document.getParagraphs(); 82 // 增强型for循环语句,前面一个为声明语句,后一个为表达式 83 for (XWPFParagraph paragraph : paragraphs) { 84 // 判断此段落是否需要替换 85 String text = paragraph.getText();// 检索文档中的所有文本 86 if (checkText(text)) { 87 List<XWPFRun> runs = paragraph.getRuns(); 88 for (XWPFRun run : runs) { 89 // 替换模板原来位置 90 Object ob = changeValue(run.toString(), textMap); 91 if (ob instanceof String) { 92 if (textMap.containsKey(run.toString())) { 93 run.setText((String) ob, 0); 94 } 95 } 96 } 97 } 98 } 99 } 100 101 /* 检查文本中是否包含指定的字符(此处为“$”),并返回值 */ 102 public static boolean checkText(String text) { 103 boolean check = false; 104 if (text.contains("$")) { 105 check = true; 106 } 107 return check; 108 } 109 110 /** 111 * 替换图片 112 * @param document 113 * @param textMap 114 * @throws Exception 115 */ 116 117 public static void changePic(XWPFDocument document, Map<String, Object> textMap) throws Exception { 118 // 获取段落集合 119 List<XWPFParagraph> paragraphs = document.getParagraphs(); 120 121 for (XWPFParagraph paragraph : paragraphs) { 122 // 判断此段落是否需要替换 123 String text = paragraph.getText(); 124 if (checkText(text)) { 125 List<XWPFRun> runs = paragraph.getRuns(); 126 for (XWPFRun run : runs) { 127 // 替换模板原来位置 128 Object ob = changeValue(run.toString(), textMap); 129 if (ob instanceof String) { 130 if (textMap.containsKey(run.toString())) { 131 run.setText("", 0); 132 try (FileInputStream is = new FileInputStream((String) ob)) { 133 run.addPicture(is, XWPFDocument.PICTURE_TYPE_PNG, (String) ob, Units.toEMU(100), 134 Units.toEMU(100)); 135 } 136 } 137 } 138 } 139 } 140 } 141 } 142 143 public static void changeTableText(XWPFDocument document, Map<String, Object> data) { 144 // 获取文件的表格 145 List<XWPFTable> tableList = document.getTables(); 146 147 // 循环所有需要进行替换的文本,进行替换 148 for (int i = 0; i < tableList.size(); i++) { 149 XWPFTable table = tableList.get(i); 150 if (checkText(table.getText())) { 151 List<XWPFTableRow> rows = table.getRows(); 152 // 遍历表格,并替换模板 153 eachTable(document, rows, data); 154 } 155 } 156 } 157 158 public static void changeTablePic(XWPFDocument document, Map<String, Object> pic) throws Exception { 159 List<XWPFTable> tableList = document.getTables(); 160 161 // 循环所有需要替换的文本,进行替换 162 for (int i = 0; i < tableList.size(); i++) { 163 XWPFTable table = tableList.get(i); 164 if (checkText(table.getText())) { 165 List<XWPFTableRow> rows = table.getRows(); 166 System.out.println("简单表格替换:" + rows); 167 // 遍历表格,并替换模板 168 eachTablePic(document, rows, pic); 169 } 170 } 171 } 172 173 public static void eachTablePic(XWPFDocument document, List<XWPFTableRow> rows, Map<String, Object> pic) 174 throws Exception { 175 for (XWPFTableRow row : rows) { 176 List<XWPFTableCell> cells = row.getTableCells(); 177 for (XWPFTableCell cell : cells) { 178 // 判断单元格是否需要替换 179 if (checkText(cell.getText())) { 180 List<XWPFParagraph> paragraphs = cell.getParagraphs(); 181 for (XWPFParagraph paragraph : paragraphs) { 182 List<XWPFRun> runs = paragraph.getRuns(); 183 for (XWPFRun run : runs) { 184 Object ob = changeValue(run.toString(), pic); 185 if (ob instanceof String) { 186 System.out.println("run" + "'" + run.toString() + "'"); 187 if (pic.containsKey(run.toString())) { 188 System.out.println("run" + run.toString() + "替换为" + ob); 189 run.setText("", 0); 190 try (FileInputStream is = new FileInputStream((String) ob)) { 191 run.addPicture(is, XWPFDocument.PICTURE_TYPE_PNG, (String) ob, Units.toEMU(100), 192 Units.toEMU(100)); 193 } 194 } else { 195 System.out.println("'" + run.toString() + "' 不匹配"); 196 } 197 } 198 } 199 } 200 } 201 } 202 } 203 } 204 205 public static Object changeValue(String value, Map<String, Object> textMap) { 206 Set<Map.Entry<String, Object>> textSets = textMap.entrySet(); 207 Object valu = ""; 208 for (Map.Entry<String, Object> textSet : textSets) { 209 // 匹配模板与替换值 格式${key} 210 String key = textSet.getKey(); 211 if (value.contains(key)) { 212 valu = textSet.getValue(); 213 } 214 } 215 return valu; 216 } 217 218 public static void eachTable(XWPFDocument document, List<XWPFTableRow> rows, Map<String, Object> textMap) { 219 for (XWPFTableRow row : rows) { 220 List<XWPFTableCell> cells = row.getTableCells(); 221 for (XWPFTableCell cell : cells) { 222 // 判断单元格是否需要替换 223 if (checkText(cell.getText())) { 224 // System.out.println("cell:" + cell.getText()); 225 List<XWPFParagraph> paragraphs = cell.getParagraphs(); 226 for (XWPFParagraph paragraph : paragraphs) { 227 List<XWPFRun> runs = paragraph.getRuns(); 228 for (XWPFRun run : runs) { 229 230 Object ob = changeValue(run.toString(), textMap); 231 if (ob instanceof String) { 232 233 System.out.println("run:" + "'" + run.toString() + "'"); 234 if (textMap.containsKey(run.toString())) { 235 System.out.println("run:" + run.toString() + "替换为" + ob); 236 run.setText((String) ob, 0); 237 } else { 238 System.out.println("'" + run.toString() + "'不匹配"); 239 } 240 } 241 } 242 } 243 } 244 } 245 } 246 } 247 }