diff --git a/internal/db/db.go b/internal/db/db.go index 3bf6685..b29cb7c 100644 --- a/internal/db/db.go +++ b/internal/db/db.go @@ -9,11 +9,11 @@ import ( ) func InitDb(rss models.Rss2, dbpath string) error { - log.Println("Initialize DB") - db, err := bolt.Open(dbpath, 0600, nil) + db, err := openDB(dbpath) if err != nil { - log.Fatal(err) + return err } + log.Println("Initialize DB") defer db.Close() err = db.Update(func(tx *bolt.Tx) error { @@ -72,11 +72,11 @@ func FindItems(rss models.Rss2, dbpath string) (*models.SendItems, error) { } func UpdateDb(dbpath string, senditems *models.SendItems) error { - log.Println("Update DB") - db, err := bolt.Open(dbpath, 0600, nil) + db, err := openDB(dbpath) if err != nil { - log.Fatal(err) + return err } + log.Println("Update DB") defer db.Close() err = db.Update(func(tx *bolt.Tx) error { @@ -98,3 +98,7 @@ func UpdateDb(dbpath string, senditems *models.SendItems) error { } return nil } + +func openDB(dbpath string) (*bolt.DB, error) { + return bolt.Open(dbpath, 0600, nil) +}