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
88
89
90
91
92
93
94
|
package main
import (
"flag"
"fmt"
"io"
"time"
migrate "github.com/rubenv/sql-migrate"
"gitlab.com/gitlab-org/gitaly/internal/praefect/config"
"gitlab.com/gitlab-org/gitaly/internal/praefect/datastore/glsql"
"gitlab.com/gitlab-org/gitaly/internal/praefect/datastore/migrations"
)
const (
sqlMigrateCmdName = "sql-migrate"
timeFmt = "2006-01-02T15:04:05"
)
type sqlMigrateSubcommand struct {
w io.Writer
ignoreUnknown bool
verbose bool
}
func newSQLMigrateSubCommand(writer io.Writer) *sqlMigrateSubcommand {
return &sqlMigrateSubcommand{w: writer}
}
func (cmd *sqlMigrateSubcommand) FlagSet() *flag.FlagSet {
flags := flag.NewFlagSet(sqlMigrateCmdName, flag.ExitOnError)
flags.BoolVar(&cmd.ignoreUnknown, "ignore-unknown", true, "ignore unknown migrations (default is true)")
flags.BoolVar(&cmd.verbose, "verbose", false, "show text of migration query (default is false)")
return flags
}
func (cmd *sqlMigrateSubcommand) Exec(flags *flag.FlagSet, conf config.Config) error {
const subCmd = progname + " " + sqlMigrateCmdName
db, clean, err := openDB(conf.DB)
if err != nil {
return err
}
defer clean()
migrationSet := migrate.MigrationSet{
IgnoreUnknown: cmd.ignoreUnknown,
TableName: migrations.MigrationTableName,
}
planSource := &migrate.MemoryMigrationSource{
Migrations: migrations.All(),
}
// Find all migrations that are currently down.
planMigrations, _, _ := migrationSet.PlanMigration(db, "postgres", planSource, migrate.Up, 0)
if len(planMigrations) == 0 {
fmt.Fprintf(cmd.w, "%s: all migrations are up\n", subCmd)
return nil
}
fmt.Fprintf(cmd.w, "%s: migrations to apply: %d\n\n", subCmd, len(planMigrations))
executed := 0
for _, mig := range planMigrations {
fmt.Fprintf(cmd.w, "= %s %v: migrating\n", time.Now().Format(timeFmt), mig.Id)
start := time.Now()
if cmd.verbose {
fmt.Fprintf(cmd.w, "\t%v\n", mig.Up)
}
n, err := glsql.MigrateSome(mig.Migration, db, cmd.ignoreUnknown)
if err != nil {
return fmt.Errorf("%s: fail: %v", time.Now().Format(timeFmt), err)
}
if n > 0 {
fmt.Fprintf(cmd.w, "== %s %v: applied (%s)\n", time.Now().Format(timeFmt), mig.Id, time.Since(start))
// Additional migrations were run. No harm, but prevents us from tracking their execution duration.
if n > 1 {
fmt.Fprintf(cmd.w, "warning: %v additional migrations were applied successfully\n", n-1)
}
} else {
fmt.Fprintf(cmd.w, "== %s %v: skipped (%s)\n", time.Now().Format(timeFmt), mig.Id, time.Since(start))
}
executed += n
}
fmt.Fprintf(cmd.w, "\n%s: OK (applied %d migrations)\n", subCmd, executed)
return nil
}
|
