mirrors/navidrome
1
0
Fork 0
mirror of https://github.com/navidrome/navidrome.git synced 2025-04-04 04:57:37 +03:00
Code Issues Projects Releases Packages Wiki Activity

refactor: better integration between db and persistence packages

Browse source 
Will address support for different DBs in the future (+1 squashed commit)
Squashed commits:
[a014757] refactor: better integration between `db` and `persistence` packages
This commit is contained in:
Deluan 2020-02-01 13:46:03 -05:00 • committed by Deluan Quintão
parent 76ca8afc84
commit 7e65bb8f20
4 changed files with 30 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

27
db/db.go
View file

@ -3,6 +3,7 @@ package db
import (
"database/sql"
"os"
"sync"
"github.com/deluan/navidrome/conf"
_ "github.com/deluan/navidrome/db/migrations"
@ -11,19 +12,35 @@ import (
"github.com/pressly/goose"
)
const driver = "sqlite3"
var (
once sync.Once
Driver = "sqlite3"
Path string
)
func EnsureDB() {
db, err := sql.Open(driver, conf.Server.DbPath)
func Init() {
once.Do(func() {
Path = conf.Server.DbPath
if Path == ":memory:" {
Path = "file::memory:?cache=shared"
conf.Server.DbPath = Path
}
log.Debug("Opening DataBase", "dbPath", Path, "driver", Driver)
})
}
func EnsureLatestVersion() {
Init()
db, err := sql.Open(Driver, Path)
defer db.Close()
if err != nil {
log.Error("Failed to open DB", err)
os.Exit(1)
}
err = goose.SetDialect(driver)
err = goose.SetDialect(Driver)
if err != nil {
log.Error("Invalid DB driver", "driver", driver, err)
log.Error("Invalid DB driver", "driver", Driver, err)
os.Exit(1)
}
err = goose.Run("up", db, "./")

2
main.go
View file

@ -11,7 +11,7 @@ func main() {
}
conf.Load()
db.EnsureDB()
db.EnsureLatestVersion()
a := CreateServer(conf.Server.MusicFolder)
a.MountRouter("/rest", CreateSubsonicAPIRouter())

21
persistence/persistence.go
View file

@ -3,18 +3,16 @@ package persistence
import (
"context"
"reflect"
"strings"
"sync"
"github.com/astaxie/beego/orm"
"github.com/deluan/navidrome/conf"
"github.com/deluan/navidrome/db"
"github.com/deluan/navidrome/log"
"github.com/deluan/navidrome/model"
)
var (
once sync.Once
driver = "sqlite3"
once sync.Once
)
type SQLStore struct {
@ -23,13 +21,7 @@ type SQLStore struct {
func New() model.DataStore {
once.Do(func() {
dbPath := conf.Server.DbPath
if dbPath == ":memory:" {
dbPath = "file::memory:?cache=shared"
}
log.Debug("Opening DataBase", "dbPath", dbPath, "driver", driver)
err := initORM(dbPath)
err := orm.RegisterDataBase("default", db.Driver, db.Path)
if err != nil {
panic(err)
}
@ -147,10 +139,3 @@ func (db *SQLStore) getOrmer() orm.Ormer {
}
return db.orm
}
func initORM(dbPath string) error {
if strings.Contains(dbPath, "postgres") {
driver = "postgres"
}
return orm.RegisterDataBase("default", driver, dbPath)
}

7
persistence/persistence_suite_test.go
View file

@ -21,10 +21,11 @@ func TestPersistence(t *testing.T) {
tests.Init(t, true)
//os.Remove("./test-123.db")
//conf.Server.DbPath = "./test-123.db"
conf.Server.DbPath = "file::memory:?cache=shared"
//conf.Server.Path = "./test-123.db"
conf.Server.DbPath = ":memory:"
db.Init()
New()
db.EnsureDB()
db.EnsureLatestVersion()
log.SetLevel(log.LevelCritical)
RegisterFailHandler(Fail)
RunSpecs(t, "Persistence Suite")