Redmage/db/db.go

37 lines
852 B
Go
Raw Normal View History

package db
import (
"database/sql"
"io/fs"
_ "github.com/mattn/go-sqlite3"
"github.com/pressly/goose/v3"
"github.com/tigorlazuardi/redmage/config"
2024-04-07 23:41:00 +07:00
"github.com/tigorlazuardi/redmage/pkg/errs"
)
var Migrations fs.FS
func Open(cfg *config.Config) (*sql.DB, error) {
2024-04-07 23:41:00 +07:00
driver := cfg.String("db.driver")
db, err := sql.Open(driver, cfg.String("db.string"))
if err != nil {
2024-04-07 23:41:00 +07:00
return db, errs.Wrapw(err, "failed to open database", "driver", driver)
}
if cfg.Bool("db.automigrate") {
2024-04-07 23:41:00 +07:00
goose.SetLogger(&gooseLogger{})
goose.SetBaseFS(Migrations)
2024-04-07 23:41:00 +07:00
if err := goose.SetDialect(driver); err != nil {
return db, errs.Wrapw(err, "failed to set goose dialect", "dialect", driver)
}
if err := goose.Up(db, "db/migrations"); err != nil {
2024-04-07 23:41:00 +07:00
return db, errs.Wrapw(err, "failed to migrate database", "dialect", driver)
}
}
return db, err
}