catch exceptions on workers and pass them to the scheduler (#93)

This commit is contained in:
Robert Nishihara
2016-06-11 15:44:56 -07:00
committed by Philipp Moritz
parent 61a0014c00
commit 41539141af
8 changed files with 38 additions and 15 deletions
+5 -3
View File
@@ -263,15 +263,17 @@ Task* Worker::receive_next_task() {
return task;
}
void Worker::notify_task_completed() {
void Worker::notify_task_completed(bool task_succeeded, std::string error_message) {
if (!connected_) {
RAY_LOG(RAY_FATAL, "Attempting to perform notify_task_completed, but connected_ = " << connected_ << ".");
}
ClientContext context;
WorkerReadyRequest request;
NotifyTaskCompletedRequest request;
request.set_workerid(workerid_);
request.set_task_succeeded(task_succeeded);
request.set_error_message(error_message);
AckReply reply;
scheduler_stub_->WorkerReady(&context, request, &reply);
scheduler_stub_->NotifyTaskCompleted(&context, request, &reply);
}
void Worker::disconnect() {