itsscb/sqlx
1
0
Fork 0
mirror of https://github.com/launchbadge/sqlx.git synced 2025-12-26 10:09:50 +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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/postgres/listen/src/main.rs
View File

@ -51,6 +51,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
async fn notify(mut pool: &PgPool) {
static COUNTER: AtomicUsize = AtomicUsize::new(0);
// Note that channel names are lower-cased by Postgres unless they are quoted
let res = pool
.execute(&*format!(
r#"

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) {