java流式分组(groupBy)生成TreeMap

发布时间 2024-01-04 23:43:32作者: 技术扫地生—楼上老刘

java流式分组(groupBy)生成TreeMap

哈哈哈,扫地生(saodisheng/sds)又回来啦。过去一坤年,啊不!应该是近一年,很少在博客上分享个人的学习笔记和平时遇到的疑难杂症了。而今,SDS 重新更新啦。

今天要记录的是一个关于流式分组 groupBy生成有序map的过程;

具体过程如下:

package com.sds.web.review;

import lombok.Data;
import lombok.experimental.Accessors;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import java.util.stream.Collectors;

/**
 * @author saodisheng
 */
public class StreamGroupByTreeMapTest {

    @Data
    @Accessors(chain = true)
    static class SdsTest {
        private String key;
        private String val;
    }

    @Test
    @DisplayName("默认分组结果")
    public void test1() {
        List<SdsTest> list = new ArrayList<SdsTest>() {
            {
                add(new SdsTest().setKey("sds2").setVal("sds2"));
                add(new SdsTest().setKey("sds1").setVal("sds1"));
            }
        };
        System.out.println("===================== 默认分组结果 ======================");
        Map<String, List<SdsTest>> defaultMap = list.stream().collect(Collectors.groupingBy(SdsTest::getKey));
        System.out.println(defaultMap.toString());
    }

    @Test
    @DisplayName("指定返回TreeMap")
    public void test2() {
        List<SdsTest> list = new ArrayList<SdsTest>() {
            {
                add(new SdsTest().setKey("sds2").setVal("sds2"));
                add(new SdsTest().setKey("sds1").setVal("sds1"));
            }
        };
        System.out.println("===================== 指定返回TreeMap ======================");
        TreeMap<String, List<SdsTest>> treeMap = list.stream().collect(Collectors.groupingBy(SdsTest::getKey, TreeMap::new, Collectors.toList()));
        System.out.println(treeMap.toString());
    }

}

image-20240104233730160