1. 使用groupBy時,如果分組的 key 為 null,會拋出異常,可以寫如下工具類規避這個問題:
import java.util.ArrayList; import java.util.Collections; import java.util.List; import java.util.Map; import java.util.function.Function; import java.util.stream.Collector; import java.util.stream.Collectors; /** * Java8 stream操作 GroupBy 設置鍵允許為null */ public class StreamGroupByUtil { public static <T, A> Collector<T, ?, Map<A, List<T>>> groupByWithNullKeys(Function<? super T, ? extends A> classifier) { return Collectors.toMap( classifier, Collections::singletonList, (List<T> oldList, List<T> newEl) -> { List<T> newList = new ArrayList<>(oldList.size() + 1); newList.addAll(oldList); newList.addAll(newEl); return newList; } ); } }
2. 使用方式:
Map<Integer, List<ClassA>> map = classAList.stream()
.collect(StreamGroupByUtil.groupByWithNullKeys(ClassA::getParentId));