From b9b490e0ec226389c6e2502fecf918f7537d8c57 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期四, 21 三月 2024 10:57:52 +0800
Subject: [PATCH] 添加Word Excel测试方法
---
src/main/java/com/lf/server/config/InitConfig.java | 6
src/main/java/com/lf/server/helper/WordHelper.java | 205 +++++++++++++++++++++++++++++++++++++---
src/main/java/com/lf/server/service/all/TestService.java | 41 ++++++++
src/main/java/com/lf/server/helper/ExcelHelper.java | 26 +++++
4 files changed, 259 insertions(+), 19 deletions(-)
diff --git a/src/main/java/com/lf/server/config/InitConfig.java b/src/main/java/com/lf/server/config/InitConfig.java
index 19a6b99..18b964c 100644
--- a/src/main/java/com/lf/server/config/InitConfig.java
+++ b/src/main/java/com/lf/server/config/InitConfig.java
@@ -33,8 +33,8 @@
@Resource
Environment env;
- //@Resource
- //TestService testService;
+ @Resource
+ TestService testService;
@Resource
AutoQueryService autoQueryService;
@@ -52,7 +52,7 @@
GdalHelper.init(env.getProperty("sys.gdal_path"));
UploadAttachService.init(env.getProperty("sys.attachTabs"));
- //testService.testRegister();
+ //testService.testExcel();
pathHelper.init();
StaticData.ADMIN = env.getProperty("sys.admin");
argsService.initSettingData();
diff --git a/src/main/java/com/lf/server/helper/ExcelHelper.java b/src/main/java/com/lf/server/helper/ExcelHelper.java
index 060653c..d0abdca 100644
--- a/src/main/java/com/lf/server/helper/ExcelHelper.java
+++ b/src/main/java/com/lf/server/helper/ExcelHelper.java
@@ -74,6 +74,32 @@
/**
* 鍐欏叆Excel妯℃澘
*
+ * @param source 婧愭枃浠讹紙妯℃澘锛�
+ * @param target 鐩綍鏂囦欢
+ * @param map 閿�煎Map
+ * @param listMap 鍒楄〃Map
+ */
+ public static <T> void writeToTemplate(String source, String target, Map<String, Object> map, Map<String, List<T>> listMap) {
+ // 鏍规嵁妯℃澘鍐欏叆鏁版嵁锛屽鏋滅洰鏍囨枃浠朵笉瀛樺湪锛屽垯鑷姩鍒涘缓鏂囦欢
+ ExcelWriter excelWriter = EasyExcel.write(target).withTemplate(source).build();
+
+ // 鍦ㄥ伐浣滅翱0涓啓鍏ユ暟鎹紝濡傛灉妯℃澘涓笉瀛樺湪缁冧範宸ヤ綔绨匡紝鍒欎細鍦ㄧ洰鏍囨枃浠朵腑鑷姩鍒涘缓
+ WriteSheet writeSheet = EasyExcel.writerSheet(0).build();
+
+ // 鍨傜洿鍐欏叆鏁版嵁锛屽鏋滆姘村钩鍐欏叆锛屽皢VERTICAL鏇挎崲涓篐ORIZONTAL
+ FillConfig fillConfig = FillConfig.builder().direction(WriteDirectionEnum.VERTICAL).build();
+
+ // 鍐欏叆鏁版嵁
+ excelWriter.fill(map, writeSheet);
+ listMap.forEach((k, v) -> excelWriter.fill(new FillWrapper(k, v), fillConfig, writeSheet));
+
+ // 缁撴潫鍐欏叆
+ excelWriter.finish();
+ }
+
+ /**
+ * 鍐欏叆Excel妯℃澘
+ *
* @param source 婧愭枃浠讹紙妯℃澘锛�
* @param target 鐩綍鏂囦欢
* @param map 鏁版嵁婧�
diff --git a/src/main/java/com/lf/server/helper/WordHelper.java b/src/main/java/com/lf/server/helper/WordHelper.java
index 5704e07..c977318 100644
--- a/src/main/java/com/lf/server/helper/WordHelper.java
+++ b/src/main/java/com/lf/server/helper/WordHelper.java
@@ -8,18 +8,22 @@
import java.io.*;
import java.util.List;
import java.util.Map;
+import java.util.Set;
/**
* Word甯姪绫�
+ *
* @author WWW
+ * @date 2024-03-21
*/
public class WordHelper {
private final static Log log = LogFactory.getLog(WordHelper.class);
/**
* 閫氳繃word妯℃澘鐢熸垚word鐨勪富鏂规硶
+ * https://blog.csdn.net/qq_46112324/article/details/128156675
*/
- public static void generateWord(String inputFile, String outPutFile, Map<String, String> insertTextMap, List<String[]> addList) {
+ public static void generateWord(String inputFile, String outPutFile, Map<String, String> textMap, List<String[]> tableList) {
FileInputStream inputStream = null;
FileOutputStream outputStream = null;
@@ -28,18 +32,23 @@
outputStream = new FileOutputStream(outPutFile);
// 鑾峰彇docx瑙ｆ瀽瀵硅薄
- XWPFDocument xwpfDocument = new XWPFDocument(inputStream);
+ XWPFDocument document = new XWPFDocument(inputStream);
// 澶勭悊鎵�鏈夋枃娈垫暟鎹紝闄や簡琛ㄦ牸
- if (null != insertTextMap && insertTextMap.size() > 0) {
- handleParagraphs(xwpfDocument, insertTextMap);
+ if (null != textMap && textMap.size() > 0) {
+ handleParagraphs(document, textMap);
+ //changeText(document, textMap);
}
- // 澶勭悊琛ㄦ牸鏁版嵁
- handleTable(xwpfDocument, insertTextMap, addList);
+ // 瑙ｆ瀽鏇挎崲琛ㄦ牸瀵硅薄
+ if (null != tableList && tableList.size() > 0) {
+ handleTable(document, textMap, tableList);
+ //changeTable(document, textMap, tableList);
+ }
// 鍐欏叆鏁版嵁
- xwpfDocument.write(outputStream);
+ document.write(outputStream);
+ document.close();
} catch (Exception ex) {
log.error(ex.getMessage(), ex);
} finally {
@@ -55,6 +64,170 @@
}
}
}
+
+ /**
+ * 鏇挎崲琛ㄦ牸瀵硅薄鏂规硶
+ *
+ * @param document docx瑙ｆ瀽瀵硅薄
+ * @param textMap 闇�瑕佹浛鎹㈢殑淇℃伅闆嗗悎
+ * @param tableList 闇�瑕佹彃鍏ョ殑琛ㄦ牸淇℃伅闆嗗悎
+ */
+ public static void changeTable(XWPFDocument document, Map<String, String> textMap, List<String[]> tableList) {
+ // 鑾峰彇琛ㄦ牸瀵硅薄闆嗗悎
+ List<XWPFTable> tables = document.getTables();
+ for (int i = 0; i < tables.size(); i++) {
+ XWPFTable table = tables.get(i);
+ if (table.getRows().size() > 1) {
+ // 鍒ゆ柇琛ㄦ牸鏄渶瑕佹浛鎹㈣繕鏄渶瑕佹彃鍏ワ紝鍒ゆ柇閫昏緫鏈�$涓烘浛鎹紝琛ㄦ牸鏃�$涓烘彃鍏�
+ if (!checkText(table.getText())) {
+ insertTable(table, tableList);
+ }
+ }
+ }
+ }
+
+ /**
+ * 鏇挎崲娈佃惤鏂囨湰
+ *
+ * @param document docx瑙ｆ瀽瀵硅薄
+ * @param textMap 闇�瑕佹浛鎹㈢殑淇℃伅闆嗗悎
+ */
+ public static void changeText(XWPFDocument document, Map<String, String> textMap) {
+ // 鑾峰彇娈佃惤闆嗗悎
+ List<XWPFTable> tables = document.getTables();
+ for (XWPFTable table : tables) {
+ if (checkText(table.getText())) {
+ List<XWPFTableRow> rows = table.getRows();
+ for (XWPFTableRow row : rows) {
+ List<XWPFTableCell> tableCells = row.getTableCells();
+ for (XWPFTableCell cell : tableCells) {
+ if (checkText(cell.getText())) {
+ List<XWPFParagraph> paragraphs = cell.getParagraphs();
+ for (XWPFParagraph paragraph : paragraphs) {
+ // 鍒ゆ柇姝ゆ钀芥椂鍊欓渶瑕佽繘琛屾浛鎹�
+ String text = paragraph.getText();
+ if (checkText(text)) {
+ List<XWPFRun> runs = paragraph.getRuns();
+ for (XWPFRun run : runs) {
+ // 鏇挎崲妯℃澘鍘熸潵浣嶇疆
+ if (checkText(run.toString())) {
+ run.setText(changeValue(paragraph.getText(), textMap), 0);
+ } else {
+ run.setText("", 0);
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+
+ /**
+ * 鍒ゆ柇鏂囨湰涓椂鍊欏寘鍚�$
+ *
+ * @param text 鏂囨湰
+ * @return 鍖呭惈杩斿洖true, 涓嶅寘鍚繑鍥瀎alse
+ */
+ public static boolean checkText(String text) {
+ boolean check = false;
+ if (text.contains("$")) {
+ check = true;
+ }
+ return check;
+ }
+
+ /**
+ * 鍖归厤浼犲叆淇℃伅闆嗗悎涓庢ā鏉�
+ *
+ * @param value 妯℃澘闇�瑕佹浛鎹㈢殑鍖哄煙
+ * @param textMap 浼犲叆淇℃伅闆嗗悎
+ * @return 妯℃澘闇�瑕佹浛鎹㈠尯鍩熶俊鎭泦鍚堝搴斿��
+ */
+ public static String changeValue(String value, Map<String, String> textMap) {
+ Set<Map.Entry<String, String>> textSets = textMap.entrySet();
+ for (Map.Entry<String, String> textSet : textSets) {
+ // 鍖归厤妯℃澘涓庢浛鎹㈠�� 鏍煎紡${key}
+ String key = "${" + textSet.getKey() + "}";
+ if (key.contains(value)) {
+ value = textSet.getValue();
+ }
+ }
+ return value;
+ }
+
+ /**
+ * 涓鸿〃鏍兼彃鍏ユ暟鎹紝琛屾暟涓嶅娣诲姞鏂拌
+ *
+ * @param table 闇�瑕佹彃鍏ユ暟鎹殑琛ㄦ牸
+ * @param tableList 鎻掑叆鏁版嵁闆嗗悎
+ */
+ public static void insertTable(XWPFTable table, List<String[]> tableList) {
+ // 璁板綍闇�瑕佹彃鍏ョ殑绗竴琛�
+ int start = -1;
+ for (int i = 0; i < table.getRows().size(); i++) {
+ for (int j = 0; j < table.getRows().get(i).getTableCells().size(); j++) {
+ if ("".equals(table.getRows().get(i).getTableCells().get(j).getText())) {
+ start = i;
+ break;
+ }
+ }
+ }
+ // 鍒涘缓琛�,鏍规嵁闇�瑕佹彃鍏ョ殑鏁版嵁娣诲姞鏂拌
+ if (start != -1) {
+ for (int i = 1; i < tableList.size(); i++) {
+ insertRow(table, start, start + i);
+ }
+ }
+
+ // 閬嶅巻琛ㄦ牸鎻掑叆鏁版嵁
+ List<XWPFTableRow> rows = table.getRows();
+ for (int i = start; i < rows.size(); i++) {
+ List<XWPFTableCell> cells = rows.get(i).getTableCells();
+ for (int j = 0; j < table.getRow(start).getTableCells().size(); j++) {
+ XWPFTableCell cell = cells.get(j);
+ cell.setText(tableList.get(0)[j]);
+ }
+ tableList.remove(0);
+ }
+ }
+
+ /**
+ * 涓鸿〃鏍兼彃鍏ユ暟鎹紝琛屾暟涓嶅娣诲姞鏂拌
+ *
+ * @param table 闇�瑕佹彃鍏ユ暟鎹殑琛ㄦ牸
+ * @param copyRowIndex 澶嶅埗琛屽彿
+ * @param newRowIndex 鏂拌鍙�
+ */
+ public static void insertRow(XWPFTable table, int copyRowIndex, int newRowIndex) {
+ // 鍦ㄨ〃鏍间腑鎸囧畾鐨勪綅缃柊澧炰竴琛�
+ XWPFTableRow targetRow = table.insertNewTableRow(newRowIndex);
+
+ // 鑾峰彇闇�瑕佸鍒惰瀵硅薄
+ XWPFTableRow copyRow = table.getRow(copyRowIndex);
+
+ // 澶嶅埗琛屽璞�
+ targetRow.getCtRow().setTrPr(copyRow.getCtRow().getTrPr());
+ //鎴栬闇�瑕佸鍒剁殑琛岀殑鍒�
+ List<XWPFTableCell> copyCells = copyRow.getTableCells();
+ // 澶嶅埗鍒楀璞�
+ XWPFTableCell targetCell = null;
+ for (XWPFTableCell copyCell : copyCells) {
+ targetCell = targetRow.addNewTableCell();
+ targetCell.getCTTc().setTcPr(copyCell.getCTTc().getTcPr());
+ if (copyCell.getParagraphs() != null && copyCell.getParagraphs().size() > 0) {
+ targetCell.getParagraphs().get(0).getCTP().setPPr(copyCell.getParagraphs().get(0).getCTP().getPPr());
+ if (copyCell.getParagraphs().get(0).getRuns() != null
+ && copyCell.getParagraphs().get(0).getRuns().size() > 0) {
+ XWPFRun cellR = targetCell.getParagraphs().get(0).createRun();
+ cellR.setBold(copyCell.getParagraphs().get(0).getRuns().get(0).isBold());
+ }
+ }
+ }
+ }
+
/**
* 澶勭悊鎵�鏈夋枃娈垫暟鎹紝闄や簡琛ㄦ牸
@@ -165,8 +338,11 @@
public static String matchesValue(String wordValue, Map<String, String> map) {
for (String s : map.keySet()) {
String s1 = "${" + s + "}";
- if (s1.equals(wordValue)) {
+ /*if (s1.equals(wordValue)) {
wordValue = map.get(s);
+ }*/
+ if (wordValue.contains(s1)) {
+ wordValue = wordValue.replace(s1, map.get(s));
}
}
@@ -177,7 +353,7 @@
* 娴嬭瘯鏄惁鍖呭惈闇�瑕佹浛鎹㈢殑鏁版嵁
*/
public static boolean isReplacement(String text) {
- return text.contains("$");
+ return text.contains("${");
}
/**
@@ -220,7 +396,7 @@
cellP.setBorderTop(tmpP.getBorderTop());
cellP.setPageBreak(tmpP.isPageBreak());
- if (tmpP.getCTP() != null&&tmpP.getCTP().getPPr() != null) {
+ if (tmpP.getCTP() != null && tmpP.getCTP().getPPr() != null) {
CTPPr tmpPpr = tmpP.getCTP().getPPr();
CTPPr cellPpr = cellP.getCTP().getPPr() != null ? cellP.getCTP().getPPr() : cellP.getCTP().addNewPPr();
@@ -257,10 +433,8 @@
CTRPr tmpRpr = tmpR.getCTR().getRPr();
if (tmpRpr.isSetRFonts()) {
CTFonts tmpFonts = tmpRpr.getRFonts();
- CTRPr cellRpr = cellR.getCTR().isSetRPr() ? cellR
- .getCTR().getRPr() : cellR.getCTR().addNewRPr();
- CTFonts cellFonts = cellRpr.isSetRFonts() ? cellRpr
- .getRFonts() : cellRpr.addNewRFonts();
+ CTRPr cellRpr = cellR.getCTR().isSetRPr() ? cellR.getCTR().getRPr() : cellR.getCTR().addNewRPr();
+ CTFonts cellFonts = cellRpr.isSetRFonts() ? cellRpr.getRFonts() : cellRpr.addNewRFonts();
cellFonts.setAscii(tmpFonts.getAscii());
cellFonts.setAsciiTheme(tmpFonts.getAsciiTheme());
cellFonts.setCs(tmpFonts.getCs());
@@ -285,8 +459,7 @@
cellSpacing.setAfter(tmpSpacing.getAfter());
}
if (tmpSpacing.getAfterAutospacing() != null) {
- cellSpacing.setAfterAutospacing(tmpSpacing
- .getAfterAutospacing());
+ cellSpacing.setAfterAutospacing(tmpSpacing.getAfterAutospacing());
}
if (tmpSpacing.getAfterLines() != null) {
cellSpacing.setAfterLines(tmpSpacing.getAfterLines());
diff --git a/src/main/java/com/lf/server/service/all/TestService.java b/src/main/java/com/lf/server/service/all/TestService.java
index ab29d61..ebfce76 100644
--- a/src/main/java/com/lf/server/service/all/TestService.java
+++ b/src/main/java/com/lf/server/service/all/TestService.java
@@ -8,6 +8,7 @@
import com.lf.server.entity.all.PermsAuthEntity;
import com.lf.server.entity.all.ResAuthEntity;
import com.lf.server.entity.bd.DlgagnpEntity;
+import com.lf.server.entity.ctrl.CountEntity;
import com.lf.server.entity.data.MetaEntity;
import com.lf.server.entity.data.MetaFileEntity;
import com.lf.server.entity.md.MdzxcgEntity;
@@ -265,4 +266,44 @@
//
}
}
+
+ public void testWord() {
+ String inputFile = "C:\\Users\\Administrator\\Desktop\\ts\\椤圭洰鏁版嵁鍒嗙被缁熻.docx";
+ String outPutFile = "C:\\Users\\Administrator\\Desktop\\ts\\NewWord.docx";
+
+ Map<String, String> textMap = new HashMap<>();
+ textMap.put("title", "娣卞湷娑插寲澶╃劧姘斿簲鎬ヨ皟宄扮珯澶栬緭绠￠亾宸ョ▼锛圶Q03T04锛�");
+
+ List<String[]> tableList = new ArrayList<>();
+ tableList.add(new String[]{"" + 1, "鏁板瓧绾垮垝鍥�", "123.54骞虫柟绫�"});
+ tableList.add(new String[]{"" + 2, "鍦扮伨鐐�", "120涓�"});
+
+ WordHelper.generateWord(inputFile, outPutFile, textMap, tableList);
+ }
+
+ public void testExcel() {
+ String source = "C:\\Users\\Administrator\\Desktop\\ts\\椤圭洰鏁版嵁鍒嗙被缁熻.xlsx";
+ String target = "C:\\Users\\Administrator\\Desktop\\ts\\NewWord.xlsx";
+
+ List<CountEntity> list = new ArrayList<>();
+ CountEntity c1 = new CountEntity();
+ c1.setNo(1);
+ c1.setM1("鏁板瓧绾垮垝鍥�");
+ c1.setM2("123.54骞虫柟绫�");
+ list.add(c1);
+
+ CountEntity c2 = new CountEntity();
+ c2.setNo(2);
+ c2.setM1("鍦扮伨鐐�");
+ c2.setM2("120涓�");
+ list.add(c2);
+
+ Map<String, Object> map = new HashMap<>(1);
+ map.put("title", "娣卞湷娑插寲澶╃劧姘斿簲鎬ヨ皟宄扮珯澶栬緭绠￠亾宸ョ▼锛圶Q03T04锛�");
+
+ Map<String, List<CountEntity>> listMap = new HashMap<>(1);
+ listMap.put("data", list);
+
+ ExcelHelper.writeToTemplate(source, target, map, listMap);
+ }
}
--
Gitblit v1.9.3