Как отфильтровать карту карты, используя поток java - PullRequest
Новая
       25

Как отфильтровать карту карты, используя поток java

2 голосов
/ 25 февраля 2020

Как я могу отфильтровать карту карты, используя поток java. Я могу сделать это, используя двойной l oop, но я думаю, что это не эффективно. 

public class MapUsage {
    static Map<Integer, Map<Integer, String>> parentMap = new HashMap<>();
    static Map<String, String> userRole = new HashMap<>();

    public static void main(String ... args){
        initializeData();
        displayMapForUser("user1", parentMap);
//        printMap(parentMap);

    }


    private static void displayMapForUser(String user1, Map<Integer, Map<Integer, String>> parentMap) {
        Integer role = new Integer(userRole.get(user1));
        Map<Integer, Map<Integer, String>> userMap = new HashMap<>();
        Map<Integer, String> childMap = new HashMap<>();
        for(Map.Entry<Integer, Map<Integer, String >> entry : parentMap.entrySet()){
            for(Map.Entry<Integer, String > entry1: entry.getValue().entrySet()){
                if(entry1.getKey().equals(role))
                    childMap.put(entry1.getKey(), entry1.getValue());
            }
            userMap.put(entry.getKey(), childMap);
        }
        printMap(userMap);
    }

    private static void printMap(Map<Integer, Map<Integer, String>> parentMap) {
        for(Map.Entry<Integer, Map<Integer,String> > entry: parentMap.entrySet()){
            System.out.println("key: "+entry.getKey());
            System.out.println("value: "+entry.getValue());
        }
    }

    private static void initializeData() {
        Map<Integer, String > childMap1 = new HashMap<>();
        Map<Integer, String > childMap2 = new HashMap<>();
        userRole.put("user1", "1");
        userRole.put("user2", "2");
        userRole.put("user3", "3");
        userRole.put("user4", "4");

        childMap1.put(1, "one");
        childMap1.put(2, "two");
        childMap1.put(3, "three");
        childMap1.put(4, "four");
        parentMap.put(1, childMap1);

        childMap2.put(1, "one");
        childMap2.put(2, "two");
        childMap2.put(3, "three");
        parentMap.put(2, childMap2);
    }
}

Вывод: 

key: 1
value: {1=one}
key: 2
value: {1=one}

1 Ответ

5 голосов
/ 25 февраля 2020

Вы можете использовать Collectors.toMap там как: 

Map<Integer, Map<Integer, String>> userMap = parentMap.entrySet().stream()
        .collect(Collectors.toMap(Map.Entry::getKey, entry -> entry.getValue()
                .entrySet().stream()
                .filter(en -> en.getKey().equals(role))
                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue))));

Но обратите внимание, что вы не добьетесь существенного повышения производительности. Просто читабельность, и вы могли бы написать это как: 

Map<Integer, Map<Integer, String>> usersMap = new HashMap<>(parentMap);
usersMap.values().forEach(innerMap -> innerMap.entrySet().removeIf(en -> !en.getKey().equals(role)));
...