Catching exception

aneoconsulting · Jul 28, 2023 · c64a5b8 · c64a5b8
1 parent 53b4b9c
commit c64a5b8
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/packages/cpp/ArmoniK.Api.Worker/source/Worker/ArmoniKWorker.cpp b/packages/cpp/ArmoniK.Api.Worker/source/Worker/ArmoniKWorker.cpp
@@ -59,20 +59,23 @@ Status API_WORKER_NAMESPACE::ArmoniKWorker::Process([[maybe_unused]] ::grpc::Ser
   TaskHandler task_handler(agent, iterator);
 
   task_handler.init();
+  try {
+    ProcessStatus status = Execute(task_handler);
 
-  ProcessStatus status = Execute(task_handler);
+    logger_.debug("Finish call C++");
 
-  logger_.debug("Finish call C++");
+    armonik::api::grpc::v1::Output output;
+    if (status.ok()) {
+      *output.mutable_ok() = armonik::api::grpc::v1::Empty();
+    } else {
+      output.mutable_error()->set_details(status.details());
+    }
 
-  armonik::api::grpc::v1::Output output;
-  if (status.ok()) {
-    *output.mutable_ok() = armonik::api::grpc::v1::Empty();
-  } else {
-    output.mutable_error()->set_details(status.details());
+    *response->mutable_output() = output;
+  } catch (const std::exception &e) {
+    return {grpc::StatusCode::UNAVAILABLE, "Error processing task", e.what()};
   }
 
-  *response->mutable_output() = output;
-
   return grpc::Status::OK;
 }