Impl AsMut for advisory lock types (launchbadge#2520) (launchbadge#2554)

The documentation for `PgAdvisoryLockGuard` lists a set of types
that should be able to be passed to it, but when actually trying
to do so, compilation would fail due to missing `AsMut` trait
implementations for those types. This commit adds the missing
`AsMut` impls so that `Transaction` and `PgConnection` can be used
as type parameters to `PgAdvisoryLockGuard`, as expected.

For reference: launchbadge#2520

Author: alilleybrinker

sqlx-core/src/transaction.rs

@@ -197,6 +197,16 @@ where
     }
 }
 
+// Implement `AsMut<DB::Connection>` so `Transaction` can be given to a
+// `PgAdvisoryLockGuard`.
+//
+// See: https://github.com/launchbadge/sqlx/issues/2520
+impl<'c, DB: Database> AsMut<DB::Connection> for Transaction<'c, DB> {
+    fn as_mut(&mut self) -> &mut DB::Connection {
+        &mut self.connection
+    }
+}
+
 impl<'c, 't, DB: Database> crate::acquire::Acquire<'t> for &'t mut Transaction<'c, DB> {
     type Database = DB;
 

sqlx-postgres/src/connection/mod.rs

@@ -213,3 +213,13 @@ impl Connection for PgConnection {
         !self.stream.write_buffer().is_empty()
     }
 }
+
+// Implement `AsMut<Self>` so that `PgConnection` can be wrapped in
+// a `PgAdvisoryLockGuard`.
+//
+// See: https://github.com/launchbadge/sqlx/issues/2520
+impl AsMut<PgConnection> for PgConnection {
+    fn as_mut(&mut self) -> &mut PgConnection {
+        self
+    }
+}