-
Notifications
You must be signed in to change notification settings - Fork 3
/
migrate.go
105 lines (95 loc) · 2.51 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
package chassix
import (
"errors"
"io/ioutil"
"os"
"github.com/golang-migrate/migrate/v4"
_ "github.com/golang-migrate/migrate/v4/database/mysql" //import mysql driver
"github.com/golang-migrate/migrate/v4/source/go_bindata"
"github.com/go-chassix/chassix/v2/logx"
)
//Migrate Run new bindataInstance and UP
func Migrate(assetNames []string, afn bindata.AssetFunc, dbURL, dialect string) error {
log := logx.New().Service("chassix").Category("migrate")
// wrap assets in Resource
s := bindata.Resource(assetNames, afn)
d, err := bindata.WithInstance(s)
if err != nil {
log.Error(err)
return errors.New("DB migrations build instance error")
}
databaseURL := ""
if dialect == "" || dialect == "mysql" {
databaseURL = "mysql://" + dbURL
}
if dialect == "sqlite3" {
databaseURL = "sqlite3://" + dbURL
}
if dialect == "postgres" {
databaseURL = "postgres://" + dbURL
}
m, err := migrate.NewWithSourceInstance("go-bindata", d, databaseURL)
if err != nil {
log.Error(err)
return errors.New("DB migrations build bindata instance error")
}
//IF ENV NOT PROD IMPORT TEST DATA
if !EnvIsProd() {
if err := m.Down(); err != nil {
log.Error("down: ", err)
}
}
upErr := m.Up() // run migrations
if upErr != nil && upErr != migrate.ErrNoChange {
log.Errorf("Run DB migrations failed,error:%s", upErr.Error())
return errors.New("DB migrations UP error " + upErr.Error())
}
//IF ENV NOT PROD IMPORT TEST DATA
if !EnvIsProd() {
fileName := os.Getenv(EnvPgTestDataFile)
log.Debugf("import data file: %s", fileName)
if fileName != "" {
if file, err := os.Open(fileName); err == nil {
// count := 0
if data, err := ioutil.ReadAll(file); err == nil {
if db, err := DB(); nil != err {
return err
} else {
db.Exec(string(data))
}
}
} else {
log.Error(err)
}
} else {
log.Error("import test data failed")
}
}
return nil
}
//ExecSQLFile danger!!! just for importing test data.
func ExecSQLFile(fileName string) error {
log := logx.New().Service("chassix").Category("execsql")
if fileName != "" {
if file, err := os.Open(fileName); err == nil {
// count := 0
if data, err := ioutil.ReadAll(file); err == nil {
if db, err := DB(); nil != err {
return err
} else {
err = db.Exec(string(data)).Error
if err != nil {
return err
}
}
}
} else {
log.Error(err)
return err
}
} else {
log.Error("import test data failed")
return errors.New("import test data failed, file cannot be empty")
}
return nil
}