diff --git a/rpc/protos/erdos_scheduler.proto b/rpc/protos/erdos_scheduler.proto index b3932caf..4f439073 100644 --- a/rpc/protos/erdos_scheduler.proto +++ b/rpc/protos/erdos_scheduler.proto @@ -44,6 +44,9 @@ service SchedulerService { /// This RPC is invoked periodically by the framework along with the current timestamp /// and the backend scheduler informs the framework about the current allocations. rpc GetPlacements(GetPlacementsRequest) returns (GetPlacementsResponse) {} + + /// Notifies the Scheduler that a Task from a particular TaskGraph has completed.option + rpc NotifyTaskCompletion(NotifyTaskCompletionRequest) returns (NotifyTaskCompletionResponse) {} } @@ -147,6 +150,17 @@ message RegisterEnvironmentReadyResponse { string message = 2; } +message NotifyTaskCompletionRequest { + string app_id = 1; + uint32 task_id = 2; + uint64 timestamp = 3; +} + +message NotifyTaskCompletionResponse { + bool success = 1; + string message = 2; +} + /// Worker related RPC structures. /// The following structures represent the data that a Worker's addition to the /// framework must pass to the scheduling backend. This may involve things like