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

storcon: cut over to the hadron impl for establishing connection

Browse Source
This commit is contained in:
Vlad Lazar
2025-07-25 17:51:04 +01:00
parent a1924e72ad
commit 7bd73eba72
1 changed files with 1 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
storage_controller/src/persistence.rs
View File

@@ -2368,30 +2368,13 @@ fn client_config_with_root_certs() -> anyhow::Result<rustls::ClientConfig> {
})
}
fn establish_connection_rustls(config: &str) -> BoxFuture<ConnectionResult<AsyncPgConnection>> {
let fut = async {
// We first set up the way we want rustls to work.
let rustls_config = client_config_with_root_certs()
.map_err(|err| ConnectionError::BadConnection(format!("{err:?}")))?;
let tls = tokio_postgres_rustls::MakeRustlsConnect::new(rustls_config);
let (client, conn) = tokio_postgres::connect(config, tls)
.await
.map_err(|e| ConnectionError::BadConnection(e.to_string()))?;
AsyncPgConnection::try_from_client_and_connection(client, conn).await
};
fut.boxed()
}
// Hadron's implementation of establish_connection_rustls which avoids hogging the tokio executor thread during
// CPU-intensive operations in postgres connection and session establishments.
// Compared to the original implementation this function performs the following tasks using spawn_blocking to avoid
// hogging the tokio executor thread:
// 1. Parsing and decoding root certificates during rustls client config setup.
// 2. The tokio_postgres::connect() call, which performs the TLS handshake and the postgres password authentication.
fn establish_connection_rustls_no_hog_thread(
config: &str,
) -> BoxFuture<ConnectionResult<AsyncPgConnection>> {
fn establish_connection_rustls(config: &str) -> BoxFuture<ConnectionResult<AsyncPgConnection>> {
let fut = async move {
// We first set up the way we want rustls to work.
let rustls_config = tokio::task::spawn_blocking(client_config_with_root_certs)