make pg_session_jwt instalation optional with a cli flag

This commit is contained in:
Ruslan Talpa
2025-06-20 10:17:32 +03:00
parent 3e36d516c2
commit 6bd15908fb
16 changed files with 87 additions and 33 deletions

View File

@@ -731,9 +731,11 @@ async fn handle_db_inner(
ComputeCredentialKeys::JwtPayload(payload)
if backend.auth_backend.is_local_proxy() =>
{
let mut client = backend.connect_to_local_postgres(ctx, conn_info).await?;
let (cli_inner, _dsc) = client.client_inner();
cli_inner.set_jwt_session(&payload).await?;
let mut client = backend.connect_to_local_postgres(ctx, conn_info, config.disable_pg_session_jwt).await?;
if !config.disable_pg_session_jwt {
let (cli_inner, _dsc) = client.client_inner();
cli_inner.set_jwt_session(&payload).await?;
}
Client::Local(client)
}
_ => {