@temporalio/core-bridge 1.14.2-canary-release-testing.0 → 1.16.0

package/sdk-core/crates/sdk/src/activity_context.rs

@@ -1,238 +0,0 @@
-use crate::app_data::AppData;
-
-use prost_types::{Duration, Timestamp};
-use std::{
-    collections::HashMap,
-    sync::Arc,
-    time::{Duration as StdDuration, SystemTime},
-};
-use temporalio_client::Priority;
-use temporalio_common::{
-    Worker,
-    protos::{
-        coresdk::{ActivityHeartbeat, activity_task},
-        temporal::api::common::v1::{Payload, RetryPolicy, WorkflowExecution},
-        utilities::TryIntoOrNone,
-    },
-};
-use tokio_util::sync::CancellationToken;
-
-/// Used within activities to get info, heartbeat management etc.
-#[derive(Clone)]
-pub struct ActContext {
-    worker: Arc<dyn Worker>,
-    app_data: Arc<AppData>,
-    cancellation_token: CancellationToken,
-    input: Vec<Payload>,
-    heartbeat_details: Vec<Payload>,
-    header_fields: HashMap<String, Payload>,
-    info: ActivityInfo,
-}
-
-#[derive(Clone)]
-pub struct ActivityInfo {
-    pub task_token: Vec<u8>,
-    pub workflow_type: String,
-    pub workflow_namespace: String,
-    pub workflow_execution: Option<WorkflowExecution>,
-    pub activity_id: String,
-    pub activity_type: String,
-    pub task_queue: String,
-    pub heartbeat_timeout: Option<StdDuration>,
-    /// Time activity was scheduled by a workflow
-    pub scheduled_time: Option<SystemTime>,
-    /// Time of activity start
-    pub started_time: Option<SystemTime>,
-    /// Time of activity timeout
-    pub deadline: Option<SystemTime>,
-    /// Attempt starts from 1, and increase by 1 for every retry, if retry policy is specified.
-    pub attempt: u32,
-    /// Time this attempt at the activity was scheduled
-    pub current_attempt_scheduled_time: Option<SystemTime>,
-    pub retry_policy: Option<RetryPolicy>,
-    pub is_local: bool,
-    /// Priority of this activity. If unset uses [Priority::default]
-    pub priority: Priority,
-}
-
-impl ActContext {
-    /// Construct new Activity Context, returning the context and the first argument to the activity
-    /// (which may be a default [Payload]).
-    pub fn new(
-        worker: Arc<dyn Worker>,
-        app_data: Arc<AppData>,
-        cancellation_token: CancellationToken,
-        task_queue: String,
-        task_token: Vec<u8>,
-        task: activity_task::Start,
-    ) -> (Self, Payload) {
-        let activity_task::Start {
-            workflow_namespace,
-            workflow_type,
-            workflow_execution,
-            activity_id,
-            activity_type,
-            header_fields,
-            mut input,
-            heartbeat_details,
-            scheduled_time,
-            current_attempt_scheduled_time,
-            started_time,
-            attempt,
-            schedule_to_close_timeout,
-            start_to_close_timeout,
-            heartbeat_timeout,
-            retry_policy,
-            is_local,
-            priority,
-        } = task;
-        let deadline = calculate_deadline(
-            scheduled_time.as_ref(),
-            started_time.as_ref(),
-            start_to_close_timeout.as_ref(),
-            schedule_to_close_timeout.as_ref(),
-        );
-        let first_arg = input.pop().unwrap_or_default();
-
-        (
-            ActContext {
-                worker,
-                app_data,
-                cancellation_token,
-                input,
-                heartbeat_details,
-                header_fields,
-                info: ActivityInfo {
-                    task_token,
-                    task_queue,
-                    workflow_type,
-                    workflow_namespace,
-                    workflow_execution,
-                    activity_id,
-                    activity_type,
-                    heartbeat_timeout: heartbeat_timeout.try_into_or_none(),
-                    scheduled_time: scheduled_time.try_into_or_none(),
-                    started_time: started_time.try_into_or_none(),
-                    deadline,
-                    attempt,
-                    current_attempt_scheduled_time: current_attempt_scheduled_time
-                        .try_into_or_none(),
-                    retry_policy,
-                    is_local,
-                    priority: priority.map(Into::into).unwrap_or_default(),
-                },
-            },
-            first_arg,
-        )
-    }
-
-    /// Returns a future the completes if and when the activity this was called inside has been
-    /// cancelled
-    pub async fn cancelled(&self) {
-        self.cancellation_token.clone().cancelled().await
-    }
-
-    /// Returns true if this activity has already been cancelled
-    pub fn is_cancelled(&self) -> bool {
-        self.cancellation_token.is_cancelled()
-    }
-
-    /// Retrieve extra parameters to the Activity. The first input is always popped and passed to
-    /// the Activity function for the currently executing activity. However, if more parameters are
-    /// passed, perhaps from another language's SDK, explicit access is available from extra_inputs
-    pub fn extra_inputs(&mut self) -> &mut [Payload] {
-        &mut self.input
-    }
-
-    /// Extract heartbeat details from last failed attempt. This is used in combination with retry policy.
-    pub fn get_heartbeat_details(&self) -> &[Payload] {
-        &self.heartbeat_details
-    }
-
-    /// RecordHeartbeat sends heartbeat for the currently executing activity
-    pub fn record_heartbeat(&self, details: Vec<Payload>) {
-        if !self.info.is_local {
-            self.worker.record_activity_heartbeat(ActivityHeartbeat {
-                task_token: self.info.task_token.clone(),
-                details,
-            })
-        }
-    }
-
-    /// Get activity info of the executing activity
-    pub fn get_info(&self) -> &ActivityInfo {
-        &self.info
-    }
-
-    /// Get headers attached to this activity
-    pub fn headers(&self) -> &HashMap<String, Payload> {
-        &self.header_fields
-    }
-
-    /// Get custom Application Data
-    pub fn app_data<T: Send + Sync + 'static>(&self) -> Option<&T> {
-        self.app_data.get::<T>()
-    }
-}
-
-/// Deadline calculation.  This is a port of
-/// https://github.com/temporalio/sdk-go/blob/8651550973088f27f678118f997839fb1bb9e62f/internal/activity.go#L225
-fn calculate_deadline(
-    scheduled_time: Option<&Timestamp>,
-    started_time: Option<&Timestamp>,
-    start_to_close_timeout: Option<&Duration>,
-    schedule_to_close_timeout: Option<&Duration>,
-) -> Option<SystemTime> {
-    match (
-        scheduled_time,
-        started_time,
-        start_to_close_timeout,
-        schedule_to_close_timeout,
-    ) {
-        (
-            Some(scheduled),
-            Some(started),
-            Some(start_to_close_timeout),
-            Some(schedule_to_close_timeout),
-        ) => {
-            let scheduled: SystemTime = maybe_convert_timestamp(scheduled)?;
-            let started: SystemTime = maybe_convert_timestamp(started)?;
-            let start_to_close_timeout: StdDuration = (*start_to_close_timeout).try_into().ok()?;
-            let schedule_to_close_timeout: StdDuration =
-                (*schedule_to_close_timeout).try_into().ok()?;
-
-            let start_to_close_deadline: SystemTime =
-                started.checked_add(start_to_close_timeout)?;
-            if schedule_to_close_timeout > StdDuration::ZERO {
-                let schedule_to_close_deadline =
-                    scheduled.checked_add(schedule_to_close_timeout)?;
-                // Minimum of the two deadlines.
-                if schedule_to_close_deadline < start_to_close_deadline {
-                    Some(schedule_to_close_deadline)
-                } else {
-                    Some(start_to_close_deadline)
-                }
-            } else {
-                Some(start_to_close_deadline)
-            }
-        }
-        _ => None,
-    }
-}
-
-/// Helper function lifted from prost_types::Timestamp implementation to prevent double cloning in
-/// error construction
-fn maybe_convert_timestamp(timestamp: &Timestamp) -> Option<SystemTime> {
-    let mut timestamp = *timestamp;
-    timestamp.normalize();
-
-    let system_time = if timestamp.seconds >= 0 {
-        std::time::UNIX_EPOCH.checked_add(StdDuration::from_secs(timestamp.seconds as u64))
-    } else {
-        std::time::UNIX_EPOCH.checked_sub(StdDuration::from_secs((-timestamp.seconds) as u64))
-    };
-
-    system_time.and_then(|system_time| {
-        system_time.checked_add(StdDuration::from_nanos(timestamp.nanos as u64))
-    })
-}

package/sdk-core/crates/sdk/src/app_data.rs

@@ -1,37 +0,0 @@
-use std::{
-    any::{Any, TypeId},
-    collections::HashMap,
-    fmt,
-};
-
-/// A Wrapper Type for workflow and activity app data
-#[derive(Default)]
-pub struct AppData {
-    map: HashMap<TypeId, Box<dyn Any + Send + Sync>>,
-}
-
-impl AppData {
-    /// Insert an item, overwritting duplicates
-    pub(crate) fn insert<T: Send + Sync + 'static>(&mut self, val: T) -> Option<T> {
-        self.map
-            .insert(TypeId::of::<T>(), Box::new(val))
-            .and_then(downcast_owned)
-    }
-
-    /// Get a reference to a type in the map
-    pub(crate) fn get<T: 'static>(&self) -> Option<&T> {
-        self.map
-            .get(&TypeId::of::<T>())
-            .and_then(|boxed| boxed.downcast_ref())
-    }
-}
-
-impl fmt::Debug for AppData {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        f.debug_struct("AppData").finish()
-    }
-}
-
-fn downcast_owned<T: Send + Sync + 'static>(boxed: Box<dyn Any + Send + Sync>) -> Option<T> {
-    boxed.downcast().ok().map(|boxed| *boxed)
-}

package/sdk-core/crates/sdk-core/tests/integ_tests/activity_functions.rs

@@ -1,5 +0,0 @@
-use temporalio_sdk::{ActContext, ActivityError};
-
-pub(crate) async fn echo(_ctx: ActContext, e: String) -> Result<String, ActivityError> {
-    Ok(e)
-}

package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/appdata_propagation.rs

@@ -1,61 +0,0 @@
-use crate::common::CoreWfStarter;
-use assert_matches::assert_matches;
-use std::time::Duration;
-use temporalio_client::{WfClientExt, WorkflowExecutionResult, WorkflowOptions};
-use temporalio_common::protos::coresdk::AsJsonPayloadExt;
-use temporalio_sdk::{ActContext, ActivityOptions, WfContext, WorkflowResult};
-
-const TEST_APPDATA_MESSAGE: &str = "custom app data, yay";
-
-struct Data {
-    message: String,
-}
-
-pub(crate) async fn appdata_activity_wf(ctx: WfContext) -> WorkflowResult<()> {
-    ctx.activity(ActivityOptions {
-        activity_type: "echo_activity".to_string(),
-        start_to_close_timeout: Some(Duration::from_secs(5)),
-        input: "hi!".as_json_payload().expect("serializes fine"),
-        ..Default::default()
-    })
-    .await;
-    Ok(().into())
-}
-
-#[tokio::test]
-async fn appdata_access_in_activities_and_workflows() {
-    let wf_name = "appdata_activity";
-    let mut starter = CoreWfStarter::new(wf_name);
-    let mut worker = starter.worker().await;
-    worker.inner_mut().insert_app_data(Data {
-        message: TEST_APPDATA_MESSAGE.to_owned(),
-    });
-
-    let client = starter.get_client().await;
-    worker.register_wf(wf_name.to_owned(), appdata_activity_wf);
-    worker.register_activity(
-        "echo_activity",
-        |ctx: ActContext, echo_me: String| async move {
-            let data = ctx.app_data::<Data>().expect("appdata exists. qed");
-            assert_eq!(data.message, TEST_APPDATA_MESSAGE.to_owned());
-            Ok(echo_me)
-        },
-    );
-
-    let run_id = worker
-        .submit_wf(
-            wf_name.to_owned(),
-            wf_name.to_owned(),
-            vec![],
-            WorkflowOptions::default(),
-        )
-        .await
-        .unwrap();
-    worker.run_until_done().await.unwrap();
-    let handle = client.get_untyped_workflow_handle(wf_name, run_id);
-    let res = handle
-        .get_workflow_result(Default::default())
-        .await
-        .unwrap();
-    assert_matches!(res, WorkflowExecutionResult::Succeeded(_));
-}