Fix: setting dep status on all deps, not just the first

This commit is contained in:
Laszlo Fogas 2021-05-25 13:05:31 +02:00
parent 11417f0f7c
commit 3cb8d06b78
3 changed files with 62 additions and 22 deletions

View file

@ -356,8 +356,8 @@ func (q *fifo) updateDepStatusInQueue(taskID string, status string) {
}
}
var n *list.Element
for e := q.waitingOnDeps.Front(); e != nil; e = n {
next = nil
for e := q.waitingOnDeps.Front(); e != nil; e = next {
next = e.Next()
waiting, ok := e.Value.(*Task)
for _, dep := range waiting.Dependencies {

View file

@ -207,8 +207,6 @@ func TestFifoErrors2(t *testing.T) {
task2 := &Task{
ID: "2",
Dependencies: []string{"1"},
DepStatus: make(map[string]string),
}
task3 := &Task{
@ -220,23 +218,22 @@ func TestFifoErrors2(t *testing.T) {
q := New().(*fifo)
q.PushAtOnce(noContext, []*Task{task2, task3, task1})
for i := 0; i < 2; i++ {
got, _ := q.Poll(noContext, func(*Task) bool { return true })
if got != task1 && got != task2{
t.Errorf("expect task1 or task2 returned from queue as task3 depends on them")
return
}
if got != task1 {
q.Done(noContext, got.ID, StatusSuccess)
}
if got != task2 {
q.Error(noContext, got.ID, fmt.Errorf("exitcode 1, there was an error"))
}
}
got, _ := q.Poll(noContext, func(*Task) bool { return true })
if got != task1 {
t.Errorf("expect task1 returned from queue as task2 and task3 depends on it")
return
}
q.Done(noContext, got.ID, StatusSuccess)
got, _ = q.Poll(noContext, func(*Task) bool { return true })
if got != task2 {
t.Errorf("expect task2 returned from queue")
return
}
q.Error(noContext, got.ID, fmt.Errorf("exitcode 1, there was an error"))
got, _ = q.Poll(noContext, func(*Task) bool { return true })
if got != task3 {
t.Errorf("expect task3 returned from queue")
return
@ -249,6 +246,7 @@ func TestFifoErrors2(t *testing.T) {
}
func TestFifoErrorsMultiThread(t *testing.T) {
//logrus.SetLevel(logrus.DebugLevel)
task1 := &Task{
ID: "1",
}
@ -294,7 +292,14 @@ func TestFifoErrorsMultiThread(t *testing.T) {
return
} else {
task1Processed = true
q.Done(noContext, got.ID, StatusSuccess)
q.Error(noContext, got.ID, fmt.Errorf("exitcode 1, there was an error"))
go func() {
for {
fmt.Printf("Worker spawned\n")
got, _ := q.Poll(noContext, func(*Task) bool { return true })
obtainedWorkCh <- got
}
}()
}
} else if !task2Processed {
if got != task2 {
@ -302,7 +307,14 @@ func TestFifoErrorsMultiThread(t *testing.T) {
return
} else {
task2Processed = true
q.Error(noContext, got.ID, fmt.Errorf("exitcode 1, there was an error"))
q.Done(noContext, got.ID, StatusSuccess)
go func() {
for {
fmt.Printf("Worker spawned\n")
got, _ := q.Poll(noContext, func(*Task) bool { return true })
obtainedWorkCh <- got
}
}()
}
} else {
if got != task3 {
@ -319,6 +331,8 @@ func TestFifoErrorsMultiThread(t *testing.T) {
}
case <-time.After(5 * time.Second):
info := q.Info(noContext)
fmt.Println(info.String())
t.Errorf("test timed out")
return
}

View file

@ -3,6 +3,8 @@ package queue
import (
"context"
"errors"
"fmt"
"strings"
)
var (
@ -61,6 +63,12 @@ func (t *Task) ShouldRun() bool {
return false
}
func (t *Task) String() string {
var sb strings.Builder
sb.WriteString(fmt.Sprintf("%s (%s) - %s", t.ID, t.Dependencies, t.DepStatus))
return sb.String()
}
func runsOnFailure(runsOn []string) bool {
for _, status := range runsOn {
if status == "failure" {
@ -98,6 +106,24 @@ type InfoT struct {
Paused bool
}
func (t *InfoT) String() string {
var sb strings.Builder
for _, task := range t.Pending {
sb.WriteString("\t" + task.String())
}
for _, task := range t.Running {
sb.WriteString("\t" + task.String())
}
for _, task := range t.WaitingOnDeps {
sb.WriteString("\t" + task.String())
}
return sb.String()
}
// Filter filters tasks in the queue. If the Filter returns false,
// the Task is skipped and not returned to the subscriber.
type Filter func(*Task) bool