work stealing scheduler for agent fleets: leases, heartbeats, retries with provenance attached.
1 package sched
2
3 // Baux de travail. Un agent ne "prend" pas une tache, il obtient un bail
4 // qui expire : si l'agent meurt, la tache revient d'elle meme.
5
6 import "time"
7
8 type Lease struct {
9 TaskID string
10 Agent string
11 ExpiresAt time.Time
12 Renewals int
13 }
14
15 // Le vol de tache prefere le bail le plus ancien, pas la tache la plus
16 // courte : sinon un agent rapide draine la file courte et la longue traine
17 // meurt de faim (voir issue #11).
18 func (s *Scheduler) Steal(now time.Time) (*Lease, bool) {
19 var oldest *Lease
20 for _, l := range s.expired(now) {
21 if oldest == nil || l.ExpiresAt.Before(oldest.ExpiresAt) {
22 oldest = l
23 }
24 }
25 if oldest == nil {
26 return nil, false
27 }
28 return oldest, true
29 }
30
31 // Renew etend le bail. Le compteur de renouvellements sert au ledger de
32 // retry : on veut savoir quel agent a echoue, et combien de fois.
33 func (l *Lease) Renew(d time.Duration) {
34 l.ExpiresAt = time.Now().Add(d)
35 l.Renewals++
36 }