Skip to content

Commit

Permalink
[RPC] Define RPC call to notify backend of Task completion.
Browse files Browse the repository at this point in the history
  • Loading branch information
sukritkalra committed Feb 21, 2024
1 parent 093c8ce commit 99f1351
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions rpc/protos/erdos_scheduler.proto
Original file line number Diff line number Diff line change
Expand Up @@ -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) {}
}


Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 99f1351

Please sign in to comment.