go-pkg/asynq
2
0
Fork 0
mirror of https://github.com/hibiken/asynq.git synced 2024-12-25 23:32:17 +08:00
Code Issues Projects Releases Wiki Activity

暴漏出来M,可以知道哪些数据是注册成功，方便维护更新

Browse Source
This commit is contained in:
xiaohan 2024-12-10 22:48:49 +08:00
parent 1e87f20b06
commit 51b4451935
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
periodic_task_manager.go
View File

@ -22,7 +22,7 @@ type PeriodicTaskManager struct {
syncInterval time.Duration syncInterval time.Duration
done chan (struct{}) done chan (struct{})
wg sync.WaitGroup wg sync.WaitGroup
m map[string]struct{} // map[hash]entryID M map[string]struct{} // map[hash]entryID
} }
type PeriodicTaskManagerOpts struct { type PeriodicTaskManagerOpts struct {
@ -69,7 +69,7 @@ func NewPeriodicTaskManager(opts PeriodicTaskManagerOpts) (*PeriodicTaskManager,
p: opts.PeriodicTaskConfigProvider, p: opts.PeriodicTaskConfigProvider,
syncInterval: syncInterval, syncInterval: syncInterval,
done: make(chan struct{}), done: make(chan struct{}),
m: make(map[string]struct{}), M: make(map[string]struct{}),
}, nil }, nil
} }
@ -188,7 +188,7 @@ func (mgr *PeriodicTaskManager) add(configs []*PeriodicTaskConfig) {
c.Cronspec, c.Task.Type(), err) c.Cronspec, c.Task.Type(), err)
continue continue
} }
mgr.m[entryID] = struct{}{} // ? m[string]struct{} mgr.M[entryID] = struct{}{} // m[string]struct{}
mgr.s.logger.Infof("Successfully registered periodic task: cronspec=%q task=%q, entryID=%s", mgr.s.logger.Infof("Successfully registered periodic task: cronspec=%q task=%q, entryID=%s",
c.Cronspec, c.Task.Type(), entryID) c.Cronspec, c.Task.Type(), entryID)
} }
@ -200,7 +200,7 @@ func (mgr *PeriodicTaskManager) remove(removed map[string]struct{}) {
mgr.s.logger.Errorf("Failed to unregister periodic task: %v", err) mgr.s.logger.Errorf("Failed to unregister periodic task: %v", err)
continue continue
} }
delete(mgr.m, entryID) delete(mgr.M, entryID)
mgr.s.logger.Infof("Successfully unregistered periodic task: entryID=%s", entryID) mgr.s.logger.Infof("Successfully unregistered periodic task: entryID=%s", entryID)
} }
} }
@ -232,7 +232,7 @@ func (mgr *PeriodicTaskManager) diffRemoved(configs []*PeriodicTaskConfig) map[s
newConfigs[c.ID] = struct{}{} // empty value since we don't have entryID yet newConfigs[c.ID] = struct{}{} // empty value since we don't have entryID yet
} }
removed := make(map[string]struct{}) removed := make(map[string]struct{})
for k, _ := range mgr.m { for k, _ := range mgr.M {
// test whether existing config is present in the incoming configs // test whether existing config is present in the incoming configs
if _, found := newConfigs[k]; !found { if _, found := newConfigs[k]; !found {
removed[k] = struct{}{} removed[k] = struct{}{}
@ -246,7 +246,7 @@ func (mgr *PeriodicTaskManager) diffRemoved(configs []*PeriodicTaskConfig) map[s
func (mgr *PeriodicTaskManager) diffAdded(configs []*PeriodicTaskConfig) []*PeriodicTaskConfig { func (mgr *PeriodicTaskManager) diffAdded(configs []*PeriodicTaskConfig) []*PeriodicTaskConfig {
var added []*PeriodicTaskConfig var added []*PeriodicTaskConfig
for _, c := range configs { for _, c := range configs {
if _, found := mgr.m[c.ID]; !found { if _, found := mgr.M[c.ID]; !found {
added = append(added, c) added = append(added, c)
} }
} }