feat: initial version

database.go

@@ -0,0 +1,66 @@
+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)
+}