From 5cd87bd537cb0f4f86d9ee7a0151b84a7ea2f522 Mon Sep 17 00:00:00 2001
From: Paul Schaub <vanitasvitae@fsfe.org>
Date: Tue, 21 Aug 2018 14:47:39 +0200
Subject: [PATCH] Fix MultiMap.remove()

---
 .../src/main/java/org/pgpainless/util/MultiMap.java  | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

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<K, V> {
         }
     }
 
-    public void remove(K o) {
-        for (Set<V> values : map.values()) {
-            values.remove(o);
-        }
+    public void removeAll(K o) {
+        map.remove(o);
+    }
+
+    public void remove(K o, V v) {
+        Set<V> vs = map.get(o);
+        if (vs == null) return;
+        vs.remove(v);
     }
 
     public void putAll(Map<? extends K, ? extends Set<V>> _map) {