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
95
96
97
98
99
|
package cmd
import (
"database/sql"
"fmt"
"github.com/golang-migrate/migrate/v4"
"github.com/golang-migrate/migrate/v4/database/sqlite"
"github.com/golang-migrate/migrate/v4/source/iofs"
"github.com/spf13/cobra"
"gitrepo.ru/neonxp/gorum/migrations"
)
var migrateCmd = &cobra.Command{
Use: "migrate",
Short: "Migrate db",
Long: `Up and down migrations`,
}
func init() {
migrateCmd.AddCommand(
&cobra.Command{
Use: "up",
Short: "Migrate up",
Long: `Up migrations`,
RunE: up,
},
&cobra.Command{
Use: "down",
Short: "Migrate down",
Long: `Down migrations`,
RunE: down,
},
&cobra.Command{
Use: "drop",
Short: "Drop db",
Long: `Deletes everything in the database`,
RunE: drop,
},
)
}
func up(_ *cobra.Command, _ []string) error {
m, err := initMigrate()
if err != nil {
return fmt.Errorf("open migration failed: %w", err)
}
defer m.Close()
if err := m.Up(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("do migration failed: %w", err)
}
return nil
}
func down(_ *cobra.Command, _ []string) error {
m, err := initMigrate()
if err != nil {
return fmt.Errorf("open migration failed: %w", err)
}
defer m.Close()
if err := m.Down(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("do migration failed: %w", err)
}
return nil
}
func drop(_ *cobra.Command, _ []string) error {
m, err := initMigrate()
if err != nil {
return fmt.Errorf("open migration failed: %w", err)
}
defer m.Close()
if err := m.Drop(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("do migration failed: %w", err)
}
return nil
}
func initMigrate() (*migrate.Migrate, error) {
db, err := sql.Open("sqlite3", dbFile)
if err != nil {
return nil, fmt.Errorf("open db failed: %w", err)
}
defer db.Close()
driver, err := sqlite.WithInstance(db, &sqlite.Config{})
if err != nil {
return nil, fmt.Errorf("failed create migration driver: %w", err)
}
sourceDriver, err := iofs.New(migrations.FS, ".")
if err != nil {
return nil, fmt.Errorf("failed open migrations: %w", err)
}
return migrate.NewWithInstance("fs", sourceDriver, "sqlite3", driver)
}
|
