From 07e1b8230feb9c3d5f6ccfcb0bfebe147caae9d5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Frosteg=C3=A5rd?= <joakim.frostegard@gmail.com>
Date: Mon, 1 Nov 2021 21:40:41 +0100
Subject: [PATCH] aquatic_ws: reclaim pending_scrape_slab memory after entry
 removal

---
 aquatic_ws/src/lib/network.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/aquatic_ws/src/lib/network.rs b/aquatic_ws/src/lib/network.rs
index f2682d1..e777e0c 100644
--- a/aquatic_ws/src/lib/network.rs
+++ b/aquatic_ws/src/lib/network.rs
@@ -416,7 +416,10 @@ impl ConnectionWriter {
                     if let Some(out_message) = opt_message {
                         self.send_out_message(&out_message).await?;
 
-                        RefCell::borrow_mut(&self.pending_scrape_slab).remove(pending_scrape_id.0);
+                        let mut slab = RefCell::borrow_mut(&self.pending_scrape_slab);
+                        
+                        slab.remove(pending_scrape_id.0);
+                        slab.shrink_to_fit();
                     }
                 }
                 out_message => {