From dfc846e4834b83b051bd8a3870a229010832265d Mon Sep 17 00:00:00 2001
From: Richard Newman <rnewman@twinql.com>
Date: Mon, 17 Apr 2017 21:10:32 -0700
Subject: [PATCH] Part 3: define keep_intersected_keys.

We'll use this to drop unneeded values from input maps, if lazy callers
reuse a general-purpose map for multiple queries.
---
 query-algebrizer/src/clauses/mod.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/query-algebrizer/src/clauses/mod.rs b/query-algebrizer/src/clauses/mod.rs
index c8407583..e606a24e 100644
--- a/query-algebrizer/src/clauses/mod.rs
+++ b/query-algebrizer/src/clauses/mod.rs
@@ -89,6 +89,7 @@ trait Contains<K, T> {
 
 trait Intersection<K> {
     fn with_intersected_keys(&self, ks: &BTreeSet<K>) -> Self;
+    fn keep_intersected_keys(&mut self, ks: &BTreeSet<K>);
 }
 
 impl<K: Ord, T> Contains<K, T> for BTreeSet<K> {
@@ -108,6 +109,22 @@ impl<K: Clone + Ord, V: Clone> Intersection<K> for BTreeMap<K, V> {
             .filter_map(|(k, v)| ks.when_contains(k, || (k.clone(), v.clone())))
             .collect()
     }
+
+    /// Remove all keys from the map that are not present in `ks`.
+    /// This implementation is terrible because there's no mutable iterator for BTreeMap.
+    fn keep_intersected_keys(&mut self, ks: &BTreeSet<K>) {
+        let mut to_remove = Vec::with_capacity(self.len() - ks.len());
+        {
+        for k in self.keys() {
+            if !ks.contains(k) {
+                to_remove.push(k.clone())
+            }
+        }
+        }
+        for k in to_remove.into_iter() {
+            self.remove(&k);
+        }
+    }
 }
 
 /// A `ConjoiningClauses` (CC) is a collection of clauses that are combined with `JOIN`.