Java Stream不用死记硬背，公司技术大牛封装的工具类！牛的不行

yuhaiqiang2023-11-27 7:09

前公司有技术大佬对Java Stream 进行二次封装，使用起来非常爽，全公司都在用。

我自己照着写了一遍，改了名字。

一共整理了10个工具方法，可以满足 Collection、List、Set、Map 之间各种类型转化。例如

  1. 将 Collection<OrderItem> 转化为 List<OrderItem>
  2. 将 Collection<OrderItem> 转化为 Set<OrderItem>
  3. 将 List<OrderItem> 转化为 List<Long>
  4. 将 Set<OrderItem> 转化为 Set<Long>
  5. 将 Collection<OrderItem> 转化为 List<Long>
  6. 将 Collection<OrderItem> 转化为 Set<Long>
  7. 从Collection<OrderItem>中提取 Key, Map 的 Value 就是类型 OrderItem
  8. 从Collection<OrderItem>中提取 Key, Map 的 Value 根据 OrderItem 类型进行转化。
  9. 将 Map<Long, OrderItem> 中的value 转化为 Map<Long, Double>
  10. value 转化时，lamada表达式可以使用（v)->{}， 也可以使用 （k，v）->{ }。

看 Collection 集合类型到 Map类型的转化。

Collection 转化为 Map

由于 List 和 Set 是 Collection 类型的子类，所以只需要实现Collection 类型转化为 Map 类型即可。 Collection转化为 Map 共分两个方法

  1. 从Collection<OrderItem>到 Map<Key, OrderItem>，提取 Key, Map 的 Value 就是类型 OrderItem
  2. 从Collection<OrderItem>到 Map<Key，Value> ，提取 Key, Map 的 Value 根据 OrderItem 类型进行转化。

使用样例

代码示例中把Set<OrderItem> 转化为 Map<Long, OrderItem> 和 Map<Long ,Double>。

Java 复制代码 
@Test
public void testToMap() {
    Collection<OrderItem> collection = coll;
    Set<OrderItem> set = toSet(collection);

    Map<Long, OrderItem> map = toMap(set, OrderItem::getOrderId);
}

@Test
public void testToMapV2() {
    Collection<OrderItem> collection = coll;
    Set<OrderItem> set = toSet(collection);

    Map<Long, Double> map = toMap(set, OrderItem::getOrderId, OrderItem::getActPrice);
}

代码展示

Java 复制代码 
public static <T, K> Map<K, T> toMap(Collection<T> collection, Function<? super T, ? extends K> keyMapper) {
    return toMap(collection, keyMapper, Function.identity());
}

public static <T, K, V> Map<K, V> toMap(Collection<T> collection,
                                        Function<? super T, ? extends K> keyFunction,
                                        Function<? super T, ? extends V> valueFunction) {
    return toMap(collection, keyFunction, valueFunction, pickLast());
}

public static <T, K, V> Map<K, V> toMap(Collection<T> collection,
                                        Function<? super T, ? extends K> keyFunction,
                                        Function<? super T, ? extends V> valueFunction,
                                        BinaryOperator<V> mergeFunction) {
    if (CollectionUtils.isEmpty(collection)) {
        return new HashMap<>(0);
    }

    return collection.stream().collect(Collectors.toMap(keyFunction, valueFunction, mergeFunction));
}

public static <T> BinaryOperator<T> pickFirst() {
    return (k1, k2) -> k1;
}
public static <T> BinaryOperator<T> pickSecond() {
    return (k1, k2) -> k2;
}

Map格式转换

转换 Map 的 Value

  1. 将 Map<Long, OrderItem> 中的value 转化为 Map<Long, Double>
  2. value 转化时，lamada表达式可以使用（v)->{}， 也可以使用 （k，v）->{ }。

测试样例

Java 复制代码 
@Test
public void testConvertValue() {
    Collection<OrderItem> collection = coll;
    Set<OrderItem> set = toSet(collection);

    Map<Long, OrderItem> map = toMap(set, OrderItem::getOrderId);

    Map<Long, Double> orderId2Price = convertMapValue(map, item -> item.getActPrice());
    Map<Long, String> orderId2Token = convertMapValue(map, (id, item) -> id + item.getName());

}

代码展示

Java 复制代码 
public static <K, V, C> Map<K, C> convertMapValue(Map<K, V> map, 
                        BiFunction<K, V, C> valueFunction,
                        BinaryOperator<C> mergeFunction) {
    if (isEmpty(map)) {
        return new HashMap<>();
    }
    return map.entrySet().stream().collect(Collectors.toMap(
            e -> e.getKey(),
            e -> valueFunction.apply(e.getKey(), e.getValue()),
            mergeFunction
    ));
}

public static <K, V, C> Map<K, C> convertValue(Map<K, V> originMap, BiFunction<K, V, C> valueConverter) {
    return convertValue(originMap, valueConverter, Lambdas.pickLast());
}

public static <T> BinaryOperator<T> pickFirst() {
    return (k1, k2) -> k1;
}
public static <T> BinaryOperator<T> pickSecond() {
    return (k1, k2) -> k2;
}

集合类型转化

Collection 和 List、Set 的转化

  1. 将 Collection<OrderItem> 转化为 List<OrderItem>
  2. 将 Collection<OrderItem> 转化为 Set<OrderItem>
js 复制代码 
public static <T> List<T> toList(Collection<T> collection) {
    if (collection == null) {
        return new ArrayList<>();
    }
    if (collection instanceof List) {
        return (List<T>) collection;
    }
    return collection.stream().collect(Collectors.toList());
}

public static <T> Set<T> toSet(Collection<T> collection) {
    if (collection == null) {
        return new HashSet<>();
    }
    if (collection instanceof Set) {
        return (Set<T>) collection;
    }
    return collection.stream().collect(Collectors.toSet());
}

测试样例

Java 复制代码 
@Test//将集合 Collection 转化为 List
public void testToList() {
    Collection<OrderItem> collection = coll;
    List<OrderItem> list = toList(coll);
}

@Test//将集合 Collection 转化为 Set
public void testToSet() {
    Collection<OrderItem> collection = coll;
    Set<OrderItem> set = toSet(collection);
}

List和 Set 是 Collection 集合类型的子类，所以无需再转化。

List、Set 类型之间的转换

业务中有时候需要将 List<A> 转化为 List<B>。如何实现工具类呢？

Java 复制代码 
public static <T, R> List<R> map(List<T> collection, Function<T, R> mapper) {
    return collection.stream().map(mapper).collect(Collectors.toList());
}

public static <T, R> Set<R> map(Set<T> collection, Function<T, R> mapper) {
    return collection.stream().map(mapper).collect(Collectors.toSet());
}

public static <T, R> List<R> mapToList(Collection<T> collection, Function<T, R> mapper) {
    return collection.stream().map(mapper).collect(Collectors.toList());
}

public static <T, R> Set<R> mapToSet(Collection<T> collection, Function<T, R> mapper) {
    return collection.stream().map(mapper).collect(Collectors.toSet());
}

测试样例

  1. 将 List<OrderItem> 转化为 List<Long>
  2. 将 Set<OrderItem> 转化为 Set<Long>
  3. 将 Collection<OrderItem> 转化为 List<Long>
  4. 将 Collection<OrderItem> 转化为 Set<Long>
Java 复制代码 
@Test
public void testMapToList() {
    Collection<OrderItem> collection = coll;
    List<OrderItem> list = toList(coll);

    List<Long> orderIdList = map(list, (item) -> item.getOrderId());
}

@Test
public void testMapToSet() {
    Collection<OrderItem> collection = coll;
    Set<OrderItem> set = toSet(coll);

    Set<Long> orderIdSet = map(set, (item) -> item.getOrderId());
}

@Test
public void testMapToList2() {
    Collection<OrderItem> collection = coll;

    List<Long> orderIdList = mapToList(collection, (item) -> item.getOrderId());
}

@Test
public void testMapToSetV2() {
    Collection<OrderItem> collection = coll;
    
    Set<Long> orderIdSet = mapToSet(collection, (item) -> item.getOrderId());

}

总结一下 以上样例包含了如下的映射场景

  1. 将 Collection<OrderItem> 转化为 List<OrderItem>
  2. 将 Collection<OrderItem> 转化为 Set<OrderItem>
  3. 将 List<OrderItem> 转化为 List<Long>
  4. 将 Set<OrderItem> 转化为 Set<Long>
  5. 将 Collection<OrderItem> 转化为 List<Long>
  6. 将 Collection<OrderItem> 转化为 Set<Long>
  7. 从Collection<OrderItem>中提取 Key, Map 的 Value 就是类型 OrderItem
  8. 从Collection<OrderItem>中提取 Key, Map 的 Value 根据 OrderItem 类型进行转化。
  9. 将 Map<Long, OrderItem> 中的value 转化为 Map<Long, Double>
  10. value 转化时，lamada表达式可以使用（v)->{}， 也可以使用 （k，v）->{ }。
相关推荐
浮游本尊30 分钟前
一次合同同步背后的多阶段流水线：从外部主数据到本地歧义消解
后端
lv__pf31 分钟前
springboot原理
java·spring boot·后端
段小二1 小时前
服务一重启全丢了——Spring AI Alibaba Agent 三层持久化完整方案
java·后端
UIUV2 小时前
Go语言入门到精通学习笔记
后端·go·编程语言
lizhongxuan2 小时前
开发 Agent 的坑
后端
段小二2 小时前
Agent 自动把机票改错了，推理完全正确——这才是真正的风险
java·后端
itjinyin2 小时前
ShardingSphere-jdbc 5.5.0 + spring boot 基础配置 - 实战篇
java·spring boot·后端
Victor3562 小时前
MongoDB（91）如何在MongoDB中使用TTL索引？
后端
老王以为2 小时前
前端重生之 - 前端视角下的 Python
前端·后端·python
热门推荐
012026年4月技术前沿：AI大模型爆发、智能体革命与量子安全新纪元02GitHub 镜像站点032026 年 AI 编程助手全面对比评测：Cursor vs Copilot vs Claude Code vs GitHub Copilot Free04Claude Code Windows 兼容性问题：指定版本 2.1.112 可解决05AI Weekly | 2026年4月第二周 · GitHub热门项目与AI发展趋势深度解析06UBUNTU Claude Code 报错 claude native binary not installed07从限购到畅通：GLM-5.1 Coding Plan接入攻略08GPT-6发布日深度解析-Symphony架构200万Token实战09从零部署 Hermes Agent：一只"会成长的 AI 马"保姆级安装教程10零成本！Ollama本地部署国产大模型全指南（支持Kimi-K2.5/GLM-5/Qwen，新手秒上手）