Merge pull request #16 from vpday/master

hellokaton · web-flow · commit 1727c18f3933 · 2018-12-11T20:34:24.000+08:00
✨ to specify the style of a column by condition.
diff --git a/src/main/java/io/github/biezhi/excel/plus/writer/ExcelWriter.java b/src/main/java/io/github/biezhi/excel/plus/writer/ExcelWriter.java
@@ -20,24 +20,15 @@
 import io.github.biezhi.excel.plus.exception.ExcelException;
 import io.github.biezhi.excel.plus.utils.ExcelUtils;
 import io.github.biezhi.excel.plus.utils.Pair;
-
 import org.apache.poi.hssf.usermodel.HSSFWorkbook;
-import org.apache.poi.ss.usermodel.Cell;
-import org.apache.poi.ss.usermodel.CellStyle;
-import org.apache.poi.ss.usermodel.Row;
-import org.apache.poi.ss.usermodel.Sheet;
-import org.apache.poi.ss.usermodel.Workbook;
-import org.apache.poi.ss.usermodel.WorkbookFactory;
+import org.apache.poi.ss.usermodel.*;
 import org.apache.poi.ss.util.CellRangeAddress;
 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 
 import java.io.InputStream;
 import java.io.OutputStream;
-import java.math.BigDecimal;
-import java.util.Collection;
-import java.util.Comparator;
-import java.util.Iterator;
-import java.util.List;
+import java.util.*;
+import java.util.function.Predicate;
 import java.util.stream.Collectors;
 
 /**
@@ -68,6 +59,8 @@ default <T> void export(Exporter<T> exporter, OutputStream outputStream) throws
             CellStyle headerStyle;
             CellStyle columnStyle = null;
             CellStyle titleStyle;
+            List<CellStyle> specialColumnStyles = new ArrayList<>();
+            List<Predicate<String>> specialConditions = new ArrayList<>();
 
             T data0 = data.iterator().next();
             // Set Excel header
@@ -110,6 +103,11 @@ default <T> void export(Exporter<T> exporter, OutputStream outputStream) throws
                     columnStyle = this.defaultColumnStyle(workbook);
                 }
 
+                if (null != exporter.getSpecialColumn()) {
+                    exporter.getSpecialColumn().values().forEach(var -> specialColumnStyles.add(var.apply(workbook)));
+                    specialConditions = new ArrayList<>(exporter.getSpecialColumn().keySet());
+                }
+
                 String headerTitle = exporter.getHeaderTitle();
                 int    colIndex    = 0;
                 if (null != headerTitle) {
@@ -123,7 +121,7 @@ default <T> void export(Exporter<T> exporter, OutputStream outputStream) throws
 
             }
 
-            this.writeRows(sheet, columnStyle, null, iterator, startRow, columnIndexes);
+            this.writeRows(sheet, columnStyle, specialColumnStyles, null, specialConditions, iterator, startRow, columnIndexes);
 
             workbook.write(outputStream);
             outputStream.flush();
@@ -171,24 +169,45 @@ default void writeColumnNames(int rowIndex, CellStyle headerStyle, Sheet sheet,
      *
      * @param sheet       work sheet
      * @param columnStyle each column style in the row.
+     * @param specialColumnStyles each special column style in the row.
      * @param rowStyle    row style
+     * @param specialConditions    the judgment conditions for a special column
      * @param iterator    row data iterator
      * @param startRow    from the beginning of the line, the default is 1
      * @param <T>         Java Type
      */
-    default <T> void writeRows(Sheet sheet, CellStyle columnStyle, CellStyle rowStyle, Iterator<T> iterator, int startRow, List<Integer> columnIndexes) {
+    default <T> void writeRows(Sheet sheet, CellStyle columnStyle, List<CellStyle> specialColumnStyles, CellStyle rowStyle, List<Predicate<String>> specialConditions, Iterator<T> iterator, int startRow, List<Integer> columnIndexes) {
         for (int rowNum = startRow; iterator.hasNext(); rowNum++) {
             T   item = iterator.next();
             Row row  = sheet.createRow(rowNum);
             if (null != rowStyle) {
                 row.setRowStyle(rowStyle);
             }
+
+            boolean isSpecialColumn = false;
+            int index = -1;
+            if (null != specialColumnStyles && null != specialConditions) {
+                here:
+                for (Integer col : columnIndexes) {
+                    String value = ExcelUtils.getColumnValue(item, col);
+                    for (int i = 0, length = specialConditions.size(); i < length; i++) {
+                        index = i;
+                        isSpecialColumn = specialConditions.get(i).test(value);
+                        if (isSpecialColumn) {
+                            break here;
+                        }
+                    }
+                }
+            }
+
             Iterator<Integer> colIt = columnIndexes.iterator();
             while (colIt.hasNext()) {
                 int    col   = colIt.next();
                 Cell   cell  = row.createCell(col);
                 String value = ExcelUtils.getColumnValue(item, col);
-                if (null != columnStyle) {
+                if (isSpecialColumn) {
+                    cell.setCellStyle(specialColumnStyles.get(index));
+                } else if (null != columnStyle) {
                     cell.setCellStyle(columnStyle);
                 }
                 if (null != value) {
diff --git a/src/main/java/io/github/biezhi/excel/plus/writer/Exporter.java b/src/main/java/io/github/biezhi/excel/plus/writer/Exporter.java
@@ -23,7 +23,10 @@
 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 
 import java.util.Collection;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.Function;
+import java.util.function.Predicate;
 
 /**
  * Excel exporter
@@ -45,6 +48,7 @@ public class Exporter<T> {
     private Function<Workbook, CellStyle> titleStyle;
     private Function<Workbook, CellStyle> headerStyle;
     private Function<Workbook, CellStyle> columnStyle;
+    private Map<Predicate<String>,Function<Workbook, CellStyle>> specialColumn = new ConcurrentHashMap<>(16);
 
     public static <T> Exporter<T> create(Collection<T> data) {
         Exporter<T> exporter = new Exporter<>();
@@ -77,6 +81,11 @@ public Exporter<T> columnStyle(Function<Workbook, CellStyle> function) {
         return this;
     }
 
+    public Exporter<T> specialColumn(Map<Predicate<String>,Function<Workbook, CellStyle>> specialColumn) {
+        this.specialColumn.putAll(specialColumn);
+        return this;
+    }
+
     public Exporter<T> byTemplate(String templatePath) {
         this.templatePath = templatePath;
         return this;
@@ -103,6 +112,10 @@ public Function<Workbook, CellStyle> getColumnStyle() {
         return columnStyle;
     }
 
+    public Map<Predicate<String>, Function<Workbook, CellStyle>> getSpecialColumn() {
+        return specialColumn;
+    }
+
     public String getTemplatePath() {
         return templatePath;
     }
diff --git a/src/test/java/io/github/biezhi/excel/plus/ExcelPlusTest.java b/src/test/java/io/github/biezhi/excel/plus/ExcelPlusTest.java
@@ -1,17 +1,16 @@
 package io.github.biezhi.excel.plus;
 
+import io.github.biezhi.excel.plus.enums.ExcelType;
 import io.github.biezhi.excel.plus.enums.ParseType;
 import io.github.biezhi.excel.plus.exception.ExcelException;
-import io.github.biezhi.excel.plus.exception.ParseException;
 import io.github.biezhi.excel.plus.model.CardSecret;
 import io.github.biezhi.excel.plus.reader.Reader;
 import io.github.biezhi.excel.plus.writer.Exporter;
 import io.github.biezhi.excel.plus.writer.ResponseWrapper;
 import org.apache.poi.hssf.util.HSSFColor;
-import org.apache.poi.ss.usermodel.CellStyle;
-import org.apache.poi.ss.usermodel.FillPatternType;
-import org.apache.poi.ss.usermodel.Font;
-import org.apache.poi.ss.usermodel.HorizontalAlignment;
+import org.apache.poi.ss.usermodel.*;
+import org.apache.poi.xssf.usermodel.XSSFCellStyle;
+import org.apache.poi.xssf.usermodel.XSSFColor;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -21,7 +20,11 @@
 import java.net.URLDecoder;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.Objects;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.function.Function;
+import java.util.function.Predicate;
 import java.util.stream.Collectors;
 
 /**
@@ -144,6 +147,49 @@ public void testExportByTpl() throws ExcelException {
         ).writeAsFile(new File("template_rows.xls"));
     }
 
+    @Test
+    public void testExportBySpecials() throws ExcelException {
+        List<CardSecret> cardSecrets = this.buildCardSecrets();
+
+        Map<Predicate<String>, Function<Workbook, CellStyle>> specialColumn = new ConcurrentHashMap<>(3);
+        specialColumn.put(var -> var.equals("vlfdzepjmlz2y43z7er4"), webHook -> {
+            XSSFCellStyle cellStyle = (XSSFCellStyle) webHook.createCellStyle();
+            cellStyle.setAlignment(HorizontalAlignment.LEFT);
+            cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
+            cellStyle.setWrapText(true);
+            XSSFColor xssfColor = new XSSFColor();
+            xssfColor.setRGB(new byte[]{(byte) 252, (byte) 228, (byte) 236});
+            cellStyle.setFillForegroundColor(xssfColor);
+            cellStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
+            Font font = webHook.createFont();
+            font.setFontName("Arial");
+            font.setFontHeightInPoints((short) 12);
+            cellStyle.setFont(font);
+            return cellStyle;
+        });
+        specialColumn.put(var -> var.equals("rasefq2rzotsmx526z6g"), webHook -> {
+            XSSFCellStyle cellStyle = (XSSFCellStyle) webHook.createCellStyle();
+            cellStyle.setAlignment(HorizontalAlignment.LEFT);
+            cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
+            cellStyle.setWrapText(true);
+            XSSFColor xssfColor = new XSSFColor();
+            xssfColor.setRGB(new byte[]{(byte) 255, (byte) 235, (byte) 238});
+            cellStyle.setFillForegroundColor(xssfColor);
+            cellStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
+            Font font = webHook.createFont();
+            font.setFontName("Arial");
+            font.setFontHeightInPoints((short) 13);
+            cellStyle.setFont(font);
+            return cellStyle;
+        });
+
+        Exporter<CardSecret> exporter = Exporter.create(cardSecrets)
+                .sheetName("卡密列表第一季数据").specialColumn(specialColumn);
+        exporter.setExcelType(ExcelType.XLSX);
+
+        excelPlus.export(exporter).writeAsFile(new File("export.xlsx"));
+    }
+
     //    @Test
     public void testDownload() throws ExcelException {
         List<CardSecret> cardSecrets = this.buildCardSecrets();
