mirror of
https://github.com/hibiken/asynq.git
synced 2024-12-25 07:12:17 +08:00
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
// Copyright 2020 Kentaro Hibino. All rights reserved.
|
|
// Use of this source code is governed by a MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package asynq
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/hibiken/asynq/internal/rdb"
|
|
)
|
|
|
|
type scheduler struct {
|
|
logger Logger
|
|
rdb *rdb.RDB
|
|
|
|
// channel to communicate back to the long running "scheduler" goroutine.
|
|
done chan struct{}
|
|
|
|
// poll interval on average
|
|
avgInterval time.Duration
|
|
|
|
// list of queues to move the tasks into.
|
|
qnames []string
|
|
}
|
|
|
|
func newScheduler(l Logger, r *rdb.RDB, avgInterval time.Duration, qcfg map[string]int) *scheduler {
|
|
var qnames []string
|
|
for q := range qcfg {
|
|
qnames = append(qnames, q)
|
|
}
|
|
return &scheduler{
|
|
logger: l,
|
|
rdb: r,
|
|
done: make(chan struct{}),
|
|
avgInterval: avgInterval,
|
|
qnames: qnames,
|
|
}
|
|
}
|
|
|
|
func (s *scheduler) terminate() {
|
|
s.logger.Info("Scheduler shutting down...")
|
|
// Signal the scheduler goroutine to stop polling.
|
|
s.done <- struct{}{}
|
|
}
|
|
|
|
// start starts the "scheduler" goroutine.
|
|
func (s *scheduler) start(wg *sync.WaitGroup) {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
for {
|
|
select {
|
|
case <-s.done:
|
|
s.logger.Info("Scheduler done")
|
|
return
|
|
case <-time.After(s.avgInterval):
|
|
s.exec()
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (s *scheduler) exec() {
|
|
if err := s.rdb.CheckAndEnqueue(s.qnames...); err != nil {
|
|
s.logger.Error("Could not enqueue scheduled tasks: %v", err)
|
|
}
|
|
}
|