2
0
mirror of https://github.com/hibiken/asynq.git synced 2025-10-21 21:46:12 +08:00

chore(): 从内存里面读取任务

This commit is contained in:
pacinochen
2022-02-24 20:13:34 +08:00
parent e55a693fa0
commit 2f7d7a88dd
2 changed files with 44 additions and 47 deletions

View File

@@ -15,14 +15,12 @@ import (
"github.com/hibiken/asynq/internal/base"
"github.com/hibiken/asynq/internal/errors"
"github.com/hibiken/asynq/internal/rdb"
"github.com/robfig/cron/v3"
)
// Inspector is a client interface to inspect and mutate the state of
// queues and tasks.
type Inspector struct {
rdb *rdb.RDB
cron *cron.Cron
rdb *rdb.RDB
}
// New returns a new instance of Inspector.
@@ -32,8 +30,7 @@ func NewInspector(r RedisConnOpt) *Inspector {
panic(fmt.Sprintf("inspeq: unsupported RedisConnOpt type %T", r))
}
return &Inspector{
rdb: rdb.NewRDB(c),
cron: cron.New(cron.WithLocation(time.Local)),
rdb: rdb.NewRDB(c),
}
}
@@ -799,55 +796,30 @@ type SchedulerEntry struct {
// currently running schedulers.
func (i *Inspector) SchedulerEntries() ([]*SchedulerEntry, error) {
var entries []*SchedulerEntry
fmt.Printf("start to record log file")
res, err := i.rdb.ListSchedulerEntries()
if err != nil {
fmt.Errorf("ListSchedulerEntries err:%s", err.Error())
return nil, err
}
// 如果查询不到数据 从内存里面获取数据
if len(res) == 0 {
fmt.Println("search entries from mem")
for _, entry := range i.cron.Entries() {
job := entry.Job.(*enqueueJob)
var opts []Option
opt := stringifyOptions(job.opts)
for _, s := range opt {
if o, err := parseOption(s); err == nil {
// ignore bad data
opts = append(opts, o)
}
for _, e := range res {
task := NewTask(e.Type, e.Payload)
var opts []Option
for _, s := range e.Opts {
if o, err := parseOption(s); err == nil {
// ignore bad data
opts = append(opts, o)
}
entries = append(entries, &SchedulerEntry{
ID: job.id.String(),
Spec: job.cronspec,
Task: job.task,
Opts: opts,
Next: entry.Next,
Prev: entry.Prev,
})
}
fmt.Printf("entries lens is %d \n", len(res))
} else {
for _, e := range res {
task := NewTask(e.Type, e.Payload)
var opts []Option
for _, s := range e.Opts {
if o, err := parseOption(s); err == nil {
// ignore bad data
opts = append(opts, o)
}
}
entries = append(entries, &SchedulerEntry{
ID: e.ID,
Spec: e.Spec,
Task: task,
Opts: opts,
Next: e.Next,
Prev: e.Prev,
})
}
entries = append(entries, &SchedulerEntry{
ID: e.ID,
Spec: e.Spec,
Task: task,
Opts: opts,
Next: e.Next,
Prev: e.Prev,
})
}
return entries, nil
}

View File

@@ -297,3 +297,28 @@ func (s *Scheduler) clearHistory() {
}
}
}
func (s *Scheduler) GetEntries() ([]*SchedulerEntry, error) {
var entries []*SchedulerEntry
for _, entry := range s.cron.Entries() {
job := entry.Job.(*enqueueJob)
var opts []Option
opt := stringifyOptions(job.opts)
for _, s := range opt {
if o, err := parseOption(s); err == nil {
// ignore bad data
opts = append(opts, o)
}
}
entries = append(entries, &SchedulerEntry{
ID: job.id.String(),
Spec: job.cronspec,
Task: job.task,
Opts: opts,
Next: entry.Next,
Prev: entry.Prev,
})
}
return entries, nil
}