//Execute executes a stage using the supplied mediator
func (e *Engine) Execute(stage stages.Stage, mediator stages.Mediator) stages.Mediator {
mediator.Type = "start"
name := stage.GetStageName()
log.Debugf("----- Execute %v start ------\n", name)
go func() {
*stage.GetInputCh() <- mediator
close(*stage.GetInputCh())
}()
go e.ExecuteStage(stage)
e.waitCloseOutputCh(stage)
return e.waitMonitorChFinalized(name)
}
// 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)
}
func execute(stage stages.Stage) stages.Mediator {
mon := make(chan stages.Mediator)
e := &Engine{
MonitorCh: &mon,
Resources: &pipelines.Resources{
Reporter: &messengers.FakeMessenger{},
},
EnvVariables: config.NewEnvVariables(),
}
go e.ExecuteStage(stage)
mediator := stages.Mediator{States: make(map[string]string), Type: "start"}
go func() {
*stage.GetInputCh() <- mediator
close(*stage.GetInputCh())
}()
for {
_, ok := <-*stage.GetOutputCh()
if !ok {
break
}
}
var m stages.Mediator
acm := stages.Mediator{States: make(map[string]string)}
for {
m = <-mon
for k, v := range m.States {
acm.States[k] = v
}
if m.Type == "end" {
break
}
}
return acm
}