/
data.go
48 lines (44 loc) · 960 Bytes
/
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
package data
import (
"github.com/go-kratos/kratos/v2/log"
"github.com/google/wire"
"github.com/matteo-gz/prof/internal/conf"
)
var ProviderSet = wire.NewSet(NewData, NewRepo)
type Data struct {
file *file
task *task
log *log.Helper
}
func NewData(c *conf.Data, logger log.Logger) (dataData *Data, cleanup func(), err error) {
dataData = &Data{
log: log.NewHelper(logger),
}
dataData.file, err = newFile(c.StorageDir, logger)
if err != nil {
return
}
dataData.task = newTask(logger)
cleanup = func() {
log.NewHelper(logger).Info("closing the data resources")
dataData.task.close()
}
return
}
func newTask(logger log.Logger) *task {
t := &task{
list: make(map[string]*Proxy),
timerCheck: 60,
log: log.NewHelper(logger),
}
go t.Timer()
return t
}
func newProxy(dir string, l *log.Helper) *Proxy {
return &Proxy{
dir: dir,
fileType: getFileType(dir),
lifeTime: 60 * 30, // 30 min
log: l,
}
}