-
Notifications
You must be signed in to change notification settings - Fork 210
/
migrations.go
87 lines (79 loc) · 2.08 KB
/
migrations.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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package sql
import (
"bufio"
"bytes"
"embed"
"fmt"
"path"
"sort"
"strconv"
"strings"
)
//go:embed migrations/*.sql
var embedded embed.FS
type migration struct {
order int
name string
content *bufio.Scanner
}
// Migrations is interface for migrations provider.
type Migrations func(Executor) error
func embeddedMigrations(db Executor) error {
files, err := embedded.ReadDir("migrations")
if err != nil {
return fmt.Errorf("readdir migrations: %w", err)
}
var migrations []migration
for _, file := range files {
parts := strings.Split(file.Name(), "_")
if len(parts) < 1 {
return fmt.Errorf("invalid migration %s", file.Name())
}
order, err := strconv.Atoi(parts[0])
if err != nil {
return fmt.Errorf("invalid migration %s: %w", file.Name(), err)
}
fpath := path.Join("migrations", file.Name())
content, err := embedded.ReadFile(fpath)
if err != nil {
return fmt.Errorf("readfile %s: %w", fpath, err)
}
scanner := bufio.NewScanner(bytes.NewBuffer(content))
scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {
if i := bytes.Index(data, []byte(";")); i >= 0 {
return i + 1, data[0 : i+1], nil
}
return 0, nil, nil
})
migrations = append(migrations, migration{
order: order,
name: file.Name(),
content: scanner,
})
}
sort.Slice(migrations, func(i, j int) bool {
return migrations[i].order < migrations[j].order
})
var current int
if _, err := db.Exec("PRAGMA user_version;", nil, func(stmt *Statement) bool {
current = stmt.ColumnInt(0)
return true
}); err != nil {
return fmt.Errorf("read user_version %w", err)
}
for _, m := range migrations {
if m.order <= current {
continue
}
for m.content.Scan() {
if _, err := db.Exec(m.content.Text(), nil, nil); err != nil {
return fmt.Errorf("exec %s: %w", m.content.Text(), err)
}
}
// binding values in pragma statement is not allowed
if _, err := db.Exec(fmt.Sprintf("PRAGMA user_version = %d;", m.order), nil, nil); err != nil {
return fmt.Errorf("update user_version to %d: %w", m.order, err)
}
}
return nil
}