在Java中使用Collectors.toMap時，如何跳過將條目添加到新地圖？

Question

因此，這裏是我的代碼：

HashMap<String, List<Foo.Builder>> strFooMap = new HashMap<>(); 
// add stuff to strFooMap and do other things 
return new HashMap<>(strFooMap.entrySet() 
    .stream() 
    .collect(Collectors.toMap(Entry::getKey, 
     entry -> entry 
      .getValue() 
      .stream() 
      .filter(builder -> { 
       // do some conditional stuff to filter 
      }) 
      .map(builder -> { 
       return builder.build(); 
      }) 
      .collect(Collectors.toList()) 
     ) 
    ) 
); 

所以有的時候，做filter後，在我的List<Foo.Builder>將被過濾掉所有的項目。然後在我的最後HashMap，我有這樣的條目：{"someStr" -> []}。我想刪除值爲空列表的鍵值對。我會如何去做這件事？在.stream().collect(Collectors.toMap(...))的代碼中可以這樣做嗎？謝謝！

Answer 1

其中一個解決方案是在收集結果映射之前移動值轉換。它爲您提供了過濾空值的可能性

List<Foo> transformValue(List<Foo.Builder> value) { 
    //original value transformation logic 
} 

<K,V> Entry<K,V> pair(K key, V value){ 
    return new AbstractMap.SimpleEntry<>(key, value); 
} 

strFooMap.entrySet().stream() 
     .map(entry -> pair(entry.getKey(), transformValue(entry.getValue())) 
     .filter(entry -> !entry.getValue().isEmpty()) 
     .collect(toMap(Entry::getKey, Entry::getValue));