// TODO: codes here are slightly different from java since the protobuf.net does not generate the HasXXX method, may want to switch to proto-port later
/// <summary>
/// Processes the given ContextControlProto to launch / close / suspend Tasks and Contexts.
/// This also triggers the HeartBeatManager to send a heartbeat with the result of this operation.
/// </summary>
/// <param name="controlMessage"></param>
public void HandleTaskControl(ContextControlProto controlMessage)
{
try
{
byte[] message = controlMessage.task_message;
if (controlMessage.add_context != null && controlMessage.remove_context != null)
{
Org.Apache.REEF.Utilities.Diagnostics.Exceptions.Throw(new InvalidOperationException("Received a message with both add and remove context. This is unsupported."), LOGGER);
}
if (controlMessage.add_context != null)
{
LOGGER.Log(Level.Info, "AddContext");
AddContext(controlMessage.add_context);
// support submitContextAndTask()
if (controlMessage.start_task != null)
{
LOGGER.Log(Level.Info, "StartTask");
StartTask(controlMessage.start_task);
}
else
{
// We need to trigger a heartbeat here. In other cases, the heartbeat will be triggered by the TaskRuntime
// Therefore this call can not go into addContext
LOGGER.Log(Level.Info, "Trigger Heartbeat");
_heartBeatManager.OnNext();
}
}
else if (controlMessage.remove_context != null)
{
LOGGER.Log(Level.Info, string.Format(CultureInfo.InvariantCulture, "RemoveContext with id {0}", controlMessage.remove_context.context_id));
RemoveContext(controlMessage.remove_context.context_id);
}
else if (controlMessage.start_task != null)
{
LOGGER.Log(Level.Info, "StartTask only");
StartTask(controlMessage.start_task);
}
else if (controlMessage.stop_task != null)
{
LOGGER.Log(Level.Info, "CloseTask");
_contextStack.Peek().CloseTask(message);
}
else if (controlMessage.suspend_task != null)
{
LOGGER.Log(Level.Info, "SuspendTask");
_contextStack.Peek().SuspendTask(message);
}
else if (controlMessage.task_message != null)
{
LOGGER.Log(Level.Info, "DeliverTaskMessage");
_contextStack.Peek().DeliverTaskMessage(message);
}
else if (controlMessage.context_message != null)
{
LOGGER.Log(Level.Info, "Handle context contol message");
ContextMessageProto contextMessageProto = controlMessage.context_message;
bool deliveredMessage = false;
foreach (ContextRuntime context in _contextStack)
{
if (context.Id.Equals(contextMessageProto.context_id))
{
LOGGER.Log(Level.Info, string.Format(CultureInfo.InvariantCulture, "Handle context message {0}", controlMessage.context_message.message));
context.HandleContextMessaage(controlMessage.context_message.message);
deliveredMessage = true;
break;
}
}
if (!deliveredMessage)
{
InvalidOperationException e = new InvalidOperationException(string.Format(CultureInfo.InvariantCulture, "Sent message to unknown context {0}", contextMessageProto.context_id));
Org.Apache.REEF.Utilities.Diagnostics.Exceptions.Throw(e, LOGGER);
}
}
else
{
InvalidOperationException e = new InvalidOperationException(string.Format(CultureInfo.InvariantCulture, "Unknown task control message: {0}", controlMessage.ToString()));
Org.Apache.REEF.Utilities.Diagnostics.Exceptions.Throw(e, LOGGER);
}
}
catch (Exception e)
{
if (e is TaskClientCodeException)
{
HandleTaskException((TaskClientCodeException)e);
}
else if (e is ContextClientCodeException)
{
HandlContextException((ContextClientCodeException)e);
}
Org.Apache.REEF.Utilities.Diagnostics.Exceptions.CaughtAndThrow(e, Level.Error, LOGGER);
}
}
private void Heartbeat()
{
_heartBeatManager.OnNext(ToProto());
}