Collectors.toMapで任意のMapクラスを返す

Collectors.toMapは通常、Mapクラスのインスタンスを返却します。

Collectors.toMapの例

enum ENUM {
    ENUM_1
    ,ENUM_2
    ,ENUM_3
    ,ENUM_4
    ,ENUM_5
    ,ENUM_6
}


Map<ENUM, Integer> map = Stream.of(ENUM.values())
        .collect(Collectors.toMap(
            e -> e
            ,e -> e.ordinal()
        ));

HashMap、LinkedHashMap、TreeMapなど、Mapクラスの任意の子クラスを返したい場合は、引数4つの Collectors.toMap を使えば可能になります。

Collectors.toMap

toMap(Function<? super T,? extends K> keyMapper, Function<? super T,? extends U> valueMapper, BinaryOperator<U> mergeFunction, Supplier<M> mapSupplier)

Collectors.toMapでHashMapを返す例

HashMap<ENUM, Integer> map= Stream.of(ENUM.values())
        .collect(Collectors.toMap(
            e -> e
            ,e -> e.ordinal()
            ,(e1, e2) -> e1
            ,HashMap::new
        ));

Collectors.toMapでEnumMapを返したい場合ですが、EnumMapは引数なしコンストラクタを持ってません。この場合どう実装すれば良いかというと、コンストラクタ参照は無理なのでラムダ式で実装すればOKです。

Collectors.toMapでEnumMapを返す例1

EnumMap<ENUM, Integer> map = Stream.of(ENUM.values())
        .collect(Collectors.toMap(
            e -> e
            ,e -> e.ordinal()
            ,(e1, e2) -> e1
            ,() -> {
                EnumMap<ENUM, Integer> ret = new EnumMap<>(ENUM.class);
                return ret;
            }
        ));

いろいろ省略すると以下の様に書くことも出来ます。

Collectors.toMapでEnumMapを返す例2

EnumMap<ENUM, Integer> map = Stream.of(ENUM.values())
        .collect(Collectors.toMap(
            e -> e
            ,e -> e.ordinal()
            ,(e1, e2) -> e1
            ,() -> new EnumMap<>(ENUM.class)
        ));