rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-07-07 14:10:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

use pg_quote

Browse Source
This commit is contained in:
Conrad Ludgate
2024-10-16 12:05:27 +01:00
parent 028f5291f8
commit 59d197f6f6
1 changed files with 2 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
compute_tools/src/compute.rs
View File

@@ -20,7 +20,6 @@ use futures::stream::FuturesUnordered;
use futures::StreamExt;
use nix::unistd::Pid;
use postgres::error::SqlState;
use postgres::Row;
use postgres::{Client, NoTls};
use tracing::{debug, error, info, instrument, warn};
use utils::id::{TenantId, TimelineId};
@@ -1394,14 +1393,8 @@ LIMIT 100",
.map(|row| row.get(0));
// sanitize the inputs as postgres idents.
// should we instead use `Escaping::pg_quote`?
let format_query = "SELECT quote_ident($1) ext_name, quote_ident($2) ext_version";
let format: Row = db_client
.query_one(format_query, &[&ext_name, &ext_version])
.await
.with_context(|| format!("Failed to execute query: {}", format_query))?;
let ext_name: String = format.get("ext_name");
let ext_version: String = format.get("ext_version");
let ext_name: String = ext_name.to_string().pg_quote();
let ext_version: String = ext_version.to_string().pg_quote();
if let Some(installed_version) = version {
if installed_version == ext_version {