Fix: exit auto close db conn

pull/134/head
zijiren233 2 years ago
parent 0f7e0df299
commit 871310e4f1

@ -11,6 +11,7 @@ import (
"github.com/synctv-org/synctv/cmd/flags"
"github.com/synctv-org/synctv/internal/conf"
"github.com/synctv-org/synctv/internal/db"
sysnotify "github.com/synctv-org/synctv/internal/sysNotify"
"github.com/synctv-org/synctv/internal/version"
"github.com/synctv-org/synctv/utils"
"gorm.io/driver/mysql"
@ -41,6 +42,12 @@ func InitDatabase(ctx context.Context) (err error) {
if err != nil {
log.Fatalf("failed to get sqlDB: %s", err.Error())
}
err = sysnotify.RegisterSysNotifyTask(0, sysnotify.NewSysNotifyTask("database", sysnotify.NotifyTypeEXIT, func() error {
return sqlDB.Close()
}))
if err != nil {
log.Fatalf("failed to register sysnotify task: %s", err.Error())
}
if conf.Conf.Database.Type != conf.DatabaseTypeSqlite3 {
initRawDB(sqlDB)
}

Loading…
Cancel
Save