/
configDB.go
43 lines (37 loc) · 1.05 KB
/
configDB.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
package models
import (
"database/sql"
"fmt"
"log"
// library for conenct postgresql
_ "github.com/lib/pq"
"github.com/spf13/viper"
)
var (
// DB variable for connection DB postgresql
DB *sql.DB
)
// Connect function for checking connection to postgresql
func Connect() error {
host := viper.GetString("configDB.host")
port := viper.GetString("configDB.port")
user := viper.GetString("configDB.user")
password := viper.GetString("configDB.password")
dbname := viper.GetString("configDB.dbname")
// psqlInfo := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname)
psqlInfo := fmt.Sprintf("postgres://%s:%s@%s:%s/%s?sslmode=disable", user, password, host, port, dbname)
// log.Println(psqlInfo)
result, err := sql.Open("postgres", psqlInfo)
if err != nil {
// utils.FailError(err, "Check your config file, Database not connect")
return err
}
// defer result.Close()
err = result.Ping()
if err != nil {
log.Println("Error DB Ping : ", err)
return err
}
DB = result
return nil
}