-
Notifications
You must be signed in to change notification settings - Fork 1
/
migration.go
41 lines (34 loc) · 978 Bytes
/
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
package postgres
import (
"github.com/golang-migrate/migrate/v4"
_ "github.com/golang-migrate/migrate/v4/database/postgres"
_ "github.com/golang-migrate/migrate/v4/source/file"
"github.com/sirupsen/logrus"
)
const MigrationsUrl = "file://migrations"
func Migrate(postgresConn string, logger *logrus.Logger) error {
m, err := migrate.New(MigrationsUrl, postgresConn)
if err != nil {
return err
}
m.Log = &logrusMigrateAdapter{logger}
err = m.Up()
if err != nil {
if err == migrate.ErrNoChange {
logger.Info("No new postgres migrations found")
} else {
return err
}
}
return nil
}
type logrusMigrateAdapter struct {
logger *logrus.Logger
}
func (a *logrusMigrateAdapter) Printf(format string, v ...interface{}) {
// TODO: No logrus categories? Maybe switch loggers...
a.logger.WithField("category", "postgresmigration").Printf(format, v...)
}
func (a *logrusMigrateAdapter) Verbose() bool {
return a.logger.IsLevelEnabled(logrus.DebugLevel)
}