Java8 stream 操作 GroupBy 設置鍵允許為null


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));

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM