From 366820bac8abb2e9566cd946f26642f87f3fd710 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Hells=C3=A9n?= <tlovinator@gmail.com>
Date: Sun, 4 Feb 2024 05:36:51 +0100
Subject: [PATCH] Close DB on exit

---
 main.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index aa2488d..c196811 100644
--- a/main.go
+++ b/main.go
@@ -16,9 +16,6 @@ import (
 	"gorm.io/gorm"
 )
 
-// Database connection
-var db *gorm.DB
-
 func main() {
 	log.Println("Starting FeedVault...")
 	db, err := gorm.Open(sqlite.Open("feedvault.db"), &gorm.Config{})
@@ -26,6 +23,12 @@ func main() {
 		panic("Failed to connect to database")
 	}
 
+	sqlDB, err := db.DB()
+	if err != nil {
+		panic("Failed to get database connection")
+	}
+	defer sqlDB.Close()
+
 	// Migrate the schema
 	err = db.AutoMigrate(&Feed{}, &Item{}, &Person{}, &Image{}, &Enclosure{}, &DublinCoreExtension{}, &ITunesFeedExtension{}, &ITunesItemExtension{}, &ITunesCategory{}, &ITunesOwner{}, &Extension{})
 	if err != nil {