/
db.go
102 lines (81 loc) · 2.46 KB
/
db.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
package db
import (
"database/sql"
"fmt"
"log"
"os"
"github.com/go-gorp/gorp"
_redis "github.com/go-redis/redis/v7"
// _ "github.com/lib/pq" //import postgres
_ "github.com/go-sql-driver/mysql"
)
//DB ...
type DB struct {
*sql.DB
}
var db *gorp.DbMap
//Init ...
func Init() {
// dbinfo := fmt.Sprintf("user=%s password=%s dbname=%s sslmode=disable", os.Getenv("DB_USER"), os.Getenv("DB_PASS"), os.Getenv("DB_NAME"))
dbinfo := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", os.Getenv("DB_USER"), os.Getenv("DB_PASS"), os.Getenv("DB_HOST"), os.Getenv("DB_PORT"), os.Getenv("DB_NAME"))
var err error
db, err = ConnectDB(dbinfo)
if err != nil {
log.Fatal(err)
}
}
//ConnectDB ...
func ConnectDB(dataSourceName string) (*gorp.DbMap, error) {
db, err := sql.Open("mysql", dataSourceName)
// db, err := sql.Open("postgres", dataSourceName)
checkErr(err, "db not connect")
if err = db.Ping(); err != nil {
return nil, err
}
// dbmap := &gorp.DbMap{Db: db, Dialect: gorp.PostgresDialect{}}
dbmap := &gorp.DbMap{Db: db, Dialect: gorp.MySQLDialect{"InnoDB", "UTF8"}}
dbmap.TraceOn("[gorp]", log.New(os.Stdout, "golang-gin:", log.Lmicroseconds)) //Trace database requests
// dbmap := &gorp.DbMap{Db: db, Dialect: gorp.PostgresDialect{}}
//dbmap.TraceOn("[gorp]", log.New(os.Stdout, "golang-gin:", log.Lmicroseconds)) //Trace database requests
// Defines the table for use by GORP
// This is a function we will create soon.
// dbmap.AddTableWithName(Pengaduan{}, "article").SetKeys(false, "ID")
// err = dbmap.CreateTablesIfNotExists()
// checkErr(err, "Create tables failed")
return dbmap, nil
}
func checkErr(err error, msg string) {
if err != nil {
log.Fatalln(msg, err)
}
}
//GetDB ...
func GetDB() *gorp.DbMap {
return db
}
//RedisClient ...
var RedisClient *_redis.Client
//InitRedis ...
func InitRedis(selectDB ...int) {
var redisHost = os.Getenv("REDIS_HOST")
var redisPassword = os.Getenv("REDIS_PASSWORD")
RedisClient = _redis.NewClient(&_redis.Options{
Addr: redisHost,
Password: redisPassword,
DB: selectDB[0],
// DialTimeout: 10 * time.Second,
// ReadTimeout: 30 * time.Second,
// WriteTimeout: 30 * time.Second,
// PoolSize: 10,
// PoolTimeout: 30 * time.Second,
// IdleTimeout: 500 * time.Millisecond,
// IdleCheckFrequency: 500 * time.Millisecond,
// TLSConfig: &tls.Config{
// InsecureSkipVerify: true,
// },
})
}
//GetRedis ...
func GetRedis() *_redis.Client {
return RedisClient
}