Update SlickOffsetStore.scala

apache · Jun 12, 2024 · 589b32f · 589b32f
1 parent aa6862f
commit 589b32f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/slick/src/main/scala/org/apache/pekko/projection/slick/internal/SlickOffsetStore.scala b/slick/src/main/scala/org/apache/pekko/projection/slick/internal/SlickOffsetStore.scala
@@ -45,7 +45,7 @@ import slick.jdbc.JdbcProfile
     system: ActorSystem[_],
     databaseConfig: DatabaseConfig[P],
     slickSettings: SlickSettings,
-    clock: Clock) {
+    clock: Clock) extends AutoCloseable {
 
   def this(system: ActorSystem[_], databaseConfig: DatabaseConfig[P], slickSettings: SlickSettings) =
     this(system, databaseConfig, slickSettings, Clock.systemUTC())
@@ -220,4 +220,8 @@ import slick.jdbc.JdbcProfile
 
     db.run(action).map(_ => Done)(ExecutionContexts.parasitic)
   }
+
+  def close(): Unit = {
+    db.close()
+  }
 }