IDK what i'm doing here

Author: slinkydeveloper

Cargo.toml

@@ -12,6 +12,7 @@ default = ["http_server", "rand", "uuid"]
 hyper = ["dep:hyper", "http-body-util", "restate-sdk-shared-core/http"]
 http_server = ["hyper", "hyper/server", "hyper/http2", "hyper-util", "tokio/net", "tokio/signal", "tokio/macros"]
 
+
 [dependencies]
 bytes = "1.6.1"
 futures = "0.3"
@@ -23,8 +24,7 @@ pin-project-lite = "0.2"
 rand = { version = "0.8.5", optional = true }
 regress = "0.10"
 restate-sdk-macros = { version = "0.3.2", path = "macros" }
-restate-sdk-shared-core = "0.1.0"
-sha2 = "=0.11.0-pre.3"
+restate-sdk-shared-core = { git = "https://github.com/restatedev/sdk-shared-core.git", branch = "main", features = ["request_identity", "sha2_random_seed", "http"] }
 serde = "1.0"
 serde_json = "1.0"
 thiserror = "1.0.63"

examples/cron.rs

@@ -77,7 +77,7 @@ impl PeriodicTaskImpl {
             .object_client::<PeriodicTaskClient>(context.key())
             .run()
             // And send with a delay
-            .send_with_delay(Duration::from_secs(10));
+            .send_after(Duration::from_secs(10));
     }
 }
 

src/context/mod.rs

@@ -8,7 +8,7 @@ use std::time::Duration;
 
 mod request;
 mod run;
-pub use request::{Request, RequestTarget};
+pub use request::{CallFuture, InvocationHandle, Request, RequestTarget};
 pub use run::{RunClosure, RunFuture, RunRetryPolicy};
 
 pub type HeaderMap = http::HeaderMap<String>;
@@ -433,6 +433,11 @@ pub trait ContextClient<'ctx>: private::SealedContext<'ctx> {
         Request::new(self.inner_context(), request_target, req)
     }
 
+    /// Create an [`InvocationHandle`] from an invocation id.
+    fn invocation_handle(&self, invocation_id: String) -> impl InvocationHandle + 'ctx {
+        self.inner_context().invocation_handle(invocation_id)
+    }
+
     /// Create a service client. The service client is generated by the [`restate_sdk_macros::service`] macro with the same name of the trait suffixed with `Client`.
     ///
     /// ```rust,no_run
@@ -454,7 +459,7 @@ pub trait ContextClient<'ctx>: private::SealedContext<'ctx> {
     /// client.handle().send();
     ///
     /// // Schedule the request to be executed later
-    /// client.handle().send_with_delay(Duration::from_secs(60));
+    /// client.handle().send_after(Duration::from_secs(60));
     /// # }
     /// ```
     fn service_client<C>(&self) -> C
@@ -485,7 +490,7 @@ pub trait ContextClient<'ctx>: private::SealedContext<'ctx> {
     /// client.handle().send();
     ///
     /// // Schedule the request to be executed later
-    /// client.handle().send_with_delay(Duration::from_secs(60));
+    /// client.handle().send_after(Duration::from_secs(60));
     /// # }
     /// ```
     fn object_client<C>(&self, key: impl Into<String>) -> C
@@ -516,7 +521,7 @@ pub trait ContextClient<'ctx>: private::SealedContext<'ctx> {
     /// client.handle().send();
     ///
     /// // Schedule the request to be executed later
-    /// client.handle().send_with_delay(Duration::from_secs(60));
+    /// client.handle().send_after(Duration::from_secs(60));
     /// # }
     /// ```
     fn workflow_client<C>(&self, key: impl Into<String>) -> C

src/context/request.rs

@@ -87,7 +87,7 @@ impl<'a, Req, Res> Request<'a, Req, Res> {
     }
 
     /// Call a service. This returns a future encapsulating the response.
-    pub fn call(self) -> impl Future<Output = Result<Res, TerminalError>> + Send
+    pub fn call(self) -> impl CallFuture<Result<Res, TerminalError>> + Send
     where
         Req: Serialize + 'static,
         Res: Deserialize + 'static,
@@ -96,18 +96,25 @@ impl<'a, Req, Res> Request<'a, Req, Res> {
     }
 
     /// Send the request to the service, without waiting for the response.
-    pub fn send(self)
+    pub fn send(self) -> impl InvocationHandle
     where
         Req: Serialize + 'static,
     {
         self.ctx.send(self.request_target, self.req, None)
     }
 
     /// Schedule the request to the service, without waiting for the response.
-    pub fn send_with_delay(self, duration: Duration)
+    pub fn send_after(self, delay: Duration) -> impl InvocationHandle
     where
         Req: Serialize + 'static,
     {
-        self.ctx.send(self.request_target, self.req, Some(duration))
+        self.ctx.send(self.request_target, self.req, Some(delay))
     }
 }
+
+pub trait InvocationHandle {
+    fn invocation_id(&self) -> impl Future<Output = Result<String, TerminalError>> + Send;
+    fn cancel(&self);
+}
+
+pub trait CallFuture<O>: Future<Output = O> + InvocationHandle {}