-
Notifications
You must be signed in to change notification settings - Fork 1
/
data.go
110 lines (95 loc) · 2.76 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
109
110
package data
import (
"context"
"github.com/go-kratos/kratos/v2/registry"
"github.com/go-redis/redis"
consul "github.com/go-kratos/consul/registry"
"github.com/go-kratos/kratos/v2/log"
_ "github.com/go-sql-driver/mysql"
"github.com/google/wire"
consulAPI "github.com/hashicorp/consul/api"
"github.com/lalifeier/vvgo-mall/app/auth/service/internal/conf"
"github.com/lalifeier/vvgo-mall/app/auth/service/internal/data/ent"
"github.com/lalifeier/vvgo-mall/app/auth/service/internal/data/ent/migrate"
_ "github.com/lalifeier/vvgo-mall/app/auth/service/internal/data/ent/runtime"
)
// ProviderSet is data providers.
var ProviderSet = wire.NewSet(NewData, NewEntClient, NewRedisClient, NewDiscovery, NewRegistrar)
// Data .
type Data struct {
log *log.Helper
db *ent.Client
rdb *redis.Client
}
// NewData .
func NewData(logger log.Logger, entClient *ent.Client, redisClient *redis.Client) (*Data, func(), error) {
log := log.NewHelper(log.With(logger, "module", "auth-service/data"))
d := &Data{
log: log,
db: entClient,
rdb: redisClient,
}
cleanup := func() {
if err := d.db.Close(); err != nil {
log.Error(err)
}
}
return d, cleanup, nil
}
func NewDiscovery(conf *conf.Registry) registry.Discovery {
c := consulAPI.DefaultConfig()
c.Address = conf.Consul.Address
c.Scheme = conf.Consul.Scheme
cli, err := consulAPI.NewClient(c)
if err != nil {
panic(err)
}
r := consul.New(cli, consul.WithHealthCheck(false))
return r
}
func NewRegistrar(conf *conf.Registry) registry.Registrar {
c := consulAPI.DefaultConfig()
c.Address = conf.Consul.Address
c.Scheme = conf.Consul.Scheme
cli, err := consulAPI.NewClient(c)
if err != nil {
panic(err)
}
r := consul.New(cli, consul.WithHealthCheck(false))
return r
}
func NewEntClient(conf *conf.Data, logger log.Logger) *ent.Client {
log := log.NewHelper(log.With(logger, "module", "auth-service/data/ent"))
client, err := ent.Open(
conf.Database.Driver,
conf.Database.Source,
ent.Debug(),
)
if err != nil {
log.Fatalf("failed opening connection to db: %v", err)
}
// Run the auto migration tool.
if err := client.Debug().Schema.Create(
context.Background(),
migrate.WithDropIndex(true),
migrate.WithDropColumn(true),
); err != nil {
log.Fatalf("failed creating schema resources: %v", err)
}
return client
}
func NewRedisClient(c *conf.Data, logger log.Logger) *redis.Client {
client := redis.NewClient(&redis.Options{
Addr: c.Redis.Network,
Password: c.Redis.Password,
DB: int(c.Redis.Db),
DialTimeout: c.Redis.DialTimeout.AsDuration(),
ReadTimeout: c.Redis.ReadTimeout.AsDuration(),
WriteTimeout: c.Redis.WriteTimeout.AsDuration(),
})
// err := client.Ping().Err()
// if err != nil {
// log.Fatalf("redis connect error: %v", err)
// }
return client
}