diff --git a/pgpainless-core/src/main/java/org/pgpainless/util/MultiMap.java b/pgpainless-core/src/main/java/org/pgpainless/util/MultiMap.java index ee5e70f8..c3579e8e 100644 --- a/pgpainless-core/src/main/java/org/pgpainless/util/MultiMap.java +++ b/pgpainless-core/src/main/java/org/pgpainless/util/MultiMap.java @@ -79,10 +79,14 @@ public class MultiMap { } } - public void remove(K o) { - for (Set values : map.values()) { - values.remove(o); - } + public void removeAll(K o) { + map.remove(o); + } + + public void remove(K o, V v) { + Set vs = map.get(o); + if (vs == null) return; + vs.remove(v); } public void putAll(Map> _map) {