This repository has been archived by the owner on Oct 9, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 62
/
migrate.go
142 lines (127 loc) · 4.52 KB
/
migrate.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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package entrypoints
import (
"context"
"github.com/flyteorg/flyteadmin/pkg/runtime"
"github.com/flyteorg/flytestdlib/promutils"
"github.com/flyteorg/flytestdlib/logger"
"github.com/flyteorg/flyteadmin/pkg/repositories/config"
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/postgres" // Required to import database driver.
"github.com/spf13/cobra"
gormigrate "gopkg.in/gormigrate.v1"
)
var parentMigrateCmd = &cobra.Command{
Use: "migrate",
Short: "This command controls migration behavior for the Flyte admin database. Please choose a subcommand.",
}
var migrationsScope = promutils.NewScope("migrations")
var migrateScope = migrationsScope.NewSubScope("migrate")
var rollbackScope = promutils.NewScope("migrations").NewSubScope("rollback")
// This runs all the migrations
var migrateCmd = &cobra.Command{
Use: "run",
Short: "This command will run all the migrations for the database",
Run: func(cmd *cobra.Command, args []string) {
ctx := context.Background()
configuration := runtime.NewConfigurationProvider()
databaseConfig := configuration.ApplicationConfiguration().GetDbConfig()
postgresConfigProvider := config.NewPostgresConfigProvider(config.DbConfig{
BaseConfig: config.BaseConfig{
IsDebug: databaseConfig.Debug,
},
Host: databaseConfig.Host,
Port: databaseConfig.Port,
DbName: databaseConfig.DbName,
User: databaseConfig.User,
Password: databaseConfig.Password,
ExtraOptions: databaseConfig.ExtraOptions,
}, migrateScope)
db, err := gorm.Open(postgresConfigProvider.GetType(), postgresConfigProvider.GetArgs())
if err != nil {
logger.Fatal(ctx, err)
}
defer db.Close()
db.LogMode(true)
if err = db.DB().Ping(); err != nil {
logger.Fatal(ctx, err)
}
m := gormigrate.New(db, gormigrate.DefaultOptions, config.Migrations)
if err = m.Migrate(); err != nil {
logger.Fatalf(ctx, "Could not migrate: %v", err)
}
logger.Infof(ctx, "Migration ran successfully")
},
}
// Rollback the latest migration
var rollbackCmd = &cobra.Command{
Use: "rollback",
Short: "This command will rollback one migration",
Run: func(cmd *cobra.Command, args []string) {
ctx := context.Background()
configuration := runtime.NewConfigurationProvider()
databaseConfig := configuration.ApplicationConfiguration().GetDbConfig()
postgresConfigProvider := config.NewPostgresConfigProvider(config.DbConfig{
BaseConfig: config.BaseConfig{
IsDebug: databaseConfig.Debug,
},
Host: databaseConfig.Host,
Port: databaseConfig.Port,
DbName: databaseConfig.DbName,
User: databaseConfig.User,
Password: databaseConfig.Password,
ExtraOptions: databaseConfig.ExtraOptions,
}, rollbackScope)
db, err := gorm.Open(postgresConfigProvider.GetType(), postgresConfigProvider.GetArgs())
if err != nil {
logger.Fatal(ctx, err)
}
defer db.Close()
db.LogMode(true)
if err = db.DB().Ping(); err != nil {
logger.Fatal(ctx, err)
}
m := gormigrate.New(db, gormigrate.DefaultOptions, config.Migrations)
err = m.RollbackLast()
if err != nil {
logger.Fatalf(ctx, "Could not rollback latest migration: %v", err)
}
logger.Infof(ctx, "Rolled back one migration successfully")
},
}
// This seeds the database with project values
var seedProjectsCmd = &cobra.Command{
Use: "seed-projects",
Short: "Seed projects in the database.",
Run: func(cmd *cobra.Command, args []string) {
ctx := context.Background()
configuration := runtime.NewConfigurationProvider()
databaseConfig := configuration.ApplicationConfiguration().GetDbConfig()
postgresConfigProvider := config.NewPostgresConfigProvider(config.DbConfig{
BaseConfig: config.BaseConfig{
IsDebug: databaseConfig.Debug,
},
Host: databaseConfig.Host,
Port: databaseConfig.Port,
DbName: databaseConfig.DbName,
User: databaseConfig.User,
Password: databaseConfig.Password,
ExtraOptions: databaseConfig.ExtraOptions,
}, migrateScope)
db, err := gorm.Open(postgresConfigProvider.GetType(), postgresConfigProvider.GetArgs())
if err != nil {
logger.Fatal(ctx, err)
}
defer db.Close()
db.LogMode(true)
if err = config.SeedProjects(db, args); err != nil {
logger.Fatalf(ctx, "Could not add projects to database with err: %v", err)
}
logger.Infof(ctx, "Successfully added projects to database")
},
}
func init() {
RootCmd.AddCommand(parentMigrateCmd)
parentMigrateCmd.AddCommand(migrateCmd)
parentMigrateCmd.AddCommand(rollbackCmd)
parentMigrateCmd.AddCommand(seedProjectsCmd)
}