Some checks failed
Backend Tests / Static Checks (push) Has been cancelled
Backend Tests / Tests (other) (push) Has been cancelled
Backend Tests / Tests (plugin) (push) Has been cancelled
Backend Tests / Tests (server) (push) Has been cancelled
Backend Tests / Tests (store) (push) Has been cancelled
Build Canary Image / build-frontend (push) Has been cancelled
Build Canary Image / build-push (linux/amd64) (push) Has been cancelled
Build Canary Image / build-push (linux/arm64) (push) Has been cancelled
Build Canary Image / merge (push) Has been cancelled
Frontend Tests / Lint (push) Has been cancelled
Frontend Tests / Build (push) Has been cancelled
Proto Linter / Lint Protos (push) Has been cancelled
33 lines
781 B
Go
33 lines
781 B
Go
package db
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/usememos/memos/internal/profile"
|
|
"github.com/usememos/memos/store"
|
|
"github.com/usememos/memos/store/db/mysql"
|
|
"github.com/usememos/memos/store/db/postgres"
|
|
"github.com/usememos/memos/store/db/sqlite"
|
|
)
|
|
|
|
// NewDBDriver creates new db driver based on profile.
|
|
func NewDBDriver(profile *profile.Profile) (store.Driver, error) {
|
|
var driver store.Driver
|
|
var err error
|
|
|
|
switch profile.Driver {
|
|
case "sqlite":
|
|
driver, err = sqlite.NewDB(profile)
|
|
case "mysql":
|
|
driver, err = mysql.NewDB(profile)
|
|
case "postgres":
|
|
driver, err = postgres.NewDB(profile)
|
|
default:
|
|
return nil, errors.New("unknown db driver")
|
|
}
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to create db driver")
|
|
}
|
|
return driver, nil
|
|
}
|