Applying comment

raw-labs · Jun 24, 2024 · 55fa740 · 55fa740
1 parent a992994
commit 55fa740
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sql-client/src/main/scala/raw/client/sql/SqlConnectionPool.scala b/sql-client/src/main/scala/raw/client/sql/SqlConnectionPool.scala
@@ -11,7 +11,7 @@
  */
 
 package raw.client.sql
-import com.google.common.cache.{CacheBuilder, CacheLoader, LoadingCache}
+import com.google.common.cache.{CacheBuilder, CacheLoader, LoadingCache, RemovalNotification}
 import com.typesafe.scalalogging.StrictLogging
 import com.zaxxer.hikari.{HikariConfig, HikariDataSource}
 import raw.creds.api.CredentialsService
@@ -68,6 +68,10 @@ class SqlConnectionPool(credentialsService: CredentialsService)(implicit setting
     .newBuilder()
     .maximumSize(connectionCacheSize)
     .expireAfterAccess(connectionCachePeriod)
+    .removalListener((notification: RemovalNotification[AuthenticatedUser, HikariDataSource]) => {
+      logger.info(s"Shutting down SQL connection pool for database ${notification.getValue.getJdbcUrl}")
+      RawUtils.withSuppressNonFatalException(notification.getValue.close())
+    })
     .build(dbCacheLoader)
 
   @throws[SQLException]