decrement pool size when connection is released

launchbadge · Dec 18, 2020 · 07fd531 · 07fd531
1 parent df1b9b0
commit 07fd531
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/sqlx-core/src/pool/connection.rs b/sqlx-core/src/pool/connection.rs
@@ -56,9 +56,13 @@ impl<DB: Database> DerefMut for PoolConnection<DB> {
 }
 
 impl<DB: Database> PoolConnection<DB> {
-    // explicitly release a connection from the pool
+    /// Explicitly release a connection from the pool
     pub fn release(mut self) -> DB::Connection {
-        self.live.take().expect("PoolConnection double-dropped").raw
+        self.live
+            .take()
+            .expect("PoolConnection double-dropped")
+            .float(&self.pool)
+            .detach()
     }
 }
 
@@ -154,6 +158,10 @@ impl<'s, DB: Database> Floating<'s, Live<DB>> {
         }
     }
 
+    pub fn detach(self) -> DB::Connection {
+        self.inner.raw
+    }
+
     pub fn into_idle(self) -> Floating<'s, Idle<DB>> {
         Floating {
             inner: self.inner.into_idle(),