Использование ComputeIfAbsent & ComputeIfPresent для размещения списка на карте - PullRequest
Новая
       5

Использование ComputeIfAbsent & ComputeIfPresent для размещения списка на карте

0 голосов
/ 30 апреля 2020

У меня есть блок кода, который хорошо работает для меня: 

for (String word : distinctWordsInOneLigne) {
                        Map<String, List<Integer>> map = new HashMap<>();
                        if (!word.isEmpty()) {
                            List<Integer> linePositionsOfWord = new LinkedList<>();
                            if (currentLine.contains(word)) {
                                linePositionsOfWord.add(numLine);
                                if (mapAllWordsPositionInFilesInFolder.containsKey(word)) {
                                    Map<String, List<Integer>> mapList = mapAllWordsPositionInFilesInFolder.get(word);
                                    if (mapList.containsKey(filePath)) {
                                        List<Integer> list = mapList.get(filePath);
                                        list.add(numLine);
                                    } else {
                                        mapList.put(filePath, linePositionsOfWord);
                                    }
                                } else {
                                    map.put(filePath, linePositionsOfWord);
                                    mapAllWordsPositionInFilesInFolder.put(word, map);
                                }
                            }
                        }
                    }

NB: Map<String, Map<String, List<Integer>>> mapAllWordsPositionInFilesInFolder = new HashMap<>(); Результат примерно такой: 

{word1={file2.txt=[7], file1.txt=[1, 2]}, word2={file2.txt=[1, 2, 9, 13], file5.txt=[2, 3, 9]}}

Теперь я хочу получить немного результат, но теперь, используя ComputeIfAbsent & ComputeIfPresent вместо containsKey и все это if ... else.

Я пробовал это, но не работает: 

mapAllWordsPositionInFilesInFolder.computeIfAbsent(word,v -> new HashMap<>())
                                        .computeIfAbsent(filePath,  val -> linePositionsOfWord);

mapAllWordsPositionInFilesInFolder.computeIfPresent(word,(k,v)->{
                                    v.computeIfPresent(filePath, (x, y) -> linePositionsOfWord.add(numLine));
                                    return v;
                                });

Мне нужна помощь, пожалуйста ! благодарю :))

1 Ответ

0 голосов
/ 30 апреля 2020

Вы бы не использовали computeIfPresent() для этого, но вы бы использовали computeIfAbsent() так: 

for (String word : distinctWordsInOneLigne) {
    if (! word.isEmpty() && currentLine.contains(word)) {
        mapAllWordsPositionInFilesInFolder.computeIfAbsent(word, k -> new HashMap<>())
                                          .computeIfAbsent(filePath, k -> new LinkedList<>())
                                          .add(numLine);
    }
}

Оригинальный код был написан очень плохо. Даже без использования computeIfPresent() его можно много почистить, исключив повторный код. Вот как было написано , если бы было написано: 

for (String word : distinctWordsInOneLigne) {
    if (! word.isEmpty() && currentLine.contains(word)) {
        Map<String, List<Integer>> mapList = mapAllWordsPositionInFilesInFolder.get(word);
        if (mapList == null) {
            mapList = new HashMap<>();
            mapAllWordsPositionInFilesInFolder.put(word, mapList);
        }
        List<Integer> linePositionsOfWord = mapList.get(filePath);
        if (linePositionsOfWord == null) {
            linePositionsOfWord = new LinkedList<>();
            mapList.put(filePath, linePositionsOfWord);
        }
        linePositionsOfWord.add(numLine);
    }
}

С помощью встраивания можно уменьшить до: 

for (String word : distinctWordsInOneLigne) {
    if (! word.isEmpty() && currentLine.contains(word)) {
        Map<String, List<Integer>> mapList = mapAllWordsPositionInFilesInFolder.get(word);
        if (mapList == null)
            mapAllWordsPositionInFilesInFolder.put(word, mapList = new HashMap<>());
        List<Integer> linePositionsOfWord = mapList.get(filePath);
        if (linePositionsOfWord == null)
            mapList.put(filePath, linePositionsOfWord = new LinkedList<>());
        linePositionsOfWord.add(numLine);
    }
}
...