将Java流筛选为1个且仅为1个元素

Question

我正在尝试使用Java 8streams来查找linkedlist中的元素。但是，我想保证只有一个和筛选条件匹配。

取此代码：

public static void main(String[] args) {

    LinkedList<User> users = new LinkedList<>();
    users.add(new User(1, "User1"));
    users.add(new User(2, "User2"));
    users.add(new User(3, "User3"));

    User match = users.stream().filter((user) -> user.getId() == 1).findAny().get();
    System.out.println(match.toString());
}

static class User {

    @Override
    public String toString() {
        return id + " - " + username;
    }

    int id;
    String username;

    public User() {
    }

    public User(int id, String username) {
        this.id = id;
        this.username = username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getUsername() {
        return username;
    }

    public int getId() {
        return id;
    }
}

此代码根据用户的ID查找他们。但不能保证有多少user与筛选器匹配。

将筛选器行更改为：

User match = users.stream().filter((user) -> user.getId() < 0).findAny().get();

将抛出nosuchelementexception（好！）

但是，如果有多个匹配，我希望它抛出一个错误。有办法做到这一点吗？

Answer 1

public static <T> Collector<T, ?, T> toSingleton() {
    return Collectors.collectingAndThen(
            Collectors.toList(),
            list -> {
                if (list.size() != 1) {
                    throw new IllegalStateException();
                }
                return list.get(0);
            }
    );
}

我们使用Collectors.CollectingAndThen通过以下方式构造所需的收集器

1. 使用Collectors.ToList()收集器收集列表中的对象。
2. 在结尾处应用一个额外的结束器，该结束器返回单个元素-或者如果List.Size！=1则抛出IllegalStateException。

用作：

User resultUser = users.stream()
        .filter(user -> user.getId() > 0)
        .collect(toSingleton());

然后，您可以任意自定义此收集器，例如，将异常作为构造函数中的参数，将其调整为允许两个值，等等。

您可以使用包含peek()和AtomicInteger的“变通方法”，但实际上您不应该使用它。

您可以做的只是将其收集到列表中，如下所示：

LinkedList<User> users = new LinkedList<>();
users.add(new User(1, "User1"));
users.add(new User(2, "User2"));
users.add(new User(3, "User3"));
List<User> resultUserList = users.stream()
        .filter(user -> user.getId() == 1)
        .collect(Collectors.toList());
if (resultUserList.size() != 1) {
    throw new IllegalStateException();
}
User resultUser = resultUserList.get(0);