Fix compile

Author: Sushisource

temporalio/bridge/src/worker.rs

@@ -15,8 +15,9 @@ use temporal_sdk_core::api::errors::PollError;
 use temporal_sdk_core::replay::{HistoryForReplay, ReplayWorkerInput};
 use temporal_sdk_core_api::errors::WorkflowErrorType;
 use temporal_sdk_core_api::worker::{
-    SlotInfo, SlotInfoTrait, SlotKind, SlotKindType, SlotMarkUsedContext, SlotReleaseContext,
-    SlotReservationContext, SlotSupplier as SlotSupplierTrait, SlotSupplierPermit,
+    PollerBehavior, SlotInfo, SlotInfoTrait, SlotKind, SlotKindType, SlotMarkUsedContext,
+    SlotReleaseContext, SlotReservationContext, SlotSupplier as SlotSupplierTrait,
+    SlotSupplierPermit,
 };
 use temporal_sdk_core_api::Worker;
 use temporal_sdk_core_protos::coresdk::workflow_completion::WorkflowActivationCompletion;
@@ -44,7 +45,7 @@ pub struct WorkerRef {
 pub struct WorkerConfig {
     namespace: String,
     task_queue: String,
-    build_id: String,
+    versioning_strategy: WorkerVersioningStrategy,
     identity_override: Option<String>,
     max_cached_workflows: usize,
     tuner: TunerHolder,
@@ -58,11 +59,34 @@ pub struct WorkerConfig {
     max_activities_per_second: Option<f64>,
     max_task_queue_activities_per_second: Option<f64>,
     graceful_shutdown_period_millis: u64,
-    use_worker_versioning: bool,
     nondeterminism_as_workflow_fail: bool,
     nondeterminism_as_workflow_fail_for_types: HashSet<String>,
 }
 
+/// Recreates [temporal_sdk_core_api::worker::WorkerVersioningStrategy]
+#[derive(FromPyObject)]
+pub enum WorkerVersioningStrategy {
+    None { build_id: String },
+    WorkerDeploymentBased(WorkerDeploymentOptions),
+    LegacyBuildIdBased { build_id: String },
+}
+
+/// Recreates [temporal_sdk_core_api::worker::WorkerDeploymentOptions]
+#[derive(FromPyObject)]
+pub struct WorkerDeploymentOptions {
+    pub version: WorkerDeploymentVersion,
+    pub use_worker_versioning: bool,
+    /// This is a [enums::v1::VersioningBehavior] represented as i32
+    pub default_versioning_behavior: i32,
+}
+
+/// Recreates [temporal_sdk_core_api::worker::WorkerDeploymentVersion]
+#[derive(FromPyObject)]
+pub struct WorkerDeploymentVersion {
+    pub deployment_name: String,
+    pub build_id: String,
+}
+
 #[derive(FromPyObject)]
 pub struct TunerHolder {
     workflow_slot_supplier: SlotSupplier,
@@ -559,16 +583,21 @@ fn convert_worker_config(
     task_locals: Arc<OnceLock<pyo3_asyncio::TaskLocals>>,
 ) -> PyResult<temporal_sdk_core::WorkerConfig> {
     let converted_tuner = convert_tuner_holder(conf.tuner, task_locals)?;
+    let converted_versioning_strategy = convert_versioning_strategy(conf.versioning_strategy);
     temporal_sdk_core::WorkerConfigBuilder::default()
         .namespace(conf.namespace)
         .task_queue(conf.task_queue)
-        .worker_build_id(conf.build_id)
+        .versioning_strategy(converted_versioning_strategy)
         .client_identity_override(conf.identity_override)
         .max_cached_workflows(conf.max_cached_workflows)
-        .max_concurrent_wft_polls(conf.max_concurrent_workflow_task_polls)
+        .workflow_task_poller_behavior(PollerBehavior::SimpleMaximum(
+            conf.max_concurrent_workflow_task_polls,
+        ))
         .tuner(Arc::new(converted_tuner))
         .nonsticky_to_sticky_poll_ratio(conf.nonsticky_to_sticky_poll_ratio)
-        .max_concurrent_at_polls(conf.max_concurrent_activity_task_polls)
+        .activity_task_poller_behavior(PollerBehavior::SimpleMaximum(
+            conf.max_concurrent_activity_task_polls,
+        ))
         .no_remote_activities(conf.no_remote_activities)
         .sticky_queue_schedule_to_start_timeout(Duration::from_millis(
             conf.sticky_queue_schedule_to_start_timeout_millis,
@@ -585,7 +614,6 @@ fn convert_worker_config(
         // auto-cancel-activity behavior of shutdown will not occur, so we
         // always set it even if 0.
         .graceful_shutdown_period(Duration::from_millis(conf.graceful_shutdown_period_millis))
-        .use_worker_versioning(conf.use_worker_versioning)
         .workflow_failure_errors(if conf.nondeterminism_as_workflow_fail {
             HashSet::from([WorkflowErrorType::Nondeterminism])
         } else {
@@ -702,6 +730,36 @@ fn convert_slot_supplier<SK: SlotKind + Send + Sync + 'static>(
     })
 }
 
+fn convert_versioning_strategy(
+    strategy: WorkerVersioningStrategy,
+) -> temporal_sdk_core_api::worker::WorkerVersioningStrategy {
+    match strategy {
+        WorkerVersioningStrategy::None { build_id } => {
+            temporal_sdk_core_api::worker::WorkerVersioningStrategy::None { build_id }
+        }
+        WorkerVersioningStrategy::WorkerDeploymentBased(worker_deployment_options) => {
+            temporal_sdk_core_api::worker::WorkerVersioningStrategy::WorkerDeploymentBased(
+                temporal_sdk_core_api::worker::WorkerDeploymentOptions {
+                    version: temporal_sdk_core_api::worker::WorkerDeploymentVersion {
+                        deployment_name: worker_deployment_options.version.deployment_name,
+                        build_id: worker_deployment_options.version.build_id,
+                    },
+                    use_worker_versioning: worker_deployment_options.use_worker_versioning,
+                    default_versioning_behavior: Some(
+                        worker_deployment_options
+                            .default_versioning_behavior
+                            .try_into()
+                            .unwrap_or_default(),
+                    ),
+                },
+            )
+        }
+        WorkerVersioningStrategy::LegacyBuildIdBased { build_id } => {
+            temporal_sdk_core_api::worker::WorkerVersioningStrategy::LegacyBuildIdBased { build_id }
+        }
+    }
+}
+
 /// For feeding histories into core during replay
 #[pyclass]
 pub struct HistoryPusher {