148 lines
3 KiB
Go
148 lines
3 KiB
Go
package main
|
|
|
|
import "time"
|
|
|
|
type Job interface {
|
|
Schedule()
|
|
Stop()
|
|
Run()
|
|
}
|
|
|
|
type jobIdentifier int
|
|
|
|
const (
|
|
JobIdCloseDecisions jobIdentifier = iota
|
|
JobIdRemindVotersJob
|
|
)
|
|
|
|
var rescheduleChannel = make(chan jobIdentifier, 1)
|
|
|
|
func JobScheduler(quitChannel chan int) {
|
|
var jobs = map[jobIdentifier]Job{
|
|
JobIdCloseDecisions: NewCloseDecisionsJob(),
|
|
JobIdRemindVotersJob: NewRemindVotersJob(),
|
|
}
|
|
logger.Infof("started job scheduler")
|
|
|
|
for {
|
|
select {
|
|
case jobId := <-rescheduleChannel:
|
|
job := jobs[jobId]
|
|
logger.Infof("reschedule job %s", job)
|
|
job.Schedule()
|
|
case <-quitChannel:
|
|
for _, job := range jobs {
|
|
job.Stop()
|
|
}
|
|
logger.Infof("stop job scheduler")
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
type CloseDecisionsJob struct {
|
|
timer *time.Timer
|
|
}
|
|
|
|
func NewCloseDecisionsJob() *CloseDecisionsJob {
|
|
job := &CloseDecisionsJob{}
|
|
job.Schedule()
|
|
return job
|
|
}
|
|
|
|
func (j *CloseDecisionsJob) Schedule() {
|
|
var nextDue *time.Time
|
|
nextDue, err := GetNextPendingDecisionDue()
|
|
if err != nil {
|
|
logger.Errorf("Could not get next pending due date")
|
|
if j.timer != nil {
|
|
j.timer.Stop()
|
|
j.timer = nil
|
|
}
|
|
return
|
|
}
|
|
if nextDue == nil {
|
|
logger.Infof("no next planned execution of CloseDecisionsJob")
|
|
j.Stop()
|
|
} else {
|
|
nextDue := nextDue.Add(time.Minute)
|
|
logger.Infof("scheduling CloseDecisionsJob for %s", nextDue)
|
|
when := nextDue.Sub(time.Now())
|
|
if j.timer != nil {
|
|
j.timer.Reset(when)
|
|
} else {
|
|
j.timer = time.AfterFunc(when, j.Run)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (j *CloseDecisionsJob) Stop() {
|
|
if j.timer != nil {
|
|
j.timer.Stop()
|
|
j.timer = nil
|
|
}
|
|
}
|
|
|
|
func (j *CloseDecisionsJob) Run() {
|
|
logger.Infof("running CloseDecisionsJob")
|
|
err := CloseDecisions()
|
|
if err != nil {
|
|
logger.Errorf("closing decisions %s", err)
|
|
}
|
|
rescheduleChannel <- JobIdCloseDecisions
|
|
}
|
|
|
|
func (j *CloseDecisionsJob) String() string {
|
|
return "CloseDecisionsJob"
|
|
}
|
|
|
|
type RemindVotersJob struct {
|
|
timer *time.Timer
|
|
}
|
|
|
|
func NewRemindVotersJob() *RemindVotersJob {
|
|
job := &RemindVotersJob{}
|
|
job.Schedule()
|
|
return job
|
|
}
|
|
|
|
func (j *RemindVotersJob) Schedule() {
|
|
year, month, day := time.Now().UTC().Date()
|
|
nextExecution := time.Date(year, month, day, 0, 0, 0, 0, time.UTC).AddDate(0, 0, 3)
|
|
logger.Infof("scheduling RemindVotersJob for %s", nextExecution)
|
|
when := nextExecution.Sub(time.Now())
|
|
if j.timer != nil {
|
|
j.timer.Reset(when)
|
|
} else {
|
|
j.timer = time.AfterFunc(when, j.Run)
|
|
}
|
|
}
|
|
|
|
func (j *RemindVotersJob) Stop() {
|
|
if j.timer != nil {
|
|
j.timer.Stop()
|
|
j.timer = nil
|
|
}
|
|
}
|
|
|
|
func (j *RemindVotersJob) Run() {
|
|
logger.Infof("running RemindVotersJob")
|
|
defer func() { rescheduleChannel <- JobIdRemindVotersJob }()
|
|
|
|
voters, err := GetReminderVoters()
|
|
if err != nil {
|
|
logger.Errorf("problem getting voters %s", err)
|
|
return
|
|
}
|
|
|
|
for _, voter := range *voters {
|
|
decisions, err := FindUnvotedDecisionsForVoter(&voter)
|
|
if err != nil {
|
|
logger.Errorf("problem getting unvoted decisions")
|
|
return
|
|
}
|
|
if len(*decisions) > 0 {
|
|
NotifyMailChannel <- &RemindVoterNotification{voter: voter, decisions: *decisions}
|
|
}
|
|
}
|
|
}
|