Java8 Создать карту группировки по ключу, содержащемуся в значениях - PullRequest
Новая
       55

Java8 Создать карту группировки по ключу, содержащемуся в значениях

0 голосов
/ 03 июля 2018

У меня есть два следующих списка строк: 

{APPLE, ORANGE, BANANA} //call it keyList
{APPLE123, ORANGEXXX, 1APPLE, APPLEEEE} //call it valueList

Желаемый вывод - HashMap<String, List<String>>, например: 

<APPLE, {APPLE123, 1APPLE, APPLEEEE}>
<ORANGE, {ORANGEXXX}>
<BANANA, {}> //also <key, null> is accepted

Я реализовал это решение (оно работает) 

HashMap<String, List<String>> myMap = new HashMap<>();
keyList.forEach(key -> {
    List<String> values = valueList.stream()
            .filter(value -> value.contains(key))
            .collect(Collectors.toList());
    myMap.put(key, values);
});

Учитывая предположение, что значение связано только с одним ключом (это ограничение моего домена), является ли это лучшим решением в java8 с точки зрения производительности и / или очистки кода? Можно ли как-то его настроить?

1 Ответ

0 голосов
/ 03 июля 2018

Если вы можете с уверенностью предположить, что каждое значение связано с ключом и только с одним ключом, вы можете пойти в следующем направлении: 

Pattern p = Pattern.compile(String.join("|", keyList));
Map<String, List<String>> map = valueList.stream()
    .collect(Collectors.groupingBy(s -> {
        Matcher m = p.matcher(s);
        if(!m.find()) throw new AssertionError();
        return m.group();
    }));

map.forEach((k,v) -> System.out.println(k+": "+v));

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

Pattern p = Pattern.compile(
    keyList.stream().map(Pattern::quote).collect(Collectors.joining("|")));

Операция collect создает группы только для существующих значений. Если вам действительно нужны все ключи, вы можете использовать 

Map<String, List<String>> map = valueList.stream()
    .collect(Collectors.groupingBy(s -> {
            Matcher m = p.matcher(s);
            if(!m.find()) throw new AssertionError();
            return m.group();
        },
        HashMap::new, // ensure mutable map
        Collectors.toList()
    ));
keyList.forEach(key -> map.putIfAbsent(key, Collections.emptyList()));

или 

Pattern p = Pattern.compile(
    keyList.stream().map(Pattern::quote)
           .collect(Collectors.joining("|", ".*(", ").*")));
Map<String, List<String>> map = valueList.stream()
    .map(p::matcher)
    .filter(Matcher::matches)
    .collect(Collectors.groupingBy(m -> m.group(1),
        HashMap::new, // ensure mutable map
        Collectors.mapping(Matcher::group, Collectors.toList())
    ));
keyList.forEach(key -> map.putIfAbsent(key, Collections.emptyList()));
...