首页 > Java > java教程 > 正文

Java Stream API:高效处理列表数据,按组合键去重并选择最新记录

DDD
发布: 2025-09-12 11:49:00
原创
972人浏览过

Java Stream API:高效处理列表数据,按组合键去重并选择最新记录

本文详细介绍了如何利用Java Stream API,特别是Collectors.toMap,对包含重复条目的对象列表进行高级过滤。教程将演示如何根据对象的多个字段(如姓名组合)确定唯一性,并在出现重复时,根据特定字段(如日期)选择最新或最符合条件的记录,从而实现数据的高效聚合与筛选。

业务场景与问题描述

在日常的数据处理中,我们经常会遇到需要从一个对象列表中提取唯一记录的场景。然而,这里的“唯一”可能并非指所有字段都完全相同,而是基于某个或某几个特定字段的组合。更进一步,当这些组合字段出现重复时,我们可能需要根据另一个字段(例如时间戳或版本号)来决定保留哪一条记录。

以员工数据为例,假设我们有一个Employee对象列表,其定义如下:

import lombok.Data;
import lombok.AllArgsConstructor;
import java.time.LocalDateTime;

@Data
@AllArgsConstructor
public class Employee {
   private String firstName;
   private String lastName;
   private double salary;
   private LocalDateTime getSalaryDate; // 更改为更具描述性的字段名
}
登录后复制

该列表中可能存在多条记录拥有相同的firstName和lastName,但salary和getSalaryDate(获取薪资的日期)不同。例如:

List<Employee> employees = new ArrayList<>();
employees.add(new Employee("John", "Smith", 10, LocalDateTime.of(2022, 9, 1, 0, 0)));
employees.add(new Employee("John", "Smith", 20, LocalDateTime.of(2022, 10, 1, 0, 0)));
employees.add(new Employee("John", "Smith", 5, LocalDateTime.of(2022, 11, 1, 0, 0)));
employees.add(new Employee("Kelly", "Jones", 12, LocalDateTime.of(2022, 3, 1, 0, 0)));
employees.add(new Employee("Sara", "Kim", 21, LocalDateTime.of(2022, 3, 1, 0, 0)));
employees.add(new Employee("Sara", "Kim", 7, LocalDateTime.of(2022, 7, 1, 0, 0)));
登录后复制

我们的目标是:对于每对唯一的firstName和lastName组合,只保留一条记录,且这条记录必须是getSalaryDate最新的那一条。

期望的输出结果应为:

立即学习Java免费学习笔记(深入)”;

表单大师AI
表单大师AI

一款基于自然语言处理技术的智能在线表单创建工具,可以帮助用户快速、高效地生成各类专业表单。

表单大师AI 74
查看详情 表单大师AI
  • "John", "Smith", 5, 2022-11-01
  • "Kelly", "Jones", 12, 2022-03-01
  • "Sara", "Kim", 7, 2022-07-01

使用Java Stream API进行高效过滤

Java 8引入的Stream API为处理集合数据提供了强大而灵活的工具。针对上述问题,我们可以利用Collectors.toMap结合自定义的合并函数(merge function)来实现。

核心思路

  1. 生成唯一键: 为了识别firstName和lastName的唯一组合,我们需要将这两个字段组合成一个唯一的键。最简单的方法是字符串拼接。
  2. 映射值: 将原始的Employee对象作为Map的值。
  3. 处理键冲突: 当不同的Employee对象生成了相同的键时(即firstName和lastName相同),Collectors.toMap允许我们提供一个合并函数来决定保留哪个值。在这个函数中,我们将比较两个Employee对象的getSalaryDate,保留日期最新的那个。
  4. 提取结果: 最终,从生成的Map中提取所有的值,即为我们所需的过滤后的Employee列表。

实现步骤与代码示例

import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

public class EmployeeFilterTutorial {

    // Employee 类定义(同上,为了完整性再次列出)
    @Data
    @AllArgsConstructor
    public static class Employee {
        private String firstName;
        private String lastName;
        private double salary;
        private LocalDateTime getSalaryDate;
    }

    public static void main(String[] args) {
        List<Employee> employees = new ArrayList<>();
        employees.add(new Employee("John", "Smith", 10, LocalDateTime.of(2022, 9, 1, 0, 0)));
        employees.add(new Employee("John", "Smith", 20, LocalDateTime.of(2022, 10, 1, 0, 0)));
        employees.add(new Employee("John", "Smith", 5, LocalDateTime.of(2022, 11, 1, 0, 0)));
        employees.add(new Employee("Kelly", "Jones", 12, LocalDateTime.of(2022, 3, 1, 0, 0)));
        employees.add(new Employee("Sara", "Kim", 21, LocalDateTime.of(2022, 3, 1, 0, 0)));
        employees.add(new Employee("Sara", "Kim", 7, LocalDateTime.of(2022, 7, 1, 0, 0)));

        Collection<Employee> filteredEmployees = employees.stream()
                .collect(Collectors.toMap(
                        // Key Mapper: 组合 firstName 和 lastName 作为唯一键
                        employee -> employee.getFirstName() + employee.getLastName(),
                        // Value Mapper: 将 Employee 对象本身作为值
                        Function.identity(),
                        // Merge Function: 处理键冲突,保留 getSalaryDate 最新的 Employee
                        (existingEmployee, newEmployee) -> 
                            existingEmployee.getSalaryDate().isAfter(newEmployee.getSalaryDate()) ? existingEmployee : newEmployee
                ))
                .values(); // 从 Map 中获取所有值,即为过滤后的 Employee 集合

        // 打印结果
        filteredEmployees.forEach(System.out::println);
    }
}
登录后复制

代码解析

  1. employees.stream(): 创建一个Employee对象的流。
  2. collect(Collectors.toMap(...)): 这是核心操作,用于将流中的元素收集到一个Map中。Collectors.toMap有三个参数:
    • keyMapper (employee -> employee.getFirstName() + employee.getLastName()): 这是一个Function,用于从流中的每个元素(Employee对象)中提取键。这里我们将firstName和lastName拼接起来,形成一个字符串作为Map的键。
    • valueMapper (Function.identity()): 这是一个Function,用于从流中的每个元素中提取值。Function.identity()是一个便捷方法,表示将元素本身作为值。
    • mergeFunction ((existingEmployee, newEmployee) -> existingEmployee.getSalaryDate().isAfter(newEmployee.getSalaryDate()) ? existingEmployee : newEmployee): 这是一个BinaryOperator,当keyMapper生成相同的键时(即发生键冲突时),它会被调用来决定保留哪个值。
      • existingEmployee:Map中已经存在的与当前键关联的Employee对象。
      • newEmployee:当前正在处理的、与当前键关联的Employee对象。
      • 我们通过比较它们的getSalaryDate,保留日期更晚(isAfter返回true)的那个Employee对象。
  3. .values(): Collectors.toMap操作完成后,返回的是一个Map<String, Employee>。我们只需要最终过滤后的Employee对象,因此调用.values()方法获取Map中所有值的集合。

运行结果

Employee(firstName=John, lastName=Smith, salary=5.0, getSalaryDate=2022-11-01T00:00)
Employee(firstName=Sara, lastName=Kim, salary=7.0, getSalaryDate=2022-07-01T00:00)
Employee(firstName=Kelly, lastName=Jones, salary=12.0, getSalaryDate=2022-03-01T00:00)
登录后复制

可以看到,输出结果与我们的预期完全一致,对于每个独特的姓名组合,都只保留了拥有最新薪资获取日期的员工记录。

注意事项与扩展

  1. 键的唯一性: 拼接字符串作为键是一种简单有效的方法。对于更复杂的场景,可以考虑自定义一个Pair类或record(Java 16+)来封装多个字段作为键,并确保正确实现其equals()和hashCode()方法。
  2. LocalDateTime的比较: LocalDateTime提供了isAfter()、isBefore()和isEqual()等方法,使得日期时间比较非常直观和安全。
  3. 空值处理: 如果getSalaryDate字段可能为null,则在合并函数中需要额外添加null值检查,以避免NullPointerException。例如:
    (e1, e2) -> {
        if (e1.getSalaryDate() == null) return e2;
        if (e2.getSalaryDate() == null) return e1;
        return e1.getSalaryDate().isAfter(e2.getSalaryDate()) ? e1 : e2;
    }
    登录后复制
  4. 性能考量: 对于非常大的数据集,Collectors.toMap通常表现良好,因为它只进行一次遍历。然而,键的生成(尤其是字符串拼接)会产生额外的对象,这在极端性能敏感的场景下可能需要优化。
  5. 选择其他条件: 如果需要选择日期最旧的记录,只需将合并函数中的比较逻辑反转即可:e1.getSalaryDate().isBefore(e2.getSalaryDate()) ? e1 : e2。

总结

通过本教程,我们学习了如何巧妙地运用Java Stream API中的Collectors.toMap,结合自定义的键映射和合并函数,来解决复杂的数据过滤和聚合问题。这种方法不仅代码简洁、可读性强,而且在处理大量数据时表现出良好的性能。掌握这种模式,将大大提升在Java中进行数据处理的效率和灵活性。

以上就是Java Stream API:高效处理列表数据,按组合键去重并选择最新记录的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号