-
Notifications
You must be signed in to change notification settings - Fork 12
/
database.go
121 lines (104 loc) · 2.77 KB
/
database.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
package database
import (
"fmt"
"sync"
"code.cloudfoundry.org/lager"
"github.com/huaweicloud/huaweicloud-service-broker/pkg/config"
"github.com/jinzhu/gorm"
// import mysql driver
_ "github.com/jinzhu/gorm/dialects/mysql"
)
// BackDBConnection is used to operate on database
var BackDBConnection *gorm.DB
// once is used to init BackDBConnection
var once sync.Once
// New Database Connection
func New(logger lager.Logger, config config.Config) error {
once.Do(func() {
// Connect Data
backDBConnection, err := ConnectBackDatabase(logger, config)
if err != nil {
logger.Error("Error connecting to back database", err)
panic(err)
}
err = UpgradeBackDatabase(logger, backDBConnection)
if err != nil {
logger.Error("Error upgrading back database", err)
panic(err)
}
BackDBConnection = backDBConnection
})
return nil
}
// ConnectBackDatabase connect back database
func ConnectBackDatabase(logger lager.Logger, config config.Config) (*gorm.DB, error) {
// connect to back database
connStr := fmt.Sprintf(
"%v:%v@tcp(%v:%v)/%v?charset=utf8&parseTime=True&loc=Local",
config.BackDatabase.DatabaseUsername,
config.BackDatabase.DatabasePassword,
config.BackDatabase.DatabaseHost,
config.BackDatabase.DatabasePort,
config.BackDatabase.DatabaseName)
return gorm.Open(config.BackDatabase.DatabaseType, connStr)
}
// UpgradeBackDatabase upgrades tables and datas
func UpgradeBackDatabase(logger lager.Logger, backdatabase *gorm.DB) error {
// upgrades defines
upgrades := make([]func() error, 1)
upgrades[0] = func() error {
// table upgrades
if err := backdatabase.Exec(UpgradesTableSQL).Error; err != nil {
return err
}
// table instance_details
if err := backdatabase.Exec(InstanceDetailsTableSQL).Error; err != nil {
return err
}
// table bind_details
if err := backdatabase.Exec(BindDetailsTableSQL).Error; err != nil {
return err
}
// table operation_details
if err := backdatabase.Exec(OperationDetailsTableSQL).Error; err != nil {
return err
}
return nil
}
// Get UpgradeID
var lastUpgrade = -1
if backdatabase.HasTable(UpgradesTableName) {
var ups []Upgrades
err := backdatabase.Order("upgrade_id desc").Find(&ups).Error
if err != nil {
logger.Error("Error getting upgrades", err)
return err
}
// Get value
if len(ups) > 0 {
lastUpgrade = ups[0].UpgradeID
}
}
// Exec upgrades one by one
for index := lastUpgrade + 1; index < len(upgrades); index++ {
// Begin transaction
tx := backdatabase.Begin()
// Exec upgrades
err := upgrades[index]()
if err != nil {
tx.Rollback()
return err
}
// Save last upgrade
err = backdatabase.Save(&Upgrades{
UpgradeID: index,
}).Error
if err != nil {
tx.Rollback()
return err
}
// End transaction
tx.Commit()
}
return nil
}