Introduce interface to export InvocationStatus (#526)

restatedev · Jun 26, 2023 · 001aca9 · 001aca9
1 parent a41878a
commit 001aca9
Show file tree

Hide file tree

Showing 6 changed files with 286 additions and 35 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/invoker/Cargo.toml b/src/invoker/Cargo.toml
@@ -17,6 +17,7 @@ bytes-utils = { workspace = true }
 restate_common = { workspace = true, features = ["serde"] }
 restate_hyper_util = { workspace = true }
 restate_fs_util = { workspace = true }
+restate_futures_util = { workspace = true }
 drain = { workspace = true }
 futures = { workspace = true }
 hyper = { workspace = true, features = ["http1", "http2", "client", "tcp", "stream", "runtime"] }

diff --git a/src/invoker/src/invocation_task.rs b/src/invoker/src/invocation_task.rs
@@ -17,7 +17,7 @@ use hyper::{http, Body, Request, Response, Uri};
 use opentelemetry::propagation::TextMapPropagator;
 use opentelemetry::sdk::propagation::TraceContextPropagator;
 use opentelemetry_http::HeaderInjector;
-use restate_common::errors::{InvocationError, UserErrorCode};
+use restate_common::errors::{InvocationError, InvocationErrorCode, UserErrorCode};
 use restate_common::types::{
  EnrichedRawEntry, EntryIndex, PartitionLeaderEpoch, ServiceInvocationId,
  ServiceInvocationSpanContext,
@@ -96,15 +96,20 @@ impl InvokerError for InvocationTaskError {
  fn is_transient(&self) -> bool {
  true
  }
-}
 
-impl From<InvocationTaskError> for InvocationError {
- fn from(value: InvocationTaskError) -> Self {
- match value {
- InvocationTaskError::Invocation(e) => e,
+ fn to_invocation_error(&self) -> InvocationError {
+ match self {
+ InvocationTaskError::Invocation(e) => e.clone(),
  e => InvocationError::new(UserErrorCode::Internal, e.to_string()),
  }
  }
+
+ fn as_invocation_error_code(&self) -> InvocationErrorCode {
+ match self {
+ InvocationTaskError::Invocation(e) => e.code(),
+ _ => UserErrorCode::Internal.into(),
+ }
+ }
 }
 
 // Copy pasted from hyper::Error