Java Stream API - 按对象内部列表的项目分组

Question

有没有办法实现以下示例代码，利用 Java Stream API 而不必创建HashMap并将其填充到 double forEach es 中？ 我试图使用groupingBy和flatMap但找不到出路。

有一个电影列表，其中每个电影都有一个流派（字符串）列表......

class Movie {
    List<String> genres;
}

List<Movie> movies = new ArrayList<>();

...我想按类型对所有电影进行分组

Map<String, List<Movie>> moviesByGenre = new HashMap();
movies.stream()
        .forEach(movie -> movie.getGenres()
                .stream()
                .forEach(genre -> moviesByGenre
                        .computeIfAbsent(genre, k -> new ArrayList<>())
                        .add(movie)));

Answer 1

这很棘手，因为您无法为每个Movie定义一个键，因为这样的对象可以出现在多个键下。

据我所知，最好的解决方案与您的相同：

Map<String, List<Movie>> groupedMovies = new HashMap<>();
movies.forEach(movie -> {
    movie.getGenres().forEach(genre ->
        groupedMovies.computeIfAbsent(genre, g -> new ArrayList<>()).add(movie)
    );
});

如果你想“转换”这个片段到Java的流，你必须开始与你有什么-这是个人风格。 使用flatMap和distinct从每个Movie提取它们以避免重复。 然后使用Collector.toMap获得所需的输出。

• Key： Function.identity()将每个独特的genre映射为一个键本身。
• 值：使用另一个Stream过滤掉包含特定genre的电影以将它们分配给键。

Map<String, List<Movie>> groupedMovies2 = movies.stream()
    .map(Movie::getGenres)
    .flatMap(List::stream)
    .distinct()
    .collect(Collectors.toMap(
            Function.identity(),
            genre -> movies.stream()
                           .filter(movie -> movie.getGenres().contains(genre))
                           .collect(Collectors.toList())));

第一个片段中的程序方法更快、更容易阅读和理解。 我不建议在这里使用java-stream 。

Answer 2

首先，我们可能想要创建一个Pair或Tuple ，如下所示：

    public static class Pair {
        Movie movie;
        String genre;

        public Pair(Movie movie, String genre) {
            this.movie = movie;
            this.genre = genre;
        }
       
       // Getters omitted

    }

这也可以是 Java 17 中的Record 。

接下来，我们可以使用toMap执行以下toMap ：

Map<String, List<Movie>> map = movies.stream()
                .flatMap(movie -> movie.getGenres().stream().map(genre -> new Pair(movie, genre)))
                .collect(Collectors.toMap(Pair::getGenre, pair -> new ArrayList<>(List.of(pair.getMovie())),
                        (existing, current) -> {
                            existing.addAll(current);
                            return existing;
                        }));

或者我们可以使用groupBy ：

Map<String, List<Movie>> map = movies.stream()
                .flatMap(movie -> movie.getGenres().stream().map(genre -> new Pair(movie, genre)))
                .collect(Collectors.groupingBy(Pair::getGenre,
                        HashMap::new,
                        Collectors.mapping(Pair::getMovie, Collectors.toList())));

本质上，我们为每部电影创建了一个Pair及其所有可能的genre 。 在我们拥有这些对之后，我们以某种方式将它们分组以摆脱Pair 。

Answer 3

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import static java.util.stream.Collectors.groupingBy;


class Movie {
        String type;
        String name;
        Movie(String type, String name)
        {
            this.type = type;
            this.name = name;
        }

    public Movie(String type) {
            this.type = type;
    }

    public String getType() {
        return type;
    }
    public String getName()
    {
        return name;
    }

    public static void main(String... args) {
        List<Movie> posts = new ArrayList<>();
        posts.add( new Movie("HORROR", "movie1"));
        posts.add( new Movie("HORROR", "movie1"));
        posts.add( new Movie("HORROR", "movie1"));
        posts.add( new Movie("COMEDY", "movie1"));
        posts.add( new Movie("COMEDY", "movie2"));
        posts.add( new Movie("COMEDY", "movie3"));
        posts.add( new Movie("COMEDY", "movie2"));


        Map<String, List<Movie>>postsPerType = posts.stream()
            .collect(groupingBy(Movie::getType));

        postsPerType.forEach((k,v) -> System.out.println("Key = "
            + k + ", Value = " + v.size()));

        Map<String, List<Movie>>postsPerName = posts.stream()
            .collect(groupingBy(Movie::getName));
        postsPerName.forEach((k,v) -> System.out.println("Key = "
            + k + ", Value = " + v.size()));

    }
}

Key = HORROR, Value = 3
Key = COMEDY, Value = 4
Key = movie3, Value = 1
Key = movie2, Value = 2
Key = movie1, Value = 4