Added initial support for PostgreSQL

This commit is contained in:
Deluan
2020-01-14 19:20:47 -05:00
parent a167669717
commit 9922ba5994
5 changed files with 31 additions and 8 deletions
+11 -3
View File
@@ -1,17 +1,22 @@
package persistence
import (
"strings"
"sync"
"github.com/astaxie/beego/orm"
"github.com/cloudsonic/sonic-server/conf"
"github.com/cloudsonic/sonic-server/log"
_ "github.com/lib/pq"
_ "github.com/mattn/go-sqlite3"
)
const batchSize = 100
var once sync.Once
var (
once sync.Once
driver = "sqlite3"
)
func Db() orm.Ormer {
once.Do(func() {
@@ -23,7 +28,7 @@ func Db() orm.Ormer {
if err != nil {
panic(err)
}
log.Debug("Opening SQLite DB from: " + dbPath)
log.Debug("Opening DB from: "+dbPath, "driver", driver)
})
return orm.NewOrm()
}
@@ -62,7 +67,10 @@ func initORM(dbPath string) error {
orm.RegisterModel(new(Property))
orm.RegisterModel(new(Playlist))
orm.RegisterModel(new(Search))
err := orm.RegisterDataBase("default", "sqlite3", dbPath)
if strings.Contains(dbPath, "postgres") {
driver = "postgres"
}
err := orm.RegisterDataBase("default", driver, dbPath)
if err != nil {
panic(err)
}