支持自定义显示Excel属性列

RuoYi

2024-11-07 f381ce3fe1dc281b49bbaaa4d4a5d496630eae5d

支持自定义显示Excel属性列

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

@@ -195,6 +195,11 @@
    public Class<T> clazz;



    /**

     * 需要显示列属性

     */

    public String[] includeFields;



    /**

     * 需要排除列属性

     */

    public String[] excludeFields;

@@ -205,10 +210,19 @@
    }



    /**

     * 仅在Excel中显示列属性

     *

     * @param fields 列属性名 示例[单个"name"/多个"id","name"]

     */

    public void showColumn(String... fields)

    {

        this.includeFields = fields;

    }



    /**

     * 隐藏Excel中列属性

     *

     * @param fields 列属性名 示例[单个"name"/多个"id","name"]

     * @throws Exception

     */

    public void hideColumn(String... fields)

    {

@@ -1488,46 +1502,86 @@
        List<Field> tempFields = new ArrayList<>();

        tempFields.addAll(Arrays.asList(clazz.getSuperclass().getDeclaredFields()));

        tempFields.addAll(Arrays.asList(clazz.getDeclaredFields()));

        for (Field field : tempFields)

        if (StringUtils.isNotEmpty(includeFields))

        {

            if (!ArrayUtils.contains(this.excludeFields, field.getName()))

            for (Field field : tempFields)

            {

                // 单注解

                if (field.isAnnotationPresent(Excel.class))

                if (ArrayUtils.contains(this.includeFields, field.getName()) || field.isAnnotationPresent(Excels.class))

                {

                    Excel attr = field.getAnnotation(Excel.class);

                    if (attr != null && (attr.type() == Type.ALL || attr.type() == type))

                    addField(fields, field);

                }

            }

        }

        else if (StringUtils.isNotEmpty(excludeFields))

        {

            for (Field field : tempFields)

            {

                if (!ArrayUtils.contains(this.excludeFields, field.getName()))

                {

                    addField(fields, field);

                }

            }

        }

        else

        {

            for (Field field : tempFields)

            {

                addField(fields, field);

            }

        }

        return fields;

    }



    /**

     * 添加字段信息

     */

    public void addField(List<Object[]> fields, Field field)

    {

        // 单注解

        if (field.isAnnotationPresent(Excel.class))

        {

            Excel attr = field.getAnnotation(Excel.class);

            if (attr != null && (attr.type() == Type.ALL || attr.type() == type))

            {

                field.setAccessible(true);

                fields.add(new Object[] { field, attr });

            }

            if (Collection.class.isAssignableFrom(field.getType()))

            {

                subMethod = getSubMethod(field.getName(), clazz);

                ParameterizedType pt = (ParameterizedType) field.getGenericType();

                Class<?> subClass = (Class<?>) pt.getActualTypeArguments()[0];

                this.subFields = FieldUtils.getFieldsListWithAnnotation(subClass, Excel.class);

            }

        }



        // 多注解

        if (field.isAnnotationPresent(Excels.class))

        {

            Excels attrs = field.getAnnotation(Excels.class);

            Excel[] excels = attrs.value();

            for (Excel attr : excels)

            {

                if (StringUtils.isNotEmpty(includeFields))

                {

                    if (ArrayUtils.contains(this.includeFields, field.getName() + "." + attr.targetAttr())

                            && (attr != null && (attr.type() == Type.ALL || attr.type() == type)))

                    {

                        field.setAccessible(true);

                        fields.add(new Object[] { field, attr });

                    }

                    if (Collection.class.isAssignableFrom(field.getType()))

                    {

                        subMethod = getSubMethod(field.getName(), clazz);

                        ParameterizedType pt = (ParameterizedType) field.getGenericType();

                        Class<?> subClass = (Class<?>) pt.getActualTypeArguments()[0];

                        this.subFields = FieldUtils.getFieldsListWithAnnotation(subClass, Excel.class);

                    }

                }



                // 多注解

                if (field.isAnnotationPresent(Excels.class))

                else

                {

                    Excels attrs = field.getAnnotation(Excels.class);

                    Excel[] excels = attrs.value();

                    for (Excel attr : excels)

                    if (!ArrayUtils.contains(this.excludeFields, field.getName() + "." + attr.targetAttr())

                            && (attr != null && (attr.type() == Type.ALL || attr.type() == type)))

                    {

                        if (!ArrayUtils.contains(this.excludeFields, field.getName() + "." + attr.targetAttr())

                                && (attr != null && (attr.type() == Type.ALL || attr.type() == type)))

                        {

                            field.setAccessible(true);

                            fields.add(new Object[] { field, attr });

                        }

                        field.setAccessible(true);

                        fields.add(new Object[] { field, attr });

                    }

                }

            }

        }

        return fields;

    }



    /**