[ENH] Propagate error codes through rust services

rust/error/Cargo.toml

@@ -7,3 +7,4 @@ edition = "2021"
 path = "src/lib.rs"
 
 [dependencies]
+tonic = { workspace = true }

rust/error/src/lib.rs

@@ -54,3 +54,28 @@ impl ChromaError for Box<dyn ChromaError> {
         self.as_ref().code()
     }
 }
+
+impl From<ErrorCodes> for tonic::Code {
+    fn from(err: ErrorCodes) -> tonic::Code {
+        match err {
+            ErrorCodes::Success => tonic::Code::Ok,
+            ErrorCodes::Cancelled => tonic::Code::Cancelled,
+            ErrorCodes::Unknown => tonic::Code::Unknown,
+            ErrorCodes::InvalidArgument => tonic::Code::InvalidArgument,
+            ErrorCodes::DeadlineExceeded => tonic::Code::DeadlineExceeded,
+            ErrorCodes::NotFound => tonic::Code::NotFound,
+            ErrorCodes::AlreadyExists => tonic::Code::AlreadyExists,
+            ErrorCodes::PermissionDenied => tonic::Code::PermissionDenied,
+            ErrorCodes::ResourceExhausted => tonic::Code::ResourceExhausted,
+            ErrorCodes::FailedPrecondition => tonic::Code::FailedPrecondition,
+            ErrorCodes::Aborted => tonic::Code::Aborted,
+            ErrorCodes::OutOfRange => tonic::Code::OutOfRange,
+            ErrorCodes::Unimplemented => tonic::Code::Unimplemented,
+            ErrorCodes::Internal => tonic::Code::Internal,
+            ErrorCodes::Unavailable => tonic::Code::Unavailable,
+            ErrorCodes::DataLoss => tonic::Code::DataLoss,
+            ErrorCodes::Unauthenticated => tonic::Code::Unauthenticated,
+            ErrorCodes::VersionMismatch => tonic::Code::Internal,
+        }
+    }
+}

rust/worker/src/server.rs

@@ -205,10 +205,11 @@ impl WorkerServer {
         let result = match result {
             Ok(result) => result,
             Err(e) => {
-                return Err(Status::internal(format!(
-                    "Error running orchestrator: {}",
-                    e
-                )));
+                tracing::error!("Error running orchestrator: {}", e);
+                return Err(Status::new(
+                    e.code().into(),
+                    format!("Error running orchestrator: {}", e),
+                ));
             }
         };
 
@@ -308,10 +309,11 @@ impl WorkerServer {
         let mut result = match result {
             Ok(result) => result,
             Err(e) => {
-                return Err(Status::internal(format!(
-                    "Error running orchestrator: {}",
-                    e
-                )));
+                tracing::error!("Error running orchestrator: {}", e);
+                return Err(Status::new(
+                    e.code().into(),
+                    format!("Error running orchestrator: {}", e),
+                ));
             }
         };
 
@@ -435,10 +437,10 @@ impl WorkerServer {
             Ok(result) => result,
             Err(e) => {
                 tracing::error!("Error running orchestrator: {}", e);
-                return Err(Status::internal(format!(
-                    "Error running orchestrator: {}",
-                    e
-                )));
+                return Err(Status::new(
+                    e.code().into(),
+                    format!("Error running orchestrator: {}", e),
+                ));
             }
         };