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

proxy: remove connections on shutdown (#5051)

Browse Source 
## Problem

On shutdown, proxy connections are staying open.

## Summary of changes

Remove the connections on shutdown
This commit is contained in:
Conrad Ludgate
2023-08-21 19:20:58 +01:00
committed by GitHub
parent 4a8bd866f6
commit 0b001a0001
3 changed files with 54 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
proxy/src/http/sql_over_http.rs
View File

@@ -16,7 +16,6 @@ use tokio_postgres::types::Type;
use tokio_postgres::GenericClient;
use tokio_postgres::IsolationLevel;
use tokio_postgres::Row;
use tracing::Instrument;
use url::Url;
use super::conn_pool::ConnInfo;
@@ -286,13 +285,12 @@ pub async fn handle(
};
if allow_pool {
let current_span = tracing::Span::current();
// return connection to the pool
tokio::task::spawn(
async move {
let _ = conn_pool.put(&conn_info, client).await;
}
.in_current_span(),
);
tokio::task::spawn_blocking(move || {
let _span = current_span.enter();
let _ = conn_pool.put(&conn_info, client);
});
}
result