protected void ProcessWorkerStateChangeReports(WorkerStateChangedMessage message)
{
PipelineSection pipelineSection = FindPipelineSection(message.WorkerBadge.SectionName);
if (pipelineSection == null)
{
return;
}
WorkerStatus workerStatus;
bool found = pipelineSection.WorkerStatuses.TryGetValue(message.WorkerBadge.WorkerId, out workerStatus);
if (found)
{
workerStatus.WorkerState = message.WorkerState;
}
else
{
workerStatus = new WorkerStatus()
{
WorkerBadge = message.WorkerBadge, WorkerState = message.WorkerState
};
pipelineSection.WorkerStatuses.Add(message.WorkerBadge.WorkerId, workerStatus);
Tracer.Trace("Pipeline {0}, section {1} added new worker {2} with state {3}",
PipelineId, pipelineSection.Name, message.WorkerBadge.WorkerId, workerStatus.WorkerState);
}
// Debug
//Tracer.Warning("Worker " + message.WorkerBadge.WorkerId + " reported state " + message.WorkerState);
if (pipelineSection.OrderNumber == 0 && message.WorkerState == WorkerState.Started)
{
Tracer.Info("Pipeline initialization completed in {0:0.##} seconds. First worker started filling up the pipe.",
_runTime.Elapsed.TotalSeconds);
}
// Handling first worker completion
if (pipelineSection.OrderNumber == 0 && message.WorkerState == WorkerState.Completed)
{
// First worker reported that it generated at least one message to the pipe.
PipelineStatus.PipelineState = PipelineState.FirstWorkerCompleted;
CheckPipelineForCompletion();
}
// Handling worker quit
if (message.WorkerState == WorkerState.Quit)
{
if (pipelineSection.WorkersNotQuit == 0)
{
long remainingWorkers = PipelineSections.Aggregate <PipelineSection, long>(0, (current, curPipelineSection) => current + curPipelineSection.WorkersNotQuit);
if (remainingWorkers == 0)
{
PipelineStatus.PipelineState = PipelineState.AllWorkersQuit;
}
}
}
}
private void ReportStateChange()
{
WorkerStateChangedMessage message = new WorkerStateChangedMessage(WorkAssignment, State);
var envelope = new PipeMessageEnvelope()
{
Body = message,
Label = WorkerStateChangedMessage.MessageLabel,
};
ReportPipe.Send(envelope);
}
private void ProcessReport(PipeMessageEnvelope envelope)
{
if (_reportPipeMessageCounter++ == 0)
{
_timeSinceTheFirstReportReceived.Start();
}
object message = envelope.Body;
WorkerStateChangedMessage workerStateChangedMessage = message as WorkerStateChangedMessage;
if (workerStateChangedMessage != null)
{
ProcessWorkerStateChangeReports(workerStateChangedMessage);
}
WorkerStatistics workerStatistics = message as WorkerStatistics;
if (workerStatistics != null)
{
ProcessWorkerStatistics(workerStatistics);
}
WorkerMessage workerMessage = message as WorkerMessage;
if (workerMessage != null)
{
ProcessWorkerMessage(workerMessage);
}
ProblemReport problemReport = message as ProblemReport;
if (problemReport != null)
{
ProcessProblemReport(problemReport);
}
}
private void ReportStateChange()
{
WorkerStateChangedMessage message = new WorkerStateChangedMessage(WorkAssignment, State);
var envelope = new PipeMessageEnvelope()
{
Body = message,
Label = WorkerStateChangedMessage.MessageLabel,
};
ReportPipe.Send(envelope);
}