1. 程式人生 > 實用技巧 >java流stream中的collect()方法詳解

java流stream中的collect()方法詳解

public class StreamTest {

    /**
     * stream.collect() 的本質由三個引數構成,
     * 1. Supplier 生產者, 返回最終結果
     * 2. BiConsumer<R, ? super T> accumulator 累加器
     * 第一個引數是要返回的集合, 第二個引數是遍歷過程中的每個元素,
     * 將流中每個被遍歷的元素新增到集合中
     * 3. BiConsumer<R, R> combiner 合併器, 在有並行流的時候才會有用, 一個流時程式碼不會走到這裡
     * 將第二步遍歷得到的所有流形成的list都新增到最終的list中,
     * 最後返回list1
     */
    @Test
    public void Test() {
        Stream<String> stream = Stream.of("hello", "world", "helloworld");
        // 最原始和基礎的方式
        /*
        List<String> list = stream.collect(
                ()->new ArrayList(),
                (theList, item) -> theList.add(item),
                (list1, list2) -> list1.addAll(list2)
        );
        */
        // 打印出更詳盡的過程
        List<String> listDetail = stream.collect(
                () -> {
                    ArrayList<String> arrayList = new ArrayList<>();
                    System.out.println("第一個list誕生, size: " + arrayList.size());
                    return arrayList;
                },
                (theList, item) -> {
                    System.out.println("第二個list的size: " + theList.size());
                    theList.add(item);
                },
                (list1, list2) -> {
                    System.out.println("第三個list1的size: " + list1.size());
                    System.out.println("第四個list2的size: " + list2.size());
                    list1.addAll(list2);
                }
        );
        /* 輸出
            第一個list誕生, size: 0
            第二個list的size: 0
            第二個list的size: 1
            第二個list的size: 2
        * */
        // 使用方法引用來傳遞行為, 更加清晰易懂, new(新建) -> add(累加) -> addAll(合併)
        List<String> list2 = stream.collect(LinkedList::new, LinkedList::add, LinkedList::addAll);
        String concat = stream.collect(StringBuilder::new, StringBuilder::append, StringBuilder::append).toString();
        System.out.println(concat);
    }
    
    @Test
    public void Test2() {
        Stream<String> stream = Stream.of("hello", "world", "helloworld");
        // 這樣的寫法兼具靈活和簡單
        ArrayList<String> list = stream.collect(Collectors.toCollection(ArrayList::new));
        TreeSet<String> treeSet = stream.collect(Collectors.toCollection(TreeSet::new));
        String s = stream.collect(Collectors.joining()); // 拼接成字串
        HashMap<String, String> map = stream.collect(HashMap::new, (x, y) -> {
            x.put(y, y); // 自己做自己的key
        }, HashMap::putAll);
    }
}

 

原文