67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
|
package dbsetup
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/fs"
|
||
|
|
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
"github.com/pressly/goose/v3"
|
||
|
|
"github.com/uptrace/opentelemetry-go-extra/otelsqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DatabaseConfig struct {
|
||
|
|
DatabaseURL string `name:"postgres-url" env:"POSTGRES_URL" help:"URL to use to connect to Postgres" default:"postgres://postgres:postgres@:5432/postgres?sslmode=disable"`
|
||
|
|
DatabaseDriverName string `name:"db-driver" env:"DB_DRIVER" help:"Driver to use to connect to db" default:"postgres"`
|
||
|
|
MaxOpenConnections int `name:"max-open-connections" env:"DB_MAX_CONNECTIONS" help:"Max open connections to db" default:"50"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Database struct {
|
||
|
|
cfg DatabaseConfig
|
||
|
|
}
|
||
|
|
|
||
|
|
func SetupDB(cfg DatabaseConfig) (*Database, error) {
|
||
|
|
db, err := sqlx.Connect(cfg.DatabaseDriverName, cfg.DatabaseURL)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if err := db.Close(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &Database{
|
||
|
|
cfg: cfg,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *Database) Connect() (*sqlx.DB, error) {
|
||
|
|
db, err := otelsqlx.Connect(d.cfg.DatabaseDriverName, d.cfg.DatabaseURL)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
db.SetMaxOpenConns(d.cfg.MaxOpenConnections)
|
||
|
|
return db, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *Database) RunMigrations(tableName, dir string, fs fs.FS) error {
|
||
|
|
db, err := connect(d.cfg)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
goose.SetTableName(tableName)
|
||
|
|
goose.SetBaseFS(fs)
|
||
|
|
if err := goose.SetDialect(d.gooseDriverName()); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return goose.Up(db.DB, dir)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *Database) gooseDriverName() string {
|
||
|
|
if d.cfg.DatabaseDriverName == "sqlmock" {
|
||
|
|
return "postgres"
|
||
|
|
}
|
||
|
|
return d.cfg.DatabaseDriverName
|
||
|
|
}
|
||
|
|
|
||
|
|
func connect(cfg DatabaseConfig) (*sqlx.DB, error) {
|
||
|
|
return sqlx.Connect(cfg.DatabaseDriverName, cfg.DatabaseURL)
|
||
|
|
}
|