POI动态生成word2007加强版

先看效果图：

public class GeneralTemplateWord2007Util {

public static void main(String[] args) {
// TODO Auto-generated method stub
String filePath = "C:/Users/Administrator/Desktop/doc/模板.docx";
String outFile = "C:/Users/Administrator/Desktop/生成模板.docx";
Map<String, Object> params = new HashMap<String, Object>();
params.put("font", "你好");
params.put("name", "小宝");
params.put("age", "xx");
params.put("sex", "男");
params.put("job", "肉盾");
params.put("hobby", "电商");
params.put("phone", "1717");
        

try {
GeneralTemplateWord2007Util gt = new GeneralTemplateWord2007Util();

Map<String, List<List<String>>> map=new HashMap<String, List<List<String>>>();
map.put("user", gt.generateTestData(5));
map.put("aa", gt.generateTestData(7));
map.put("mytable", gt.generateTestData(11));

gt.templateWrite(filePath, outFile, params, map);
System.out.println("生成模板成功");
} catch (Exception e) {
// TODO Auto-generated catch block
System.out.println("生成模板失败");
e.printStackTrace();
}
}


// 生成测试数据
public List<List<String>> generateTestData(int num) {
List<List<String>> resultList = new ArrayList<List<String>>();
for (int i = 1; i <= num; i++) {
List<String> list = new ArrayList<String>();
list.add("" + i);
list.add("测试_" + i);
list.add("测试2_" + i);
list.add("测试3_" + i);
list.add("测试4_" + i);
list.add("测试5_" + i);
resultList.add(list);
}
return resultList;
}


/**
* 用一个docx文档作为模板，然后替换其中的内容，再写入目标文档中。
* 
* @throws Exception
*/
public void templateWrite(String filePath, String outFile,
Map<String, Object> params, Map<String, List<List<String>>> map)
throws Exception {


InputStream is = new FileInputStream(filePath);
XWPFDocument doc = new XWPFDocument(is);
// 替换段落里面的变量
this.replaceInPara(doc, params);
// 替换表格里面的变量并插入数据
this.insertValueToTable(doc, params, map);
OutputStream os = new FileOutputStream(outFile);
doc.write(os);
this.close(os);
this.close(is);
}


/**
* 替换段落里面的变量
* 
* @param doc
*            要替换的文档
* @param params
*            参数
* @throws Exception 
*/
private void replaceInPara(XWPFDocument doc, Map<String, Object> params) throws Exception {
Iterator<XWPFParagraph> iterator = doc.getParagraphsIterator();
XWPFParagraph para;
while (iterator.hasNext()) {
para = iterator.next();
this.replaceInPara(para, params);
}
}


/**
* 替换段落里面的变量
* 
* @param para
*            要替换的段落
* @param params
*            参数
* @throws Exception 
*/
private String replaceInPara(XWPFParagraph para, Map<String, Object> params) throws Exception {
String str=null;
List<XWPFRun> runs;
XWPFRun run = null;
Matcher matcher;
Matcher mr;
int fontSize=0;
boolean setBold=false;
boolean setItalic=false;
UnderlinePatterns setUnderline=null; 
String setColor="";
int setTextPosition=0;
String setFontFamily=null;


if (this.matcher(para.getParagraphText()).find()) {
runs = para.getRuns();
for (int i = 0; i < runs.size(); i++) {
run = runs.get(i);
String runText = run.toString();
matcher = this.matcher(runText);
if (matcher.find()) {
while ((matcher = this.matcher(runText)).find()) {
runText = matcher.replaceFirst(String.valueOf(params
.get(matcher.group(1))));
}
// 直接调用XWPFRun的setText()方法设置文本时，在底层会重新创建一个XWPFRun，把文本附加在当前文本后面，
// 所以我们不能直接设值，需要先删除当前run,然后再自己手动插入一个新的run。
fontSize=run.getFontSize();
setBold=run.isBold();
setItalic=run.isItalic();
setUnderline=run.getUnderline();
setColor=run.getColor();
setTextPosition=run.getTextPosition();
setFontFamily=run.getFontFamily();

para.removeRun(i);
XWPFRun runP = para.insertNewRun(i);

runP.setText(runText);
runP.setBold(setBold);
runP.setItalic(setItalic);
runP.setUnderline(setUnderline);
runP.setColor(setColor);
runP.setTextPosition(setTextPosition);

if(fontSize!=-1) runP.setFontSize(fontSize);
if (setFontFamily != null) runP.setFontFamily(setFontFamily);
}
}
} else if ((mr=this.matcherRow(para.getParagraphText())).find()) {
str=mr.group(1)+"";
}
return str;
}


/**
* 按模版行样式填充数据,暂未实现特殊样式填充(如列合并)，只能用于普通样式(如段落间距 缩进 字体 对齐)
* 
* @param doc
*            要替换的文档
* @param params
*            参数
* @param resultList
*            需要遍历的数据
* @throws Exception
*/
private void insertValueToTable(XWPFDocument doc,
Map<String, Object> params, Map<String, List<List<String>>> map)
throws Exception {
Iterator<XWPFTable> iterator = doc.getTablesIterator();
XWPFTable table = null;
List<XWPFTableRow> rows = null;
List<XWPFParagraph> paras;
List<XWPFTableCell> tmpCells = null;// 模版列
XWPFTableRow tmpRow = null;// 匹配用
XWPFTableCell tmpCell = null;// 匹配用
int thisRow = 0;
String str=null;
while (iterator.hasNext()) {
List<XWPFTableCell> cells = null;
List<List<String>> resultList=null;
table = iterator.next();
rows = table.getRows();
for (int i = 1; i <= rows.size(); i++) {
cells = rows.get(i - 1).getTableCells();
for (XWPFTableCell cell : cells) {
paras = cell.getParagraphs();
for (XWPFParagraph para : paras) {
str=this.replaceInPara(para, params);
if (str!=null) {
thisRow = i;// 找到模板行
resultList=map.get(str);

tmpRow = rows.get(i - 1);
cells = tmpRow.getTableCells();


}
}
}
}
if (thisRow > 0 && resultList.size()>0) {
this.insertRowToTable(table, tmpRow, tmpCell, thisRow,
resultList, tmpCells, tmpCells);
} else {
System.out.println("该表格中未找到动态标志符");
return;
}
thisRow = 0;
}
}


public void insertRowToTable(XWPFTable table, XWPFTableRow tmpRow,
XWPFTableCell tmpCell, int thisRow, List<List<String>> resultList,
List<XWPFTableCell> tmpCells, List<XWPFTableCell> cells)
throws Exception {
tmpCells = tmpRow.getTableCells();
for (int i = 0, len = resultList.size(); i < len; i++) {
XWPFTableRow row = table.insertNewTableRow(thisRow + i);
row.setHeight(tmpRow.getHeight());
List<String> list = resultList.get(i);
cells = row.getTableCells();
// 插入的行会填充与表格第一行相同的列数
for (int k = 0, klen = cells.size(); k < klen; k++) {
tmpCell = tmpCells.get(k);
XWPFTableCell cell = cells.get(k);
setCellText(tmpCell, cell, list.get(k));
}
// 继续写剩余的列
for (int j = cells.size(), jlen = list.size(); j < jlen; j++) {
tmpCell = tmpCells.get(j);
XWPFTableCell cell = row.addNewTableCell();
setCellText(tmpCell, cell, list.get(j));
}
}
// 删除模版行
table.removeRow(thisRow - 1);
}


public void setCellText(XWPFTableCell tmpCell, XWPFTableCell cell,
String text) throws Exception {
CTTc cttc2 = tmpCell.getCTTc();
CTTcPr ctPr2 = cttc2.getTcPr();


CTTc cttc = cell.getCTTc();
CTTcPr ctPr = cttc.addNewTcPr();
cell.setColor(tmpCell.getColor());
// cell.setVerticalAlignment(tmpCell.getVerticalAlignment());
if (ctPr2.getTcW() != null) {
ctPr.addNewTcW().setW(ctPr2.getTcW().getW());
}
if (ctPr2.getVAlign() != null) {
ctPr.addNewVAlign().setVal(ctPr2.getVAlign().getVal());
}
if (cttc2.getPList().size() > 0) {
CTP ctp = cttc2.getPList().get(0);
if (ctp.getPPr() != null) {
if (ctp.getPPr().getJc() != null) {
cttc.getPList().get(0).addNewPPr().addNewJc()
.setVal(ctp.getPPr().getJc().getVal());
}
}
}


if (ctPr2.getTcBorders() != null) {
ctPr.setTcBorders(ctPr2.getTcBorders());
}


XWPFParagraph tmpP = tmpCell.getParagraphs().get(0);
XWPFParagraph cellP = cell.getParagraphs().get(0);
XWPFRun tmpR = null;
if (tmpP.getRuns() != null && tmpP.getRuns().size() > 0) {
tmpR = tmpP.getRuns().get(0);
}
XWPFRun cellR = cellP.createRun();
cellR.setText(text);
// 复制字体信息
if (tmpR != null) {
cellR.setBold(tmpR.isBold());
cellR.setItalic(tmpR.isItalic());
cellR.setStrike(tmpR.isStrike());
cellR.setUnderline(tmpR.getUnderline());
cellR.setColor(tmpR.getColor());
cellR.setTextPosition(tmpR.getTextPosition());
if (tmpR.getFontSize() != -1) {
cellR.setFontSize(tmpR.getFontSize());
}
if (tmpR.getFontFamily() != null) {
cellR.setFontFamily(tmpR.getFontFamily());
}
if (tmpR.getCTR() != null) {
if (tmpR.getCTR().isSetRPr()) {
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();
cellFonts.setAscii(tmpFonts.getAscii());
cellFonts.setAsciiTheme(tmpFonts.getAsciiTheme());
cellFonts.setCs(tmpFonts.getCs());
cellFonts.setCstheme(tmpFonts.getCstheme());
cellFonts.setEastAsia(tmpFonts.getEastAsia());
cellFonts.setEastAsiaTheme(tmpFonts.getEastAsiaTheme());
cellFonts.setHAnsi(tmpFonts.getHAnsi());
cellFonts.setHAnsiTheme(tmpFonts.getHAnsiTheme());
}
}
}
}
// 复制段落信息
cellP.setAlignment(tmpP.getAlignment());
cellP.setVerticalAlignment(tmpP.getVerticalAlignment());
cellP.setBorderBetween(tmpP.getBorderBetween());
cellP.setBorderBottom(tmpP.getBorderBottom());
cellP.setBorderLeft(tmpP.getBorderLeft());
cellP.setBorderRight(tmpP.getBorderRight());
cellP.setBorderTop(tmpP.getBorderTop());
cellP.setPageBreak(tmpP.isPageBreak());
if (tmpP.getCTP() != null) {
if (tmpP.getCTP().getPPr() != null) {
CTPPr tmpPPr = tmpP.getCTP().getPPr();
CTPPr cellPPr = cellP.getCTP().getPPr() != null ? cellP
.getCTP().getPPr() : cellP.getCTP().addNewPPr();
// 复制段落间距信息
CTSpacing tmpSpacing = tmpPPr.getSpacing();
if (tmpSpacing != null) {
CTSpacing cellSpacing = cellPPr.getSpacing() != null ? cellPPr
.getSpacing() : cellPPr.addNewSpacing();
if (tmpSpacing.getAfter() != null) {
cellSpacing.setAfter(tmpSpacing.getAfter());
}
if (tmpSpacing.getAfterAutospacing() != null) {
cellSpacing.setAfterAutospacing(tmpSpacing
.getAfterAutospacing());
}
if (tmpSpacing.getAfterLines() != null) {
cellSpacing.setAfterLines(tmpSpacing.getAfterLines());
}
if (tmpSpacing.getBefore() != null) {
cellSpacing.setBefore(tmpSpacing.getBefore());
}
if (tmpSpacing.getBeforeAutospacing() != null) {
cellSpacing.setBeforeAutospacing(tmpSpacing
.getBeforeAutospacing());
}
if (tmpSpacing.getBeforeLines() != null) {
cellSpacing.setBeforeLines(tmpSpacing.getBeforeLines());
}
if (tmpSpacing.getLine() != null) {
cellSpacing.setLine(tmpSpacing.getLine());
}
if (tmpSpacing.getLineRule() != null) {
cellSpacing.setLineRule(tmpSpacing.getLineRule());
}
}
// 复制段落缩进信息
CTInd tmpInd = tmpPPr.getInd();
if (tmpInd != null) {
CTInd cellInd = cellPPr.getInd() != null ? cellPPr.getInd()
: cellPPr.addNewInd();
if (tmpInd.getFirstLine() != null) {
cellInd.setFirstLine(tmpInd.getFirstLine());
}
if (tmpInd.getFirstLineChars() != null) {
cellInd.setFirstLineChars(tmpInd.getFirstLineChars());
}
if (tmpInd.getHanging() != null) {
cellInd.setHanging(tmpInd.getHanging());
}
if (tmpInd.getHangingChars() != null) {
cellInd.setHangingChars(tmpInd.getHangingChars());
}
if (tmpInd.getLeft() != null) {
cellInd.setLeft(tmpInd.getLeft());
}
if (tmpInd.getLeftChars() != null) {
cellInd.setLeftChars(tmpInd.getLeftChars());
}
if (tmpInd.getRight() != null) {
cellInd.setRight(tmpInd.getRight());
}
if (tmpInd.getRightChars() != null) {
cellInd.setRightChars(tmpInd.getRightChars());
}
}
}
}
}


/**
* 正则匹配字符串
* 
* @param str
* @return
*/
private Matcher matcher(String str) {
Pattern pattern = Pattern.compile("\\$\\{(.+?)\\}",
Pattern.CASE_INSENSITIVE);
Matcher matcher = pattern.matcher(str);
return matcher;
}


/**
* 正则匹配字符串
* 
* @param str
* @return
*/
private Matcher matcherRow(String str) {
Pattern pattern = Pattern.compile("\\$\\[(.+?)\\]",
Pattern.CASE_INSENSITIVE);
Matcher matcher = pattern.matcher(str);
return matcher;
}


/**
* 关闭输入流
* 
* @param is
*/
private void close(InputStream is) {
if (is != null) {
try {
is.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}


/**
* 关闭输出流
* 
* @param os
*/
private void close(OutputStream os) {
if (os != null) {
try {
os.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}

}