/
resource.go
46 lines (39 loc) 路 946 Bytes
/
resource.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
package resources
import (
"github.com/NghiaTranUIT/artify-core/constant"
"github.com/NghiaTranUIT/artify-core/utils"
"github.com/gobuffalo/pop"
)
type Resource struct {
Config constant.Config
PostgreSQL *pop.Connection
}
func (r *Resource) Close() {
utils.LogInfo("Closing all connection ...")
if r.Config.IsEnablePostgreSQL {
r.PostgreSQL.Close()
}
}
func Init(config constant.Config) (*Resource, error) {
r := Resource{}
r.Config = config
// PostgreSQL
if config.IsEnablePostgreSQL {
utils.LogInfo("Initializing PostgreSQL connection")
pg, err := initPostgreSQL()
if err != nil {
utils.LogError("Failed PostgreSQL connection ...", err)
return nil, err
}
utils.LogInfo("Connected PostgreSQL successfully !!!")
r.PostgreSQL = pg
}
return &r, nil
}
func initPostgreSQL() (*pop.Connection, error) {
dbSQL, err := pop.Connect("development")
if err != nil {
return dbSQL, err
}
return dbSQL, err
}