add: server shutdown, tests
This commit is contained in:
@@ -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()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user