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

add missing parts to make disable_pg_session_jwt flag work

Browse Source
This commit is contained in:
Ruslan Talpa
2025-06-20 10:23:42 +03:00
parent 6bd15908fb
commit b39f04ab99
1 changed files with 17 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
proxy/src/serverless/backend.rs
View File

@@ -315,13 +315,15 @@ impl PoolingBackend {
config
.user(&conn_info.user_info.user)
.dbname(&conn_info.dbname);
// .set_param(
// "options",
// &format!(
// "-c pg_session_jwt.jwk={}",
// serde_json::to_string(&jwk).expect("serializing jwk to json should not fail")
// ),
// );
if !disable_pg_session_jwt {
config.set_param(
"options",
&format!(
"-c pg_session_jwt.jwk={}",
serde_json::to_string(&jwk).expect("serializing jwk to json should not fail")
),
);
}
let pause = ctx.latency_timer_pause(crate::metrics::Waiting::Compute);
let (client, connection) = config.connect(&postgres_client::NoTls).await?;
@@ -345,11 +347,14 @@ impl PoolingBackend {
let (client, mut discard) = handle.inner();
debug!("setting up backend session state");
// // initiates the auth session
// if let Err(e) = client.batch_execute("select auth.init();").await {
// discard.discard();
// return Err(e.into());
// }
// initiates the auth session
if !disable_pg_session_jwt {
if let Err(e) = client.batch_execute("select auth.init();").await {
discard.discard();
return Err(e.into());
}
}
info!("backend session state initialized");
}