-
Notifications
You must be signed in to change notification settings - Fork 0
/
cron.go
41 lines (34 loc) · 1.11 KB
/
cron.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 jobs
import (
"github.com/dstgo/tracker/conf"
"github.com/dstgo/tracker/internal/handler"
"github.com/dstgo/tracker/internal/types"
"github.com/robfig/cron/v3"
"log/slog"
)
// cronLogger adapter
type cronLogger struct {
logger *slog.Logger
prefab string
}
func (c cronLogger) Info(msg string, keysAndValues ...interface{}) {
c.logger.Info(c.prefab+": "+msg, keysAndValues...)
}
func (c cronLogger) Error(err error, msg string, keysAndValues ...interface{}) {
c.logger.Error(c.prefab+": "+msg, append([]any{"err", err}, keysAndValues...)...)
}
func LoadCronJobs(dstConf conf.DstConf, lobbyHandler handler.LobbyHandler) (*cron.Cron, error) {
cronJob := cron.New(
cron.WithLogger(cronLogger{logger: slog.Default(), prefab: "CRON"}),
cron.WithLocation(types.TimeZone),
)
// lobby collector
lobbyCollector := LobbyCollector{dstConf.Lobby, lobbyHandler, cronJob}
if _, err := cronJob.AddFunc(dstConf.Lobby.CollectCron, lobbyCollector.Collect); err != nil {
return nil, err
}
if _, err := cronJob.AddFunc(dstConf.Lobby.ClearCron, lobbyCollector.Clear); err != nil {
return nil, err
}
return cronJob, nil
}