Redmage/db/db.go

57 lines
1.3 KiB
Go
Raw Normal View History

package db
import (
"database/sql"
"io/fs"
_ "github.com/mattn/go-sqlite3"
"github.com/pressly/goose/v3"
2024-04-08 15:48:45 +07:00
sqldblogger "github.com/simukti/sqldb-logger"
"github.com/tigorlazuardi/redmage/config"
2024-04-07 23:41:00 +07:00
"github.com/tigorlazuardi/redmage/pkg/errs"
2024-04-13 00:15:31 +07:00
semconv "go.opentelemetry.io/otel/semconv/v1.4.0"
"github.com/XSAM/otelsql"
)
var Migrations fs.FS
func Open(cfg *config.Config) (*sql.DB, error) {
2024-04-25 20:22:05 +07:00
driver := cfg.String("db.driver")
2024-04-25 13:05:41 +07:00
dsn := cfg.String("db.string")
db, err := OpenSilent(cfg)
if err != nil {
return db, err
}
2024-04-25 20:22:05 +07:00
if cfg.Bool("db.automigrate") {
goose.SetLogger(&gooseLogger{})
goose.SetBaseFS(Migrations)
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 {
return db, errs.Wrapw(err, "failed to migrate database", "dialect", driver)
}
}
2024-04-25 13:05:41 +07:00
db = sqldblogger.OpenDriver(dsn, db.Driver(), sqlLogger{},
sqldblogger.WithSQLQueryAsMessage(true),
)
return db, err
}
func OpenSilent(cfg *config.Config) (*sql.DB, error) {
2024-04-07 23:41:00 +07:00
driver := cfg.String("db.driver")
2024-04-08 15:48:45 +07:00
dsn := cfg.String("db.string")
2024-04-13 00:15:31 +07:00
db, err := otelsql.Open(driver, dsn, otelsql.WithAttributes(
semconv.DBSystemSqlite,
))
if err != nil {
2024-04-07 23:41:00 +07:00
return db, errs.Wrapw(err, "failed to open database", "driver", driver)
}
return db, err
}