mirror of
https://github.com/hibiken/asynq.git
synced 2024-12-25 15:22:18 +08:00
58 lines
1.1 KiB
Go
58 lines
1.1 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 (
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/hibiken/asynq/internal/rdb"
|
|
)
|
|
|
|
type scheduler struct {
|
|
rdb *rdb.RDB
|
|
|
|
// channel to communicate back to the long running "scheduler" goroutine.
|
|
done chan struct{}
|
|
|
|
// poll interval on average
|
|
avgInterval time.Duration
|
|
}
|
|
|
|
func newScheduler(r *rdb.RDB, avgInterval time.Duration) *scheduler {
|
|
return &scheduler{
|
|
rdb: r,
|
|
done: make(chan struct{}),
|
|
avgInterval: avgInterval,
|
|
}
|
|
}
|
|
|
|
func (s *scheduler) terminate() {
|
|
log.Println("[INFO] Scheduler shutting down...")
|
|
// Signal the scheduler goroutine to stop polling.
|
|
s.done <- struct{}{}
|
|
}
|
|
|
|
// start starts the "scheduler" goroutine.
|
|
func (s *scheduler) start() {
|
|
go func() {
|
|
for {
|
|
select {
|
|
case <-s.done:
|
|
log.Println("[INFO] Scheduler done.")
|
|
return
|
|
case <-time.After(s.avgInterval):
|
|
s.exec()
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (s *scheduler) exec() {
|
|
if err := s.rdb.CheckAndEnqueue(); err != nil {
|
|
log.Printf("[ERROR] could not forward scheduled tasks: %v\n", err)
|
|
}
|
|
}
|