Excel注解支持自定义数据处理器

RuoYi

2021-09-22 0a104689ed76e9ab7a98270c66f82d3fc26a9305

Excel注解支持自定义数据处理器

 ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/annotation/Excel.java

@@ -5,6 +5,7 @@
import java.lang.annotation.RetentionPolicy;

import java.lang.annotation.Target;

import java.math.BigDecimal;

import com.ruoyi.common.core.utils.poi.ExcelHandlerAdapter;



/**

 * 自定义导出Excel数据注解

@@ -103,7 +104,17 @@
    /**

     * 导出字段对齐方式（0：默认；1：靠左；2：居中；3：靠右）

     */

    Align align() default Align.AUTO;

    public Align align() default Align.AUTO;



    /**

     * 自定义数据处理器

     */

    public Class<?> handler() default ExcelHandlerAdapter.class;



    /**

     * 自定义数据处理器参数

     */

    public String[] args() default {};



    public enum Align

    {


 ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/utils/poi/ExcelHandlerAdapter.java

New file
@@ -0,0 +1,19 @@
package com.ruoyi.common.utils.poi;



/**

 * Excel数据格式处理适配器

 * 
 * @author ruoyi

 */

public interface ExcelHandlerAdapter

{

    /**

     * 格式化

     * 
     * @param value 单元格数据值

     * @param args excel注解args参数组

     *

     * @return 处理后的值

     */

    Object format(Object value, String[] args);

}


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

@@ -4,6 +4,7 @@
import java.io.InputStream;

import java.io.OutputStream;

import java.lang.reflect.Field;

import java.lang.reflect.Method;

import java.math.BigDecimal;

import java.text.DecimalFormat;

import java.util.ArrayList;

@@ -307,6 +308,10 @@
                        else if (StringUtils.isNotEmpty(attr.readConverterExp()))

                        {

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

                        }

                        else if (!attr.handler().equals(ExcelHandlerAdapter.class))

                        {

                            val = dataFormatHandlerAdapter(val, attr);

                        }

                        ReflectUtils.invokeSetter(entity, propertyName, val);

                    }

@@ -633,6 +638,10 @@
                {

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

                }

                else if (!attr.handler().equals(ExcelHandlerAdapter.class))

                {

                    cell.setCellValue(dataFormatHandlerAdapter(value, attr));

                }

                else

                {

                    // 设置列类型

@@ -780,6 +789,28 @@
    }



    /**

     * 数据处理器

     * 
     * @param value 数据值

     * @param excel 数据注解

     * @return

     */

    public String dataFormatHandlerAdapter(Object value, Excel excel)

    {

        try

        {

            Object instance = excel.handler().newInstance();

            Method formatMethod = excel.handler().getMethod("format", new Class[] { Object.class, String[].class });

            value = formatMethod.invoke(instance, value, excel.args());

        }

        catch (Exception e)

        {

            log.error("不能格式化数据 " + excel.handler(), e.getMessage());

        }

        return Convert.toStr(value);

    }



    /**

     * 合计统计信息

     */

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