bluebell/initdb.go

39 lines
870 B
Go
Raw Normal View History

2024-08-10 13:28:30 +01:00
package bluebell
import (
"database/sql"
"embed"
"errors"
"github.com/1f349/bluebell/database"
2024-08-16 16:48:50 +01:00
"github.com/golang-migrate/migrate/v4"
2024-08-10 13:28:30 +01:00
"github.com/golang-migrate/migrate/v4/database/mysql"
"github.com/golang-migrate/migrate/v4/source/iofs"
)
//go:embed database/migrations/*.sql
var migrations embed.FS
func InitDB(p string) (*database.Queries, error) {
migDrv, err := iofs.New(migrations, "database/migrations")
if err != nil {
return nil, err
}
dbOpen, err := sql.Open("mysql", p)
if err != nil {
return nil, err
}
dbDrv, err := mysql.WithInstance(dbOpen, &mysql.Config{})
if err != nil {
return nil, err
}
mig, err := migrate.NewWithInstance("iofs", migDrv, "mysql", dbDrv)
if err != nil {
return nil, err
}
err = mig.Up()
if err != nil && !errors.Is(err, migrate.ErrNoChange) {
return nil, err
}
return database.New(dbOpen), nil
}