func (e *Engine) finalizeMonitorChAfterExecute(mediators []stages.Mediator, mediator stages.Mediator) {
// Append to monitorCh
*e.MonitorCh <- mediator
for _, m := range mediators {
*e.MonitorCh <- m
}
// Set type
if mediators[0].Type == "start" {
log.Debug("Finalize monitor channel..")
mediatorEnd := stages.Mediator{States: make(map[string]string), Type: "end"}
*e.MonitorCh <- mediatorEnd
} else {
log.Debugf("Skipped finalizing")
}
}
func (e *Engine) waitAllChildStages(childStages *list.List, stage *stages.Stage) {
for childStage := childStages.Front(); childStage != nil; childStage = childStage.Next() {
s := childStage.Value.(stages.Stage)
for {
log.Debugf("Receiving child: %v", s.GetStageName())
childReceived, ok := <-*s.GetOutputCh()
if !ok {
log.Debug("Closing child output")
break
}
log.Debugf("Sending child: %v", childReceived)
*(*stage).GetOutputCh() <- childReceived
log.Debugf("Send child: %v", childReceived)
}
log.Debugf("Finished executing child: %v", s.GetStageName())
}
}
// ExecuteStage executes the supplied stage
func (e *Engine) ExecuteStage(stage stages.Stage) {
log.Debug("Receiving input")
mediatorsReceived := e.receiveInputs(stage.GetInputCh())
log.Debugf("Received input size: %v", len(mediatorsReceived))
log.Debugf("Mediator received: %+v", mediatorsReceived)
log.Debugf("Execute as parent: %+v", stage)
log.Debugf("Execute as parent name %+v", stage.GetStageName())
mediator := stages.Mediator{States: make(map[string]string)}
mediator.States[stage.GetStageName()] = e.executeStage(stage, mediatorsReceived, mediator)
log.Debugf("Sending output of stage: %+v %v", stage.GetStageName(), mediator)
*stage.GetOutputCh() <- mediator
close(*stage.GetOutputCh())
log.Debugf("Closed output of stage: %+v", stage.GetStageName())
e.finalizeMonitorChAfterExecute(mediatorsReceived, mediator)
}