feat: convenient wrapper for Postgres advisory locks

launchbadge · Mar 24, 2022 · cb78292 · cb78292
1 parent f539215
commit cb78292
Show file tree

Hide file tree

Showing 8 changed files with 527 additions and 9 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/sqlx-core/Cargo.toml b/sqlx-core/Cargo.toml
@@ -33,6 +33,7 @@ postgres = [
     "json",
     "dirs",
     "whoami",
+    "hkdf"
 ]
 mysql = [
     "sha-1",
@@ -168,6 +169,7 @@ hashlink = "0.7.0"
 # NOTE: *must* remain below 1.7.0 to allow users to avoid the `ahash` cyclic dependency problem by pinning the version
 # https://github.com/tkaitchuck/aHash/issues/95#issuecomment-874150078
 indexmap = "1.6.0"
+hkdf = { version = "0.11.0", optional = true }
 
 [dev-dependencies]
 sqlx = { version = "0.5.11", path = "..", features = ["postgres", "sqlite"] }
diff --git a/sqlx-core/src/pool/connection.rs b/sqlx-core/src/pool/connection.rs
@@ -59,6 +59,18 @@ impl<DB: Database> DerefMut for PoolConnection<DB> {
     }
 }
 
+impl<DB: Database> AsRef<DB::Connection> for PoolConnection<DB> {
+    fn as_ref(&self) -> &DB::Connection {
+        self
+    }
+}
+
+impl<DB: Database> AsMut<DB::Connection> for PoolConnection<DB> {
+    fn as_mut(&mut self) -> &mut DB::Connection {
+        self
+    }
+}
+
 impl<DB: Database> PoolConnection<DB> {
     /// Explicitly release a connection from the pool
     #[deprecated = "renamed to `.detach()` for clarity"]