-
Notifications
You must be signed in to change notification settings - Fork 1
/
migration.go
139 lines (123 loc) · 3.58 KB
/
migration.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
/*
* Copyright 2022 Armory, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package mysql
import (
"context"
"fmt"
"github.com/go-sql-driver/mysql"
"github.com/golang-migrate/migrate/v4"
_ "github.com/golang-migrate/migrate/v4/database/mysql"
_ "github.com/golang-migrate/migrate/v4/source/file"
"go.uber.org/fx"
"go.uber.org/zap"
"time"
)
const defaultMigrationPath = "./db/migrations"
type (
Migrator struct {
settings Configuration
log *zap.SugaredLogger
}
Configuration struct {
Connection string `yaml:"connection"`
// User can also be specified separately from the connection string
User string `yaml:"user"`
// Password can also be specified separately from the connection string
Password string `yaml:"password"`
// User can also be specified separately from the connection string
MigrateUser string `yaml:"migrateUser"`
// Password can also be specified separately from the connection string
MigratePassword string `yaml:"migratePassword"`
// MaxLifetime is the maximum lifetime of a connection
// from time.ParseDuration:
// A duration string is a possibly signed sequence of
// decimal numbers, each with optional fraction and a unit suffix,
// such as "300ms", "-1.5h" or "2h45m".
// Valid time units are "ns", "us" (or "µs"), "ms", "s", "m", "h".
MaxLifetime MDuration `yaml:"maxLifetime"`
MaxOpenConnections int `yaml:"maxOpenConnections"`
MaxIdleConnections int `yaml:"maxIdleConnections"`
MigrationPath string `yaml:"migrationPath"`
}
MDuration struct {
time.Duration
}
)
func (d *Configuration) ConnectionUrl(migration bool) (string, error) {
cfg, err := mysql.ParseDSN(d.Connection)
if err != nil {
return "", err
}
if migration {
cfg.User = d.MigrateUser
cfg.Passwd = d.MigratePassword
} else {
cfg.User = d.User
cfg.Passwd = d.Password
}
if migration {
return fmt.Sprintf("mysql://%s", cfg.FormatDSN()), nil
}
cfg.ParseTime = true
return cfg.FormatDSN(), nil
}
func (d *MDuration) UnmarshalJSON(data []byte) error {
s := string(data)
if len(s) > 2 && s[0] == '"' && s[len(s)-1] == '"' {
s = s[1 : len(s)-1]
}
// remove quotes
var err error
d.Duration, err = time.ParseDuration(s)
if err != nil {
return err
}
return nil
}
func NewMigrator(lc fx.Lifecycle, settings Configuration, log *zap.SugaredLogger) *Migrator {
m := &Migrator{
settings: settings,
log: log,
}
lc.Append(fx.Hook{
OnStart: func(context.Context) error {
return m.migrate()
},
})
return m
}
func (m *Migrator) migrate() error {
databaseConfig := m.settings
c, err := databaseConfig.ConnectionUrl(true)
if err != nil {
return err
}
migrationPath := databaseConfig.MigrationPath
if migrationPath == "" {
m.log.Infof("No database.migrationPath configured, defaulting to: %s", defaultMigrationPath)
migrationPath = defaultMigrationPath
}
migrationInstance, err := migrate.New(fmt.Sprintf("file://%s", migrationPath), c)
if err != nil {
return err
}
err = migrationInstance.Up()
if err == migrate.ErrNoChange {
m.log.Infof("No change detected.")
return nil
}
return err
}