/
db.go
49 lines (42 loc) · 975 Bytes
/
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
package main
import (
redis "github.com/alphazero/Go-Redis"
mongo "labix.org/v2/mgo"
"log"
)
const (
MONGO_CLIENT_URL = "localhost"
)
var (
/// redis
redisSpec = redis.DefaultSpec()
redisConn redis.Client
redisError redis.Error
/// mongo
mongoClient *mongo.Session
mongoError error
)
func setupRedisConnection() {
log.Printf("connecting to redis...")
redisConn, redisError = redis.NewSynchClientWithSpec(redisSpec)
if redisError != nil {
log.Fatal("couldn't connect to redis: ", redisError)
}
log.Printf("connected to redis.")
}
func setupMongoConnection() {
log.Printf("connecting to mongo...")
mongoClient, mongoError = mongo.Dial(MONGO_CLIENT_URL)
if mongoError != nil {
log.Fatal("couldn't connect to mongo: ", mongoError)
}
log.Printf("connected to mongo.")
}
func setupDBConnections() {
go setupRedisConnection()
go setupMongoConnection()
}
func closeDBConnections() {
mongoClient.Close()
log.Printf("closed mongo connection.")
}