Skip to content

Commit

Permalink
template: factor onTemplateRendered out of template re-renderer
Browse files Browse the repository at this point in the history
Pull this function out so it can be used by the first-pass renderer when
we restore a task from a restarted client.
  • Loading branch information
tgross committed Dec 11, 2020
1 parent 863c194 commit 815f01c
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions client/allocrunner/taskrunner/template/template.go
Original file line number Diff line number Diff line change
Expand Up @@ -368,6 +368,12 @@ func (tm *TaskTemplateManager) handleTemplateRerenders(allRenderedTime time.Time
SetFailsTask().
SetDisplayMessage(fmt.Sprintf("Template failed: %v", err)))
case <-tm.runner.TemplateRenderedCh():
tm.onTemplateRendered(handledRenders, allRenderedTime)
}
}
}

func (tm *TaskTemplateManager) onTemplateRendered(handledRenders map[string]time.Time, allRenderedTime time.Time) {
// A template has been rendered, figure out what to do
var handling []string
signals := make(map[string]struct{})
Expand Down Expand Up @@ -472,8 +478,7 @@ func (tm *TaskTemplateManager) handleTemplateRerenders(allRenderedTime time.Time
}
}
}
}
}

}

// allTemplatesNoop returns whether all the managed templates have change mode noop.
Expand Down

0 comments on commit 815f01c

Please sign in to comment.