TheLovinator/feedvault.se
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Close DB on exit

Browse source
This commit is contained in:
Joakim Hellsén 2024-02-04 05:36:51 +01:00
commit 366820bac8
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
main.go
View file

@ -16,9 +16,6 @@ import (
"gorm.io/gorm" "gorm.io/gorm"
) )
// Database connection
var db *gorm.DB
func main() { func main() {
log.Println("Starting FeedVault...") log.Println("Starting FeedVault...")
db, err := gorm.Open(sqlite.Open("feedvault.db"), &gorm.Config{}) db, err := gorm.Open(sqlite.Open("feedvault.db"), &gorm.Config{})
@ -26,6 +23,12 @@ func main() {
panic("Failed to connect to database") 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 // Migrate the schema
err = db.AutoMigrate(&Feed{}, &Item{}, &Person{}, &Image{}, &Enclosure{}, &DublinCoreExtension{}, &ITunesFeedExtension{}, &ITunesItemExtension{}, &ITunesCategory{}, &ITunesOwner{}, &Extension{}) err = db.AutoMigrate(&Feed{}, &Item{}, &Person{}, &Image{}, &Enclosure{}, &DublinCoreExtension{}, &ITunesFeedExtension{}, &ITunesItemExtension{}, &ITunesCategory{}, &ITunesOwner{}, &Extension{})
if err != nil { if err != nil {