如何为列表元素分配序列号（Java Stream API 实战）

news2024/11/15 12:10:56

摘要：

本文将详细介绍如何在Java中使用Stream API对列表中的元素进行分组，并为每组元素分配一个唯一的序列号。通过一个具体的示例和测试案例，我们将演示这一技术如何简化数据处理，提高代码效率。
在Java编程中，我们经常需要对数据进行分组和排序。Java 8引入的Stream API为我们提供了强大的数据处理能力。本文将介绍如何利用Stream API来实现一个实用的功能：为列表中的元素分配序列号。

一、问题背景

在实际应用中，我们可能会遇到这样的需求：给定一个对象列表，需要按照某种属性对这些对象进行分组，并为每个分组内的对象分配一个递增的序列号。例如，在处理客户订单时，可能需要按订单类型分组，并为每组订单分配一个序列号。

二、解决方案

为了解决这个问题，我们创建了一个名为SequenceAssigner的类，其中包含一个静态方法assignSequencesDirectly。这个方法接受三个参数：一个元素列表、一个分组函数和一个序列号设置器。
以下是SequenceAssigner类的核心代码：

import java.util.*;
import java.util.function.*;
import java.util.stream.Collectors;
public class SequenceAssigner {
    public static <T> void assignSequencesDirectly(List<T> list,
                                                   Function<T, ?> grouper,
                                                   BiConsumer<T, Integer> sequenceNumberSetter) {
        Map<Object, List<T>> grouped = list.stream()
                .collect(Collectors.groupingBy(t -> grouper.apply(t) != null ? grouper.apply(t) : ""));
        int seqNumber = 1;
        for (Map.Entry<Object, List<T>> entry : grouped.entrySet()) {
            for (T item : entry.getValue()) {
                sequenceNumberSetter.accept(item, seqNumber);
            }
            seqNumber++;
        }
    }
}

三、测试案例

为了验证我们的解决方案，我们创建了一个简单的Person类和一个测试案例。
以下是Person类的定义：

public class Person {
    private String name;
    private String surname;
    private int sequenceNumber;
    // 构造器、getter、setter 和 toString 方法
}

接下来是测试案例：

public class SequenceAssignerTest {
    public static void main(String[] args) {
        List<Person> people = new ArrayList<>();
        people.add(new Person("Alice", "Smith"));
        people.add(new Person("Bob", "Smith"));
        people.add(new Person("Charlie", "Brown"));
        people.add(new Person("David", "Smith"));
        people.add(new Person("Eva", "Brown"));
        Function<Person, String> grouper = Person::getSurname;
        BiConsumer<Person, Integer> sequenceNumberSetter = Person::setSequenceNumber;
        SequenceAssigner.assignSequencesDirectly(people, grouper, sequenceNumberSetter);
        people.forEach(System.out::println);
    }
}