/
data.go
108 lines (96 loc) · 2.65 KB
/
data.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
103
104
105
106
107
108
package data
import (
"context"
"time"
"github.com/go-kratos/kratos/v2/log"
"github.com/go-redis/redis"
_ "github.com/go-sql-driver/mysql"
"github.com/google/wire"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"github.com/ZQCard/kbk-administrator/internal/conf"
"github.com/ZQCard/kbk-administrator/pkg/middleware/requestInfo"
)
// ProviderSet is data providers.
var ProviderSet = wire.NewSet(
NewData,
NewRedisCmd,
NewMysqlCmd,
NewRedisClient,
NewAdministratorRepo)
// Data .
type Data struct {
cfg *conf.Bootstrap
logger *log.Helper
db *gorm.DB
rdb *redis.Client
}
func NewData(cfg *conf.Bootstrap, db *gorm.DB, redisCli *redis.Client, logger log.Logger) (*Data, func(), error) {
logs := log.NewHelper(log.With(logger, "module", "kratos-base-layout/data"))
cleanup := func() {
logs.Info("closing the data resources")
}
return &Data{
logger: logs,
cfg: cfg,
db: db,
rdb: redisCli,
}, cleanup, nil
}
func getDomain(ctx context.Context) string {
domain := ctx.Value(requestInfo.DomainKey)
return domain.(string)
}
func getDbWithDomain(ctx context.Context, db *gorm.DB) *gorm.DB {
domain := ctx.Value(requestInfo.DomainKey)
if domain != nil {
db = db.Where("domain = ?", domain)
}
return db
}
func NewRedisClient(conf *conf.Data) *redis.Client {
client := redis.NewClient(&redis.Options{
Addr: conf.Redis.Addr,
Password: conf.Redis.Password,
ReadTimeout: conf.Redis.ReadTimeout.AsDuration(),
WriteTimeout: conf.Redis.WriteTimeout.AsDuration(),
DialTimeout: time.Second * 2,
PoolSize: 10,
DB: int(conf.Redis.Db),
})
err := client.Ping().Err()
if err != nil {
log.Fatalf("redis connect error: %v", err)
}
return client
}
func NewRedisCmd(conf *conf.Data, logger log.Logger) redis.Cmdable {
logs := log.NewHelper(log.With(logger, "module", "serviceName/data/redis"))
client := redis.NewClient(&redis.Options{
Addr: conf.Redis.Addr,
Password: conf.Redis.Password,
ReadTimeout: conf.Redis.ReadTimeout.AsDuration(),
WriteTimeout: conf.Redis.WriteTimeout.AsDuration(),
DialTimeout: time.Second * 2,
PoolSize: 10,
})
err := client.Ping().Err()
if err != nil {
logs.Fatalf("redis connect error: %v", err)
}
return client
}
func NewMysqlCmd(conf *conf.Bootstrap, logger log.Logger) *gorm.DB {
logs := log.NewHelper(log.With(logger, "module", "serviceName/data/mysql"))
db, err := gorm.Open(mysql.Open(conf.Data.Database.Source), &gorm.Config{})
if err != nil {
logs.Fatalf("mysql connect error: %v", err)
}
// 如果是开发环境 打印sql
if conf.Env.Mode == "dev" {
db = db.Debug()
}
// 数据迁移
db.AutoMigrate(&AdministratorEntity{})
return db
}