EasyExcel自定义导出及导出顺序

1.加入依赖

<dependency>        
    <groupId>com.alibaba</groupId>
    <artifactId>easyexcel</artifactId>
    <version>3.3.3</version>
</dependency>

2.封装工具类

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.annotation.ExcelProperty;

import java.io.OutputStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;

/**
 * easyExcel 工具类
 *
 * @author 
 */
public class EasyExcelUtils {

    /**
     * 导出数据, 导出字段顺序/表头在注解ExcelProperty中定义
     *
     * @param outputStream 输出流
     * @param clazz        数据类型
     * @param data         数据集合
     */
    public static void write(
            OutputStream outputStream, Class<?> clazz, Collection<?> data
    ) {
        EasyExcel
                .write(outputStream, clazz)
                .sheet()
                .doWrite(data);
    }

    /**
     * 导出数据, 自定义 列和顺序/表头
     *
     * @param outputStream 输出流
     * @param clazz        数据类型
     * @param data         数据集合
     * @param column       需要导出的列
     * @param head         自定义excel表头
     */
    public static void write(
            OutputStream outputStream, Class<?> clazz, Collection<?> data,
            Collection<String> column, Collection<String> head
    ) {
        EasyExcel
                .write(outputStream, clazz)
                .sheet()
                .head(head(head))
                .includeColumnFieldNames(column)
                .doWrite(dataList(data, column));
    }

    /**
     * 导出数据, 自定义导出列和顺序 <br/>
     * excel表头需要从ExcelProperty中获取
     *
     * @param outputStream 输出流
     * @param clazz        数据类型
     * @param data         数据集合
     * @param column       需要导出的列
     */
    public static void write(
            OutputStream outputStream, Class<?> clazz, Collection<?> data,
            Collection<String> column
    ) {
        EasyExcel
                .write(outputStream, clazz)
                .sheet()
                .head(head(clazz, column))
                .includeColumnFieldNames(column)
                .doWrite(dataList(data, column));
    }

    /**
     * 按照指定头列表设置Excel头
     *
     * @param headList Excel头信息
     * @return
     */
    private static List<List<String>> head(Collection<String> headList) {
        List<List<String>> list = new ArrayList<>();
        for (String value : headList) {
            List<String> head = new ArrayList<>();
            head.add(value);
            list.add(head);
        }
        return list;
    }

    /**
     * 将注解中的头按传入列的顺序返回
     *
     * @param clazz      指定类
     * @param columnList 传入的列顺序
     * @return
     */
    public static List<List<String>> head(Class<?> clazz, Collection<String> columnList) {
        List<List<String>> list = new ArrayList<>();

        for (String column : columnList) {
            try {
                Field field = clazz.getDeclaredField(column);
                String[] value = field.getAnnotation(ExcelProperty.class).value();
                List<String> collect = Arrays.stream(value).collect(Collectors.toList());
                list.add(collect);
            } catch (NoSuchFieldException e) {
                e.printStackTrace();
            }
        }
        return list;
    }

    /**
     * 设置表格信息
     *
     * @param dataList 查询出的数据
     * @param column   需要显示的字段
     * @return
     */
    private static List<List<Object>> dataList(Collection<?> dataList, Collection<String> column) {
        List<List<Object>> list = new ArrayList<>();
        for (Object person : dataList) {
            List<Object> data = new ArrayList<>();
            for (String fieldName : column) {
                // 通过反射根据需要显示的字段，获取对应的属性值
                data.add(getFieldValue(fieldName, person));
            }
            list.add(data);
        }
        return list;
    }


    /**
     * 根据传入的字段获取对应的get方法，如name,对应的getName方法
     *
     * @param fieldName 字段名
     * @param obj       对象
     * @return
     */
    private static Object getFieldValue(String fieldName, Object obj) {
        try {
            String firstUpper = fieldName.substring(0, 1).toUpperCase();
            String getter = "get" + firstUpper + fieldName.substring(1);
            Method method = obj.getClass().getMethod(getter);
            return method.invoke(obj);
        } catch (Exception e) {
            return null;
        }
    }

}