itsscb/sqlx
1
0
Fork 0
mirror of https://github.com/launchbadge/sqlx.git synced 2026-03-19 08:39:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added quotes to listen/unlisten channel names to honor case sensitivity and be consistent with 'listen_all()'

Browse Source
This commit is contained in:
rage311
2020-06-02 21:30:23 -06:00
committed by Austin Bonander
parent 9b68eb19ef
commit a0d1106f90
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sqlx-core/src/postgres/listener.rs
View File

@@ -64,9 +64,10 @@ impl PgListener {
}
/// Starts listening for notifications on a channel.
/// The channel name is quoted here to ensure case sensitivity.
pub async fn listen(&mut self, channel: &str) -> Result<(), Error> {
self.connection()
.execute(&*format!("LISTEN {}", ident(channel)))
.execute(&*format!(r#"LISTEN "{}""#, ident(channel)))
.await?;
self.channels.push(channel.to_owned());
@@ -92,9 +93,10 @@ impl PgListener {
}
/// Stops listening for notifications on a channel.
/// The channel name is quoted here to ensure case sensitivity.
pub async fn unlisten(&mut self, channel: &str) -> Result<(), Error> {
self.connection()
.execute(&*format!("UNLISTEN {}", ident(channel)))
.execute(&*format!(r#"UNLISTEN "{}""#, ident(channel)))
.await?;
if let Some(pos) = self.channels.iter().position(|s| s == channel) {