ExcelTemplate实现输入excel模板，导出结果

代码功能(function)

提供输入一个excel模板，然后替换掉部分数据，之后再输出一个excel的文档
Propose a method that input a excel template and output a excel with some datas with which you want to replace the placeholder

实现方法(code implement)

Excel需要替换的数据，一定要用占位符“${}”包括，这样才可以替换，当然如果你想使用自己定义的占位符，请自行修改代码中的replaceTemplateWithSpecifiedData方法的代码

First, input excel template which contains some placeholders with "${}" that will be replaced with what you want. If you want to use your own placeholder, you should modify the code which is in the method replaceTemplateWithSpecifiedData.

GitHub code 完整代码

代码

public class ExcelTemplate {

    private SXSSFWorkbook writeWorkbook;
    private XSSFWorkbook readWorkbook;
    private XSSFSheet readSheet;

    public ExcelTemplate(String filePath) {

        try {
            readWorkbook = new XSSFWorkbook(new FileInputStream(filePath));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public ExcelTemplate(InputStream inputStream) {
        try {
            ZipSecureFile.setMinInflateRatio(-1.0d);
            readWorkbook = new XSSFWorkbook(inputStream);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }


    public String getData(int sheetIndex, int row, int column) {

        readSheet = readWorkbook.getSheetAt(sheetIndex);
        String data = "";
        Row r = readSheet.getRow(row);
        Cell c = r.getCell(column, Row.MissingCellPolicy.RETURN_BLANK_AS_NULL);

        if (c == null) {
        } else {
            if (r.getCell(column).getCellType() == HSSFCell.CELL_TYPE_STRING) {
                data = r.getCell(column).getStringCellValue();
            } else if (r.getCell(column).getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
                int intData = (int) r.getCell(column).getNumericCellValue();
                data = Integer.toString(intData);
            }
        }
        return data;
    }


    public void replaceTemplateWithSpecifiedData(int sheetIndex, Map replaceDataMap) throws Exception {
        readSheet = readWorkbook.getSheetAt(sheetIndex);
        if (readSheet != null) {
            this.replaceTemplateWithSpecifiedData(readSheet.getSheetName(), replaceDataMap);
        }
    }

    public void replaceTemplateWithSpecifiedData(String sheetName, Map replaceDataMap) throws Exception {

        readSheet = readWorkbook.getSheet(sheetName);
        if (readSheet == null) {
            throw new Exception("sheet with name " + sheetName + " not found");
        }
        readSheet.setForceFormulaRecalculation(true);

        StrSubstitutor substitutor = new StrSubstitutor(replaceDataMap, "${", "}");

        Iterator rowIterator = readSheet.iterator();
        while (rowIterator.hasNext()) {
            Row row = rowIterator.next();

            // For each row, iterate through all the columns
            Iterator cellIterator = row.cellIterator();
            while (cellIterator.hasNext()) {
                Cell cell = cellIterator.next();
                String cellValue = null;
                try {
                    cellValue = cell.getStringCellValue();
                } catch (Exception e) {
                    try {
                        cellValue = String.valueOf(cell.getNumericCellValue());
                    } catch (Exception e1) {
                        e1.printStackTrace();
                    }
                    e.printStackTrace();
                }

                if (!StringUtils.isEmpty(cellValue)) {
                    cell.setCellValue(substitutor.replace(cellValue));
                }
            }
        }
    }

    /**
     * 清除所有sheet中的占位符
     * Clear all the placeholders
     */
    public void clearRestOfPlaceholder() {
        Iterator sheetIterator = readWorkbook.sheetIterator();
        while (sheetIterator.hasNext()) {
            XSSFSheet xssfSheet = (XSSFSheet) sheetIterator.next();

            xssfSheet.setForceFormulaRecalculation(true);

            Iterator rowIterator = xssfSheet.iterator();
            while (rowIterator.hasNext()) {
                Row row = rowIterator.next();

                // For each row, iterate through all the columns
                Iterator cellIterator = row.cellIterator();
                while (cellIterator.hasNext()) {
                    Cell cell = cellIterator.next();
                    String cellValue = null;
                    try {
                        cellValue = cell.getStringCellValue();
                    } catch (Exception e) {
                        try {
                            cellValue = String.valueOf(cell.getNumericCellValue());
                        } catch (Exception e1) {
                            e1.printStackTrace();
                        }
                        e.printStackTrace();
                    }

                    if (!StringUtils.isEmpty(cellValue)) {
                        if (cellValue.startsWith("${") && cellValue.endsWith("}")) {
                            cell.setCellValue("");
                        }
                    }
                }
            }
        }
    }

    /**
     * 清除指定sheet中剩余的占位符
     * Clear placeholders in the given sheet name
     *
     * @param sheetName
     */
    public void clearRestOfPlaceholder(String sheetName) throws Exception {
        int sheetIndex = readWorkbook.getSheetIndex(sheetName);
        this.clearRestOfPlaceholder(sheetIndex);
    }

    /**
     * 清除指定sheet中剩余的占位符
     *
     * Clear placeholders in the given sheet index
     * @param sheetIndex
     */
    public void clearRestOfPlaceholder(int sheetIndex) throws Exception {
        readSheet = readWorkbook.getSheetAt(sheetIndex);
        if (readSheet == null) {
            throw new Exception("sheet with index " + sheetIndex + " not found");
        }
        readSheet.setForceFormulaRecalculation(true);

        Iterator rowIterator = readSheet.iterator();
        while (rowIterator.hasNext()) {
            Row row = rowIterator.next();

            // For each row, iterate through all the columns
            Iterator cellIterator = row.cellIterator();
            while (cellIterator.hasNext()) {
                Cell cell = cellIterator.next();
                String cellValue = null;
                try {
                    cellValue = cell.getStringCellValue();
                } catch (Exception e) {
                    try {
                        cellValue = String.valueOf(cell.getNumericCellValue());
                    } catch (Exception e1) {
                        e1.printStackTrace();
                    }
                    e.printStackTrace();
                }

                if (!StringUtils.isEmpty(cellValue)) {
                    if (cellValue.startsWith("${") && cellValue.endsWith("}")) {
                        cell.setCellValue("");
                    }
                }
            }
        }
    }

    public void writeToFile(String filePath) throws IOException {
        if (StringUtils.isEmpty(filePath)) {
            return;
        }
        FileOutputStream fout = new FileOutputStream(filePath);
        this.writeToFile(fout);
    }

    public void writeToFile(FileOutputStream fileOutputStream) throws IOException {
        if (fileOutputStream == null) {
            return;
        }
        try {
            writeWorkbook = new SXSSFWorkbook(readWorkbook);
            writeWorkbook.write(fileOutputStream);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            this.close();
        }
    }

    public void close() throws IOException {
        readWorkbook.close();
        writeWorkbook.close();
    }
}