flatMap на элементы набора - PullRequest
       26

flatMap на элементы набора

1 голос
/ 12 октября 2019

У меня есть Map<String, Set<String>>, скажем, это {"a": {"a1", "a2", "a3"}, "b": {"b1", "b2", "b3"}, "c": {"c1", "c2"}, "d": {}}

У меня есть поток наборов ключей карты, и я хочу отобразить каждый элемент потокового набора на элементы соответствующегозначение, установленное на моей карте, например,

INPUT STREAM:

{"a","b"}
{"a","c"}
{"b","c","d"}

ВЫХОДНОЙ ПОТОК:

//first set
{"a1","b1"}
{"a1","b2"}
{"a1","b3"}
{"a2","b1"}
{"a2","b2"}
{"a2","b3"}
{"a3","b1"}
{"a3","b2"}
{"a3","b3"}
//second set
{"a1","c1"}
{"a1","c2"}
{"a2","c1"}
{"a2","c2"}
{"a3","c1"}
{"a3","c2"}
//third set would be flatmapped to nothing, as "d" is mapped to an empty set

Как я могу сделать это с помощью потоков Java8?

Есть ли лучший способ сделать это, используя только API Java SE 8?

1 Ответ

0 голосов
/ 12 октября 2019

Вы можете использовать коллекции Apache Commons

<dependency>
    <groupId>org.apache.commons</groupId>
    <artifactId>commons-collections4</artifactId>
    <version>4.4</version>
</dependency>

Пример:

public class App {

    public static void main(String[] args) {
        // With an ArrayList for the values
        MultiValuedMap<String, String> map1 = new ArrayListValuedHashMap<>();
        map1.put("a", "a1");
        map1.put("a", "a2");
        map1.put("a", "a3");
        map1.put("a", "a4");
        map1.put("a", "a5");
        map1.put("b", "b1");
        map1.put("b", "b1");
        map1.put("b", "b3");
        map1.put("b", "b4");
        map1.put("b", "b5");
        map1.entries().forEach(e -> System.out.println(e.getKey() + " - " + e.getValue()));
        System.out.println("----");
        // With a HashSet for the values
        MultiValuedMap<String, String> map2 = new HashSetValuedHashMap<>(map1);
        map2.entries().forEach(e -> System.out.println(e.getKey() + " - " + e.getValue()));
    }
}

Вывод выглядит следующим образом:

a - a1
a - a2
a - a3
a - a4
a - a5
b - b1
b - b1
b - b3
b - b4
b - b5
----
a - a1
a - a2
a - a3
a - a4
a - a5
b - b3
b - b4
b - b5
b - b1
...