Удалить ключ, если значение пуста

Question

Скажем, у нас есть Map<Key, Collection<Value>> myMap и метод, который удаляет значение из коллекции, связанной с ключом. Если удаление значения выходит из коллекции пустой, мы хотим избавиться от key записи в карте:

List<Value> removeValue(Key key, Value value) { 
    List<Value> v = myMap.get(key); 
    if (v != null) { 
     v.remove(value); 
     if (v.isEmpty()) 
      myMap.remove(key); 
    } 
    return v; 
} 

Есть ли Java 8 способы достижения описанного поведения с однострочником или коротким способом ?

Answer 1

Вы можете использовать computeIfPresent для этого:

static <K, V> List<V> removeValue(K key, V value, Map<K, List<V>> map){ 
    return map.computeIfPresent(key, (k, l) -> l.remove(value) && l.isEmpty() ? null : l); 
} 

computeIfPresent применяет BiFunction к ключу и текущего значения в карте, если значение не null (если это nullcomputeIfPresent возвращается null сразу), а затем либо задает значение возвращаемому значению BiFunction, если возвращаемое значение не равно null или удаляет ключ с карты, если возвращаемое значение равно null и, наконец, возвращает новое значение.

Обратите внимание, что она ведет себя немного по-другому методу, у вас вопрос - это не будет удалить уже пустой List с карты, потому что remove вернется false. Если вы хотите удалить уже пустой List, вы можете использовать l.isEmpty() || (l.remove(value) && l.isEmpty()).