add: server shutdown, tests

This commit is contained in:
$(pass /github/name)
2024-08-20 21:56:36 +02:00
parent 6f42f715a7
commit 15d80889d1
7 changed files with 105 additions and 18 deletions

View File

@@ -3,13 +3,14 @@ package database
import (
"GoMembership/internal/models"
"GoMembership/pkg/logger"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
var DB *gorm.DB
func InitDB(dbPath string) error {
func Open(dbPath string) error {
db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
if err != nil {
@@ -26,5 +27,16 @@ func InitDB(dbPath string) error {
return err
}
DB = db
logger.Info.Print("Opened DB")
return nil
}
func Close() error {
logger.Info.Print("Closing DB")
db, err := DB.DB()
if err != nil {
return err
}
return db.Close()
}