[Java 8][Stream API] オブジェクトの List から Map を生成する方法を模索する

はじめに

Java 8 を導入し始めて既に半年以上になるけど、未だに Java 8 の Stream API で複雑な処理をしようとすると頭を抱えてしまうので、その時々の 自分なりの ベストプラクティスをメモしておくことにします。

今回のお題

Stream API に慣れてくると、データクラスのオブジェクトからなる List オブジェクトに対して、データクラスのある属性をグルーピングのキーに、もう一つ別の属性をグルーピングされた結果を保持する List の構成要素として持つような Map オブジェクトを Stream API で華麗に表現したくなるのが人間ってもんです。

今回はこの問題について、 Collectors.toMap() を使った方法と Collectors.groupingBy() と Collectors.mapping() を組み合わせて使った方法で解消してみることにしました。

実装例

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.function.BinaryOperator;
import java.util.function.Function;
import java.util.stream.Collector;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * オブジェクトのリストから、そのオブジェクトの属性を key / values にした Map を Java 8 の Stream で生成する。
 */
public class HowToBuildMapFromObjectList {
    public static void main(String[] args) {
        List<GroupUser> groupUsers = Arrays.asList(
                GroupUser.newGroupUser("でぃずにーぐみ", "泡姫（ありえる）"),
                GroupUser.newGroupUser("じぶりぐみ", "永瞳月（ととろ）"),
                GroupUser.newGroupUser("でぃずにーぐみ", "黄熊（ぷう）"),
                GroupUser.newGroupUser("じぶりぐみ", "今鹿（なうしか）")
        );

        // Collectors.toMap() を使う
        System.out.println(byCollectorsToMap(groupUsers));

        // Collectors.groupingBy() と Collectors.mapping() を組み合わせて使う
        System.out.println(byGroupingByAndMapping(groupUsers));
    }

    /**
     * {@link Collectors#toMap(Function, Function, BinaryOperator)} を使う方法。
     */
    static Map<String, List<String>> byCollectorsToMap(List<GroupUser> groupUsers) {
        return groupUsers.stream()
                .collect(Collectors.toMap(
                        GroupUser::groupName,
                        gu -> new ArrayList<>(Collections.singletonList(gu.userName())),
                        (l1, l2) -> Stream.concat(l1.stream(), l2.stream()).collect(Collectors.toList())));
    }

    /**
     * {@link Collectors#groupingBy(Function, Collector)} と
     * {@link Collectors#mapping(Function, Collector)} を組み合わせて使う方法。
     */
    static Map<String, List<String>> byGroupingByAndMapping(List<GroupUser> groupUsers) {
        return groupUsers.stream()
                .collect(Collectors.groupingBy(
                        GroupUser::groupName,
                        Collectors.mapping(
                                GroupUser::userName,
                                Collectors.toList())));
    }

    /**
     * グループ名とユーザ名を保持するよくあるクラスです。
     */
    static class GroupUser {
        private String groupName;
        private String userName;

        static GroupUser newGroupUser(String groupName, String userName) {
            GroupUser o = new GroupUser();
            o.groupName = groupName;
            o.userName = userName;
            return o;
        }

        public String groupName() {
            return groupName;
        }

        public String userName() {
            return userName;
        }
    }
}

感想

• Collectors.groupingBy() と Collectors.mapping() を組み合わせて使った実現方法が断然スマートに見えますね
• これよりもっとスマートな解決方法はないものだろうか…