log on postgres reconnect

blockworks-foundation · Mar 20, 2024 · 636dbfc · 636dbfc
1 parent de0383e
commit 636dbfc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/benchrunner-service/src/postgres/postgres_session.rs b/benchrunner-service/src/postgres/postgres_session.rs
@@ -134,7 +134,7 @@ impl PostgresSession {
             log::info!("Connecting to Postgres");
 
             if let Err(err) = connection.await {
-                log::error!("Connection to Postgres broke {err:?}");
+                log::error!("Connection to Postgres broke: {err:?}");
                 return;
             }
             log::debug!("Postgres thread shutting down");

diff --git a/benchrunner-service/src/postgres/postgres_session_cache.rs b/benchrunner-service/src/postgres/postgres_session_cache.rs
@@ -1,4 +1,5 @@
 use std::sync::Arc;
+use log::info;
 use tokio::sync::RwLock;
 use crate::postgres::postgres_session::{PostgresSession, PostgresSessionConfig};
 
@@ -20,6 +21,7 @@ impl PostgresSessionCache {
     pub async fn get_session(&self) -> anyhow::Result<PostgresSession> {
         let session = self.session.read().await;
         if session.is_closed() {
+            info!("PostgreSQL session closed - reconnecting");
             drop(session);
             let session = PostgresSession::new(self.config.as_ref().clone()).await?;
             *self.session.write().await = session.clone();