Skip to content

Commit

Permalink
fix: sendNextScheduled 移至 Server.Serve 中调用
Browse files Browse the repository at this point in the history
  • Loading branch information
caixw committed Dec 23, 2024
1 parent c9313a6 commit 77ad633
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 7 deletions.
2 changes: 1 addition & 1 deletion schedulers/scheduler.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ import "time"

// Scheduler 时间调度算法需要实现的接口
type Scheduler interface {
// Next 生成相对于 last 的下一次时间
// Next 生成相对于 last 的下一次时间
//
// 如果返回的时间值,已经小于当前时间,那么该任务会被安排在最先执行。
// 如果返回是零值,表示该调度已经终结,不会再执行该任务,后续也都应返回零值。
Expand Down
11 changes: 5 additions & 6 deletions server.go
Original file line number Diff line number Diff line change
Expand Up @@ -78,13 +78,12 @@ func (s *Server) Serve(ctx context.Context) error {
if err := s.schedule(ctx); err != nil {
return err
}
s.sendNextScheduled()
}
}
}

// sendNextScheduled 立即触发一次任务调度
//
// clear 为 true 可以在触发之前确保前一次的调度任务已经退出。
func (s *Server) sendNextScheduled() {
s.nextScheduledLocker.Lock()
if len(s.nextScheduled) == 0 {
Expand All @@ -93,7 +92,7 @@ func (s *Server) sendNextScheduled() {
s.nextScheduledLocker.Unlock()
}

// sendNextScheduled 立即触发一次任务调度
// clearAndSendNextScheduled 清除现在有任务并立即触发一次任务调度
//
// 如果有正在运行的 schedule,则会让其退出。
func (s *Server) clearAndSendNextScheduled() {
Expand Down Expand Up @@ -131,10 +130,12 @@ func (s *Server) schedule(ctx context.Context) error {
for {
select {
case <-ctx.Done():
timer.Stop()
return ctx.Err()
case <-timer.C: // 计时结束,表示 jobs 没有变化,直接跳至 LOOP 部分执行
goto LOOP
case <-s.exitSchedule:
timer.Stop()
return nil
}
}
Expand All @@ -156,11 +157,9 @@ LOOP:
go j.run(now, s.erro, s.info)
}

if len(s.exitSchedule) > 0 { // 在退出函数和执行 sendNextScheduled 清空 exitSchedule
if len(s.exitSchedule) > 0 { // 在退出函数和执行 sendNextScheduled 之前清空 exitSchedule
<-s.exitSchedule
}

s.sendNextScheduled()

return nil
}

0 comments on commit 77ad633

Please sign in to comment.