手写业务数据处理小工具

本文源码可运行的 JDK 版本要求：JDK 1.8 +

本文源码参见：https://github.com/woodwhales/woodwhales-common

一般在业务开发中，开发者会先通过 ORM 框架查询数据库数据得到数据的 List 集合，再该业务数据集合做如下处理：

List 集合转 Map 集合

按照 List 集合元素中某个属性将 List 集合转 Map 集合

在 JDK 1.8 版本中可以使用 stream 快速将 list 转 map：

// 第一种 直接返回user本身
Map<Long, User> map = userList.stream().collect(Collectors.toMap(User::getId, user -> user));

// 第二种 Function中有一个static方法identity 返回本身
Map<Long, User> map = userList.stream().collect(Collectors.toMap(User::getId, Function.identity()));

// 如果list中的age存在相同的时候,转化map的时候就会出错Duplicate key
// toMap的重载,定义key2覆盖key1的值
Map<Integer, Users> map3 = list.stream().collect(Collectors.toMap(Users::getId, Function.identity(), (key1,key2) -> key2));

笔者利用上述代码进行二次封装，简化了判空及转 map 时填写 .stream().collect… 等重复的代码。

public class DataTool {

    /**
     * list 转 map 集合
     * @param source 数据源集合
     * @param keyMapper map 集合中的 key 获取规则
     * @param valueMapper map 集合中的 value 获取规则
     * @param <K> map 集合中的 key 类型
     * @param <S> 数据源集合中元素的类型
     * @param <T> map 集合中的 value 类型
     * @return
     */
    public static <K, S, T> Map<K,T> toMap(List<S> source,
                                           Function<? super S, ? extends K> keyMapper,
                                           Function<? super S, ? extends T> valueMapper) {
        if(null == source || source.size() == 0) {
            return Collections.emptyMap();
        }

        return source.stream().collect(Collectors.toMap(keyMapper, valueMapper));
    }
}

源码参见：src/main/java/org/woodwhales/business/DataTool.java

List 集合分组

List 集合按照某种规则进行分组

在 JDK 1.8 版本中可以使用 stream 快速将 list 分组：

Map<String, List<User>> group = list.stream().collect(Collectors.groupingBy(User::getType));

笔者利用上述代码进行二次封装，简化了判空及转 map 时填写 .stream().collect… 等重复的代码。

public class DataTool {

    /**
     * 将 list 集合分组
     * @param source 数据源集合
     * @param classifier 分组规则
     * @param <K> map 集合中的 key 类型
     * @param <S> map 集合中的 value 类型
     * @return
     */
    public static <K, S> Map<K, List<S>> groupingBy(List<S> source,
                                                    Function<? super S, ? extends K> classifier) {
        if(null == source || source.size() == 0) {
            return Collections.emptyMap();
        }

        return source.stream().collect(Collectors.groupingBy(classifier));
    }
}

源码参见：src/main/java/org/woodwhales/business/DataTool.java

业务数据去重

List 集合按照指定属性去重（其中可能存在要去重的属性不存在，即“无效的”数据）。

利用 Map 集合存入元素方法 put() 的特性：已存在则返回旧的元素，不存在则返回空。

public class DataTool {
    /**
     * 对集合数据进行去重器
     * 非线程安全
     * @param source 数据源集合
     * @param deduplicateInterface 去重器接口
     * @param <K> 去重属性的类型
     * @param <T> 数据源集合中元素的类型
     * @return
     */
    public static <K, T> DeduplicateResult<T> deduplicate(List<T> source,
                                                          DeduplicateInterface<K, T> deduplicateInterface) {
        if(null == source || source.size() == 0) {
            return new DeduplicateResult<T>(source, emptyList(), emptyList(), emptyList());
        }

        Map<K, T> container =  new LinkedHashMap<>();
        // 无效的数据集合
        List<T> invalidList = new LinkedList<>();
        // 重复的数据集合
        List<T> repetitiveList = new LinkedList<>();

        for (T data : source) {
            if (!deduplicateInterface.isValid(data)) {
                invalidList.add(data);
            } else {
                K deduplicatedKey = deduplicateInterface.getDeduplicatedKey(data);
                T putData = container.put(deduplicatedKey, data);
                if(Objects.nonNull(putData)) {
                    repetitiveList.add(putData);
                }
            }
        }

        // 已去重的数据集合
        List<T> deduplicatedList = new ArrayList<>(container.values());

        return new DeduplicateResult<T>(source, invalidList, deduplicatedList, repetitiveList);
    }
}

源码参见：src/main/java/org/woodwhales/business/DataTool.java

枚举转 map 集合

根据枚举中某个属性进行枚举转 map 集合。

利用 EnumSet 类的 allOf() 方法，可以根据枚举的 Class 类型获取该枚举的所有枚举实例。当拿到集合枚举数据就可以使用 stream 进行集合转 map。

public class DataTool {
    
    /**
     * 枚举转 map 集合
     * @param sourceEnumClass 数据源枚举 Class类
     * @param keyMapper map 集合中的 key 获取规则
     * @param <K> map 集合中的 key 类型
     * @param <T> map 集合中的 value 类型
     * @return
     */
    public static <K, T extends Enum<T>> Map<K, T> enumMap(Class<T> sourceEnumClass,
                                                         Function<? super T, ? extends K> keyMapper) {
        EnumSet<T> enumSet = EnumSet.allOf(sourceEnumClass);
        return enumSet.stream().collect(Collectors.toMap(keyMapper, Function.identity()));
    }
}

源码参见：src/main/java/org/woodwhales/business/DataTool.java