perf(db): release database before exit

This commit is contained in:
Andy Hsu
2023-09-05 13:04:27 +08:00
parent b604e21c69
commit 3231c3d930
6 changed files with 26 additions and 1 deletions

View File

@ -31,3 +31,17 @@ func AutoMigrate(dst ...interface{}) error {
func GetDb() *gorm.DB {
return db
}
func Close() {
log.Info("closing db")
sqlDB, err := db.DB()
if err != nil {
log.Errorf("failed to get db: %s", err.Error())
return
}
err = sqlDB.Close()
if err != nil {
log.Errorf("failed to close db: %s", err.Error())
return
}
}