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

box authenticate task

Browse Source
This commit is contained in:
Conrad Ludgate
2025-05-29 15:55:17 +01:00
parent cfa2813446
commit 7d1c908b1b
1 changed files with 28 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
proxy/src/proxy/mod.rs
View File

@@ -354,30 +354,36 @@ pub(crate) async fn handle_client<S: AsyncRead + AsyncWrite + Unpin + Send>(
return Ok(None);
};
let user = user_info.user.clone();
let compute_creds = match cplane
.authenticate(
ctx,
&mut stream,
user_info,
mode.allow_cleartext(),
&config.authentication_config,
endpoint_rate_limiter,
)
.await
{
Ok(auth_result) => auth_result,
Err(e) => {
let db = params.get("database");
let app = params.get("application_name");
let params_span = tracing::info_span!("", ?user, ?db, ?app);
let auth_result: Result<_, ClientRequestError> = async {
let user = user_info.user.clone();
return stream
.throw_error(e, Some(ctx))
.instrument(params_span)
.await?;
match cplane
.authenticate(
ctx,
&mut stream,
user_info,
mode.allow_cleartext(),
&config.authentication_config,
endpoint_rate_limiter,
)
.await
{
Ok(auth_result) => Ok(auth_result),
Err(e) => {
let db = params.get("database");
let app = params.get("application_name");
let params_span = tracing::info_span!("", ?user, ?db, ?app);
stream
.throw_error(e, Some(ctx))
.instrument(params_span)
.await?
}
}
};
}
.boxed()
.await;
let compute_creds = auth_result?;
let compute_user_info = compute_creds.info.clone();
let params_compat = compute_user_info