svn commit: r1132306 - in /incubator/mesos/trunk/src/slave: slave.cpp slave.hpp

[be...@apache.org]

Author: benh
Date: Sun Jun  5 09:22:45 2011
New Revision: 1132306

URL: http://svn.apache.org/viewvc?rev=1132306&view=rev
Log:
Fixed slave to properly record updated task states.

Modified:
    incubator/mesos/trunk/src/slave/slave.cpp
    incubator/mesos/trunk/src/slave/slave.hpp

Modified: incubator/mesos/trunk/src/slave/slave.cpp
URL: http://svn.apache.org/viewvc/incubator/mesos/trunk/src/slave/slave.cpp?rev=1132306&r1=1132305&r2=1132306&view=diff
==============================================================================
--- incubator/mesos/trunk/src/slave/slave.cpp (original)
+++ incubator/mesos/trunk/src/slave/slave.cpp Sun Jun  5 09:22:45 2011
@@ -556,6 +556,7 @@ void Slave::statusUpdate(const Framework
   if (framework != NULL) {
     Executor* executor = framework->getExecutor(status.task_id());
     if (executor != NULL) {
+      executor->updateTaskState(status.task_id(), status.state());
       if (status.state() == TASK_FINISHED ||
           status.state() == TASK_FAILED ||
           status.state() == TASK_KILLED ||

Modified: incubator/mesos/trunk/src/slave/slave.hpp
URL: http://svn.apache.org/viewvc/incubator/mesos/trunk/src/slave/slave.hpp?rev=1132306&r1=1132305&r2=1132306&view=diff
==============================================================================
--- incubator/mesos/trunk/src/slave/slave.hpp (original)
+++ incubator/mesos/trunk/src/slave/slave.hpp Sun Jun  5 09:22:45 2011
@@ -106,6 +106,13 @@ struct Executor
     }
   }
 
+  void updateTaskState(const TaskID& taskId, TaskState state)
+  {
+    if (tasks.count(taskId) > 0) {
+      tasks[taskId]->set_state(state);
+    }
+  }
+
   const FrameworkID frameworkId;
   const ExecutorInfo info;