技术总监写的十个方法，让我精通了lambda表达式 前公司的技术总监写了工具类，对Java Stream 进行二次封装，使用起来非常爽，全公司都在用。
我自己照着写了一遍，改了名字，分享给大家。
一共整理了10个工具方法，可以满足 Collection、List、Set、Map 之间各种类型转化。例如
151
0Collection<OrderItem> 转化为 List<OrderItem>Collection<OrderItem> 转化为 Set<OrderItem>List<OrderItem> 转化为 List<Long>Set<OrderItem> 转化为 Set<Long>Collection<OrderItem> 转化为 List<Long>Collection<OrderItem> 转化为 Set<Long>Collection<OrderItem>中提取 Key, Map 的 Value 就是类型 OrderItemCollection<OrderItem>中提取 Key, Map 的 Value 根据 OrderItem 类型进行转化。Map<Long, OrderItem> 中的value 转化为 Map<Long, Double> （v)->{}， 也可以使用 （k，v）->{ }。看 Collection 集合类型到 Map类型的转化。
由于 List 和 Set 是 Collection 类型的子类，所以只需要实现Collection 类型转化为 Map 类型即可。
Collection转化为 Map 共分两个方法
Collection<OrderItem>到 Map<Key, OrderItem>，提取 Key, Map 的 Value 就是类型 OrderItemCollection<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);
}
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, pickSecond());
}
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;
}
@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());
}
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> convertMapValue(Map<K, V> originMap, BiFunction<K, V, C> valueConverter) {
return convertMapValue(originMap, valueConverter, Lambdas.pickSecond());
}
public static <T> BinaryOperator<T> pickFirst() {
return (k1, k2) -> k1;
}
public static <T> BinaryOperator<T> pickSecond() {
return (k1, k2) -> k2;
}
Collection<OrderItem> 转化为 List<OrderItem>Collection<OrderItem> 转化为 Set<OrderItem>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());
}
@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<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());
}
List<OrderItem> 转化为 List<Long>Set<OrderItem> 转化为 Set<Long>Collection<OrderItem> 转化为 List<Long>Collection<OrderItem> 转化为 Set<Long>@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());
}
总结一下 以上样例包含了如下的映射场景
Collection<OrderItem> 转化为 List<OrderItem>Collection<OrderItem> 转化为 Set<OrderItem>List<OrderItem> 转化为 List<Long>Set<OrderItem> 转化为 Set<Long>Collection<OrderItem> 转化为 List<Long>Collection<OrderItem> 转化为 Set<Long>Collection<OrderItem>中提取 Key, Map 的 Value 就是类型 OrderItemCollection<OrderItem>中提取 Key, Map 的 Value 根据 OrderItem 类型进行转化。Map<Long, OrderItem> 中的value 转化为 Map<Long, Double> （v)->{}， 也可以使用 （k，v）->{ }。免责申明：
本文系转载，版权归原作者所有，如若侵权请联系我们进行删除！
@Copyrights 2016-2023 杭州玳数科技有限公司 浙ICP备15044486号-1 浙公网安备33011002011932号
