graceful close on shutdown

src/adapters/storage/postgresql.rs

@@ -47,6 +47,12 @@ impl Postgres {
         Ok(Self { connection_pool })
     }
 
+    /// Close only when shutting down the application. Will close the connection pool even if it is
+    /// shared.
+    pub async fn close(self) {
+        self.connection_pool.close().await;
+    }
+
     pub async fn migrate(&self) -> Result<()> {
         sqlx::migrate!().run(&self.connection_pool).await?;
         Ok(())

src/main.rs

@@ -62,7 +62,7 @@ async fn main() -> Result<()> {
     launch_api_server(
         &config,
         metrics_registry,
-        storage,
+        storage.clone(),
         fuel_health_check,
         eth_health_check,
     )
@@ -74,6 +74,7 @@ async fn main() -> Result<()> {
         wallet_balance_tracker_handle,
         committer_handle,
         listener_handle,
+        storage,
     )
     .await
 }

src/setup/helpers.rs

@@ -201,6 +201,7 @@ pub async fn shut_down(
     wallet_balance_tracker_handle: JoinHandle<()>,
     committer_handle: JoinHandle<()>,
     listener_handle: JoinHandle<()>,
+    storage: Postgres,
 ) -> Result<()> {
     cancel_token.cancel();
 
@@ -213,5 +214,6 @@ pub async fn shut_down(
         handle.await?;
     }
 
+    storage.close().await;
     Ok(())
 }