temporalio 0.0.1

data/bridge/sdk-core/sdk-core-protos/src/task_token.rs

@@ -0,0 +1,38 @@
+use std::fmt::{Debug, Display, Formatter};
+
+static LOCAL_ACT_TASK_TOKEN_PREFIX: &[u8] = b"local_act_";
+
+#[derive(Hash, Eq, PartialEq, Clone, derive_more::From, derive_more::Into)]
+/// Type-safe wrapper for task token bytes
+pub struct TaskToken(pub Vec<u8>);
+
+impl TaskToken {
+    /// Task tokens for local activities are always prefixed with a special sigil so they can
+    /// be identified easily
+    pub fn new_local_activity_token(unique_data: impl IntoIterator<Item = u8>) -> Self {
+        let mut bytes = LOCAL_ACT_TASK_TOKEN_PREFIX.to_vec();
+        bytes.extend(unique_data);
+        TaskToken(bytes)
+    }
+
+    /// Returns true if the task token is for a local activity
+    pub fn is_local_activity_task(&self) -> bool {
+        self.0.starts_with(LOCAL_ACT_TASK_TOKEN_PREFIX)
+    }
+}
+
+impl Display for TaskToken {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        f.write_str(&fmt_tt(&self.0))
+    }
+}
+
+impl Debug for TaskToken {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        f.write_str(&format!("TaskToken({})", fmt_tt(&self.0)))
+    }
+}
+
+pub fn fmt_tt(tt: &[u8]) -> String {
+    base64::encode(tt)
+}

data/bridge/sdk-core/sdk-core-protos/src/utilities.rs

@@ -0,0 +1,14 @@
+pub trait TryIntoOrNone<F, T> {
+    /// Turn an option of something into an option of another thing, trying to convert along the way
+    /// and returning `None` if that conversion fails
+    fn try_into_or_none(self) -> Option<T>;
+}
+
+impl<F, T> TryIntoOrNone<F, T> for Option<F>
+where
+    F: TryInto<T>,
+{
+    fn try_into_or_none(self) -> Option<T> {
+        self.map(TryInto::try_into).transpose().ok().flatten()
+    }
+}

data/bridge/sdk-core/test-utils/Cargo.toml

@@ -0,0 +1,35 @@
+[package]
+name = "temporal-sdk-core-test-utils"
+version = "0.1.0"
+authors = ["Spencer Judge <spencer@temporal.io>"]
+edition = "2021"
+
+[[bin]]
+name = "histfetch"
+path = "src/histfetch.rs"
+
+[dependencies]
+anyhow = "1.0"
+async-trait = "0.1"
+base64 = "0.13"
+futures = "0.3"
+log = "0.4"
+parking_lot = "0.12"
+prost = "0.11"
+prost-types = "0.11"
+rand = "0.8"
+serde_json = "1.0"
+temporal-client = { path = "../client" }
+temporal-sdk = { path = "../sdk" }
+temporal-sdk-core = { path = "../core" }
+temporal-sdk-core-api = { path = "../core-api" }
+thiserror = "1.0"
+tokio = "1.1"
+tokio-util = { version = "0.7" }
+tracing = "0.1"
+url = "2.2"
+uuid = "1.1"
+
+[dependencies.temporal-sdk-core-protos]
+path = "../sdk-core-protos"
+version = "0.1"