Java 8 stream for Map <String，Set <String>>

Question

我有一个Map < Integer，Set < Integer>>。我想根据自定义方法所做的一些修改将其转换为整数列表。

现在我用两个for循环，我想知道是否有更好的方法用java做流

这里是我现有的代码：

public myMethod(Map<Integer, Set<Integer>> myMap, String a, int b) { 
List<Integer> myIntegerList = new ArrayList<>(); 
    for (int i: myMap.keySet()) { 
     for (int j: myMap.get(i)) { 
      myIntegerList.add(myCustomMethod(i, j, a.concat(b)); 
     } 
    } 
} 

public Integer myCustomMethod(int x, int y, String result) { 
... 
... 
... 

return Integer; 
} 

我想知道，如果我们可以使用java stream（）迭代整数集合？

Answer 1

试试这个：

public void myMethod(Map<Integer, Set<Integer>> myMap, String a, String b) { 
     List<Integer> myIntegerList = new ArrayList<>(); 
     for (int i: myMap.keySet()) 
      myIntegerList.addAll(myMap.get(i).stream().map(j -> myCustomMethod(i, j, a.concat(b))).collect(Collectors.toList())); 
    } 

我变变 “B” 为字符串，因为CONCAT需要字符串（但如果你需要INT，您可以使用方法Integer.toString(b)

Answer 2

List<Integer> myIntegerList = myMap.entrySet().stream() 
      .flatMap(myEntry -> 
       myEntry.getValue().stream() 
        .map(setEntry -> myCustomMethod(myEntry.getKey(), setEntry, a + b))) 
      .collect(Collectors.toList());