-
Notifications
You must be signed in to change notification settings - Fork 0
/
dbmigrate.go
73 lines (60 loc) · 1.66 KB
/
dbmigrate.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
// Package dbmigrate contains the databse schema, migrations, and seeding data.
package dbmigrate
import (
"context"
"database/sql"
_ "embed"
"errors"
"fmt"
database "github.com/aleury/service/business/sys/database/pgx"
"github.com/ardanlabs/darwin/v3"
"github.com/ardanlabs/darwin/v3/dialects/postgres"
"github.com/ardanlabs/darwin/v3/drivers/generic"
"github.com/jmoiron/sqlx"
)
var (
//go:embed sql/migrate.sql
migrateDoc string
//go:embed sql/seed.sql
seedDoc string
)
// Migrate attempts to bring the database up to date with the migrations
// defined in this package.
func Migrate(ctx context.Context, db *sqlx.DB) error {
if err := database.StatusCheck(ctx, db); err != nil {
return fmt.Errorf("status check database: %w", err)
}
driver, err := generic.New(db.DB, postgres.Dialect{})
if err != nil {
return fmt.Errorf("construct darwin driver: %w", err)
}
d := darwin.New(driver, darwin.ParseMigrations(migrateDoc))
return d.Migrate()
}
// Seed runs the seed document defined in this package against the db.
// The queries are run in a transaction and rolled back if any fail.
func Seed(ctx context.Context, db *sqlx.DB) error {
if err := database.StatusCheck(ctx, db); err != nil {
return fmt.Errorf("status check database: %w", err)
}
tx, err := db.Begin()
if err != nil {
return err
}
defer func() {
if errTx := tx.Rollback(); errTx != nil {
if errors.Is(errTx, sql.ErrTxDone) {
return
}
err = fmt.Errorf("rollback: %w", errTx)
return
}
}()
if _, err := tx.Exec(seedDoc); err != nil {
return fmt.Errorf("exec: %w", err)
}
if err := tx.Commit(); err != nil {
return fmt.Errorf("commit: %w", err)
}
return nil
}