Safely close the connection to the sqlite3 database on Ctrl^C.

ronin-rb · May 3, 2024 · 4e143f5 · 4e143f5
1 parent 5ada7f7
commit 4e143f5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/config/database.rb b/config/database.rb
@@ -11,3 +11,7 @@
             end
 
 Ronin::DB.connect(database, pool: pool_size)
+
+at_exit do
+  ActiveRecord::Base.connection_pool.disconnect!
+end