Excel注解支持自动统计数据总和

RuoYi

2020-08-26 215952da7a3b78d02d5ec83f1f69974f43dab813

 src/main/java/com/ruoyi/common/utils/poi/ExcelUtil.java

@@ -16,6 +16,7 @@
import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.UUID;

import java.util.stream.Collectors;

import org.apache.poi.hssf.usermodel.HSSFDateUtil;

@@ -102,6 +103,16 @@
     * 注解列表

     */

    private List<Object[]> fields;



    /**

     * 统计列表

     */

    private Map<Integer, Double> statistics = new HashMap<Integer, Double>();



    /**

     * 数字格式

     */

    private static final DecimalFormat DOUBLE_FORMAT = new DecimalFormat("######0.00");



    /**

     * 实体对象

@@ -200,7 +211,10 @@
                    // 设置类的私有字段属性可访问.

                    field.setAccessible(true);

                    Integer column = cellMap.get(attr.name());

                    fieldsMap.put(column, field);

                    if (column != null)

                    {

                        fieldsMap.put(column, field);

                    }

                }

            }

            for (int i = 1; i < rows; i++)

@@ -271,11 +285,11 @@
                        }

                        else if (StringUtils.isNotEmpty(attr.readConverterExp()))

                        {

                            val = reverseByExp(Convert.toStr(val), attr.readConverterExp());

                            val = reverseByExp(Convert.toStr(val), attr.readConverterExp(), attr.separator());

                        }

                        else if (StringUtils.isNotEmpty(attr.dictType()))

                        {

                            val = reverseDictByExp(attr.dictType(), Convert.toStr(val));

                            val = reverseDictByExp(Convert.toStr(val), attr.dictType(), attr.separator());

                        }

                        ReflectUtils.invokeSetter(entity, propertyName, val);

                    }

@@ -339,6 +353,7 @@
                if (Type.EXPORT.equals(type))

                {

                    fillExcelData(index, row);

                    addStatisticsRow();

                }

            }

            String filename = encodingFilename(sheetName);

@@ -445,6 +460,15 @@
        headerFont.setColor(IndexedColors.WHITE.getIndex());

        style.setFont(headerFont);

        styles.put("header", style);

        
        style = wb.createCellStyle();

        style.setAlignment(HorizontalAlignment.CENTER);

        style.setVerticalAlignment(VerticalAlignment.CENTER);

        Font totalFont = wb.createFont();

        totalFont.setFontName("Arial");

        totalFont.setFontHeightInPoints((short) 10);

        style.setFont(totalFont);

        styles.put("total", style);



        return styles;

    }

@@ -474,7 +498,7 @@
    {

        if (ColumnType.STRING == attr.cellType())

        {

            cell.setCellType(CellType.NUMERIC);

            cell.setCellType(CellType.STRING);

            cell.setCellValue(StringUtils.isNull(value) ? attr.defaultValue() : value + attr.suffix());

        }

        else if (ColumnType.NUMERIC == attr.cellType())

@@ -534,6 +558,7 @@
                Object value = getTargetValue(vo, field, attr);

                String dateFormat = attr.dateFormat();

                String readConverterExp = attr.readConverterExp();

                String separator = attr.separator();

                String dictType = attr.dictType();

                if (StringUtils.isNotEmpty(dateFormat) && StringUtils.isNotNull(value))

                {

@@ -541,17 +566,22 @@
                }

                else if (StringUtils.isNotEmpty(readConverterExp) && StringUtils.isNotNull(value))

                {

                    cell.setCellValue(convertByExp(Convert.toStr(value), readConverterExp));

                    cell.setCellValue(convertByExp(Convert.toStr(value), readConverterExp, separator));

                }

                else if (StringUtils.isNotEmpty(dictType))

                else if (StringUtils.isNotEmpty(dictType) && StringUtils.isNotNull(value))

                {

                    cell.setCellValue(convertDictByExp(dictType, Convert.toStr(value)));

                    cell.setCellValue(convertDictByExp(Convert.toStr(value), dictType, separator));

                }

                else if (value instanceof BigDecimal && -1 != attr.scale())

                {

                    cell.setCellValue((((BigDecimal) value).setScale(attr.scale(), attr.roundingMode())).toString());

                }

                else

                {

                    // 设置列类型

                    setCellVo(value, attr, cell);

                }

                addStatisticsData(column, Convert.toStr(value), attr);

            }

        }

        catch (Exception e)

@@ -623,28 +653,36 @@
     * 

     * @param propertyValue 参数值

     * @param converterExp 翻译注解

     * @param separator 分隔符

     * @return 解析后值

     * @throws Exception

     */

    public static String convertByExp(String propertyValue, String converterExp) throws Exception

    public static String convertByExp(String propertyValue, String converterExp, String separator)

    {

        try

        StringBuilder propertyString = new StringBuilder();

        String[] convertSource = converterExp.split(",");

        for (String item : convertSource)

        {

            String[] convertSource = converterExp.split(",");

            for (String item : convertSource)

            String[] itemArray = item.split("=");

            if (StringUtils.containsAny(separator, propertyValue))

            {

                String[] itemArray = item.split("=");

                for (String value : propertyValue.split(separator))

                {

                    if (itemArray[0].equals(value))

                    {

                        propertyString.append(itemArray[1] + separator);

                        break;

                    }

                }

            }

            else

            {

                if (itemArray[0].equals(propertyValue))

                {

                    return itemArray[1];

                }

            }

        }

        catch (Exception e)

        {

            throw e;

        }

        return propertyValue;

        return StringUtils.stripEnd(propertyString.toString(), separator);

    }



    /**

@@ -652,52 +690,109 @@
     * 

     * @param propertyValue 参数值

     * @param converterExp 翻译注解

     * @param separator 分隔符

     * @return 解析后值

     * @throws Exception

     */

    public static String reverseByExp(String propertyValue, String converterExp) throws Exception

    public static String reverseByExp(String propertyValue, String converterExp, String separator)

    {

        try

        StringBuilder propertyString = new StringBuilder();

        String[] convertSource = converterExp.split(",");

        for (String item : convertSource)

        {

            String[] convertSource = converterExp.split(",");

            for (String item : convertSource)

            String[] itemArray = item.split("=");

            if (StringUtils.containsAny(separator, propertyValue))

            {

                String[] itemArray = item.split("=");

                for (String value : propertyValue.split(separator))

                {

                    if (itemArray[1].equals(value))

                    {

                        propertyString.append(itemArray[0] + separator);

                        break;

                    }

                }

            }

            else

            {

                if (itemArray[1].equals(propertyValue))

                {

                    return itemArray[0];

                }

            }

        }

        catch (Exception e)

        {

            throw e;

        }

        return propertyValue;

        return StringUtils.stripEnd(propertyString.toString(), separator);

    }



    /**

     * 解析字典值

     * 

     * @param dictType 字典类型

     * @param dictValue 字典值

     * @param dictType 字典类型

     * @param separator 分隔符

     * @return 字典标签

     */

    public static String convertDictByExp(String dictType, String dictValue) throws Exception

    public static String convertDictByExp(String dictValue, String dictType, String separator)

    {

        return DictUtils.getDictLabel(dictType, dictValue);

        return DictUtils.getDictLabel(dictType, dictValue, separator);

    }



    /**

     * 反向解析值字典值

     * 

     * @param dictLabel 字典标签

     * @param dictType 字典类型

     * @param dictValue 字典标签

     * @param separator 分隔符

     * @return 字典值

     */

    public static String reverseDictByExp(String dictType, String dictLabel) throws Exception

    public static String reverseDictByExp(String dictLabel, String dictType, String separator)

    {

        return DictUtils.getDictValue(dictType, dictLabel);

        return DictUtils.getDictValue(dictType, dictLabel, separator);

    }



    /**

     * 合计统计信息

     */

    private void addStatisticsData(Integer index, String text, Excel entity)

    {

        if (entity != null && entity.isStatistics())

        {

            Double temp = 0D;

            if (!statistics.containsKey(index))

            {

                statistics.put(index, temp);

            }

            try

            {

                temp = Double.valueOf(text);

            }

            catch (NumberFormatException e)

            {

            }

            statistics.put(index, statistics.get(index) + temp);

        }

    }



    /**

     * 创建统计行

     */

    public void addStatisticsRow()

    {

        if (statistics.size() > 0)

        {

            Cell cell = null;

            Row row = sheet.createRow(sheet.getLastRowNum() + 1);

            Set<Integer> keys = statistics.keySet();

            cell = row.createCell(0);

            cell.setCellStyle(styles.get("total"));

            cell.setCellValue("合计");



            for (Integer key : keys)

            {

                cell = row.createCell(key);

                cell.setCellStyle(styles.get("total"));

                cell.setCellValue(DOUBLE_FORMAT.format(statistics.get(key)));

            }

            statistics.clear();

        }

    }



    /**

@@ -877,7 +972,7 @@
                    {

                        if ((Double) val % 1 > 0)

                        {

                            val = new DecimalFormat("0.00").format(val);

                            val = new BigDecimal(val.toString());

                        }

                        else

                        {

@@ -906,4 +1001,4 @@
        }

        return val;

    }

}
}