java8 strem的collector.toMap根据指定字段分组

最新推荐文章于 2025-03-06 14:43:11 发布

前端-鸡天帝

最新推荐文章于 2025-03-06 14:43:11 发布

阅读量2.1k

点赞数

文章标签： java 开发语言

本文链接：https://blog.csdn.net/qq_38349413/article/details/124748796

版权

本文探讨了如何使用Java 8的Stream API中Collectors.toMap方法，针对Student对象按性别分组，并解决键值重复合并的问题。通过实例代码和异常解析，展示了如何利用Function接口实现合并逻辑。最后，介绍了groupingBy作为替代方案的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文阐述用strem的collector.toMap根据指定字段分组
老规矩，上代码，从代码看问题

先new一个辅助文本的“助教”实体类

import io.swagger.annotations.ApiModelProperty;
import lombok.Builder;
import lombok.Data;

import java.io.Serializable;

/**
 * @author Mikey
 * @version 1.0
 * @date 2022/5/13 10:32
 * @description
 */
@Data
@Builder
public class Student implements Serializable {
    @ApiModelProperty("姓名")
    private String name;
    @ApiModelProperty("性别 0-man 1-woman")
    private Integer gender;
}

分组逻辑，根据性别分组，为了直观一些就直接返回student而是返回name，衍生出问题

import com.example.demo.dto.Student;
import org.springframework.util.CollectionUtils;

import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @author Mikey
 * @version 1.0
 * @date 2022/5/13 10:37
 * @description
 */
public class CollectorDemo {

    public static void main(String[] args){
        // 初始化数据
        List<Student> list = init();
        Map<Integer, String> genderMap = list.parallelStream().collect(Collectors.toMap(Student::getGender, Student::getName));
    }

    public static List<Student> init() {
        // 这里的student根据业务走db
        Student stu1 = Student.builder().name("花木兰").gender(1).build();
        Student stu2 = Student.builder().name("凯茵").gender(0).build();
        Student stu3 = Student.builder().name("梦奇").gender(0).build();
        Student stu4 = Student.builder().name("关羽").gender(0).build();
        Student stu5 = Student.builder().name("露娜").gender(1).build();
        return Arrays.asList(stu1, stu2, stu3, stu4, stu5);
    }

}

以上代码采用了Collectors.toMap指定性别分组，然后名字做结果集，期望输出类似{“0”:[“凯茵”,“关羽”,…],“1”:[“花木兰”,…]}，兴高采烈执行完后给你抛了一个异常，完了芭比Q了，如下
在这里插入图片描述
根据抛出的异常，打开源码，注释大致意思重复键需要提供一个值合并函数，就是性别代码为0的有凯茵，梦奇，关羽，当你的map key为0加了凯茵后，到梦奇了，你要如何取值

看了注释后，改造代码
PS：这里其实还需要思考一下如果出现重复值要如何合并，是过滤还是如何重复加。

 Map<Integer, String> genderMap1 = list.parallelStream().collect(Collectors.toMap(Student::getGender, Student::getName,(var1, var2) -> String.format("%s,%s", var1, var2)));

输出结果如下，
{0=凯茵,梦奇,关羽, 1=花木兰,露娜}

讲到这里本文差不多结束了，大家别仅限于Collectors.toMap，如果碰到问题没办法处理，有时其实也可以尝试下别的方案，办法总比问题多的，如Collectors.groupingBy

Map<Integer, List<Student>> genderMap2 = list.stream().collect(Collectors.groupingBy(Student::getGender));

结尾，end，拿下
在这里插入图片描述