fix(raiko): refine error return (#378)

* fix(raiko): refine error return * fix lint Signed-off-by: smtmfft <[email protected]> * fix lint Signed-off-by: smtmfft <[email protected]> * fix zk entrypoint log level Signed-off-by: smtmfft <[email protected]> * Update host/src/server/api/v2/mod.rs Co-authored-by: Petar Vujović <[email protected]> --------- Signed-off-by: smtmfft <[email protected]> Co-authored-by: Petar Vujović <[email protected]>
taikoxyz · Sep 20, 2024 · f4f818d · f4f818d
1 parent c2b0db5
commit f4f818d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 8 deletions.
diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
@@ -182,5 +182,5 @@ if [[ -n $ZK ]]; then
         update_raiko_sgx_instance_id $RAIKO_CONF_BASE_CONFIG
         update_docker_chain_specs $RAIKO_CONF_CHAIN_SPECS
 
-        RUST_LOG=debug /opt/raiko/bin/raiko-host "$@"
+        /opt/raiko/bin/raiko-host "$@"
 fi
diff --git a/host/src/proof.rs b/host/src/proof.rs
@@ -111,8 +111,8 @@ impl ProofActor {
                 }
                 result = Self::handle_message(proof_request, key.clone(), &opts, &chain_specs) => {
                     match result {
-                        Ok(()) => {
-                            info!("Host handling message");
+                        Ok(status) => {
+                            info!("Host handling message: {status:?}");
                         }
                         Err(error) => {
                             error!("Worker failed due to: {error:?}");
@@ -151,14 +151,14 @@ impl ProofActor {
         key: TaskDescriptor,
         opts: &Opts,
         chain_specs: &SupportedChainSpecs,
-    ) -> HostResult<()> {
+    ) -> HostResult<TaskStatus> {
         let mut manager = get_task_manager(&opts.clone().into());
 
         let status = manager.get_task_proving_status(&key).await?;
 
         if let Some(latest_status) = status.iter().last() {
             if !matches!(latest_status.0, TaskStatus::Registered) {
-                return Ok(());
+                return Ok(latest_status.0);
             }
         }
 
@@ -178,7 +178,8 @@ impl ProofActor {
         manager
             .update_task_progress(key, status, proof.as_deref())
             .await
-            .map_err(|e| e.into())
+            .map_err(HostError::from)?;
+        Ok(status)
     }
 }
 

diff --git a/host/src/server/api/v2/mod.rs b/host/src/server/api/v2/mod.rs
@@ -84,8 +84,14 @@ impl From<Vec<u8>> for Status {
 
 impl From<TaskStatus> for Status {
     fn from(status: TaskStatus) -> Self {
-        Self::Ok {
-            data: ProofResponse::Status { status },
+        match status {
+            TaskStatus::Success | TaskStatus::WorkInProgress | TaskStatus::Registered => Self::Ok {
+                data: ProofResponse::Status { status },
+            },
+            _ => Self::Error {
+                error: "task_failed".to_string(),
+                message: format!("Task failed with status: {status:?}"),
+            },
         }
     }
 }