-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
41 lines (34 loc) 路 1.01 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
package lib
import (
"fmt"
"github.com/go-redis/redis/v8"
"gorm.io/driver/postgres"
"gorm.io/gorm"
)
type Database struct {
Postgres *gorm.DB
Redis *redis.Client
}
func NewDatabase(env Env, logger Logger) Database {
dsn := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable TimeZone=Asia/Shanghai", env.PostgresHost, env.PostgresPort, env.PostgresUserName, env.PostgresPassword, env.PostgresDB)
postgres, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
Logger: logger.GetGormLogger(),
})
if err != nil {
logger.Info("dsn:", dsn)
logger.Panic(err)
}
logger.Info("PostgreSQL database connection established")
redisURL := fmt.Sprintf("redis://:%s@%s:%s/%s", env.RedisPassword, env.RedisHost, env.RedisPort, env.RedisName)
opt, err := redis.ParseURL(redisURL)
if err != nil {
logger.Info("redisURL:", redisURL)
logger.Panic(err)
}
redis := redis.NewClient(opt)
logger.Info("Redis database connection established")
return Database{
Postgres: postgres,
Redis: redis,
}
}