java.util.stream.Collectors с EnumSet Потока

Question

Я пытаюсь использовать вместо битовой маски Ниже приведен код

public static Set<Amenities> fromBitFlags(int bitFlag) { 
    return ALL_OPTS.stream().filter(a -> (a.ameityId & bitFlag) > 0).collect(Collectors.toSet()); 
} 

Я хотел бы вернуться EnumSet вместо обычного набора (не хочу терять на полезность EnumSet только из-за кастинга).

Нужно определить, как создать пользовательский сборщик для сбора EnumSet.

Answer 1

Вы можете использовать toCollection(Supplier):

return ALL_OPTS.stream().filter(a -> (a.ameityId & bitFlag) > 0) 
       .collect(Collectors.toCollection(() -> EnumSet.noneOf(Amenities.class))); 

Метод toCollection получает лямбда, который должен создать пустую коллекцию для хранения результата. Здесь мы создаем пустой EnumSet с использованием вызова EnumSet.noneOf. Обратите внимание, что для EnumSet вы должны всегда указывать (неявно или явно), для чего это перечисление установлено.