temporalio 0.0.2 → 0.1.0

data/bridge/src/worker.rs

@@ -1,14 +1,15 @@
 use crate::connection::Client;
 use crate::runtime::{Callback, Command, Runtime};
 use prost::Message;
-use std::sync::Arc;
 use std::sync::mpsc::Sender;
-use temporal_sdk_core::api::{Worker as WorkerTrait};
-use temporal_sdk_core_api::errors::{PollActivityError};
+use std::sync::Arc;
+use temporal_sdk_core::api::Worker as WorkerTrait;
+use temporal_sdk_core_api::errors::{PollActivityError, PollWfError};
 use temporal_sdk_core_api::worker::{WorkerConfigBuilder, WorkerConfigBuilderError};
+use temporal_sdk_core_protos::coresdk::workflow_completion::WorkflowActivationCompletion;
 use temporal_sdk_core_protos::coresdk::{ActivityHeartbeat, ActivityTaskCompletion};
 use thiserror::Error;
-use tokio::runtime::{Runtime as TokioRuntime};
+use tokio::runtime::Runtime as TokioRuntime;
 
 #[derive(Error, Debug)]
 pub enum WorkerError {
@@ -24,12 +25,24 @@ pub enum WorkerError {
     #[error(transparent)]
     UnableToPollActivityTask(#[from] temporal_sdk_core::api::errors::PollActivityError),
 
+    #[error(transparent)]
+    UnableToPollWorkflowActivation(#[from] temporal_sdk_core::api::errors::PollWfError),
+
     #[error(transparent)]
     UnableToCompleteActivityTask(#[from] temporal_sdk_core::api::errors::CompleteActivityError),
 
+    #[error(transparent)]
+    UnableToCompleteWorkflowActivation(#[from] temporal_sdk_core::api::errors::CompleteWfError),
+
     #[error("Unable to send a request. Channel is closed")]
     ChannelClosed(),
 
+    #[error("Worker has already been shutdown")]
+    WorkerHasBeenShutdown(),
+
+    #[error("Unexpected error: {0}")]
+    UnexpectedError(String),
+
     #[error("Core worker is shutting down")]
     Shutdown(),
 }
@@ -37,18 +50,20 @@ pub enum WorkerError {
 pub type WorkerResult = Result<Vec<u8>, WorkerError>;
 
 pub struct Worker {
-    core_worker: Arc<temporal_sdk_core::Worker>,
+    core_worker: Option<Arc<temporal_sdk_core::Worker>>,
     tokio_runtime: Arc<TokioRuntime>,
     callback_tx: Sender<Command>,
 }
 
 impl Worker {
     // TODO: Extend this to include full worker config
-    pub fn new(runtime: &Runtime, client: &Client, namespace: &str, task_queue: &str) -> Result<Worker, WorkerError> {
+    pub fn new(runtime: &Runtime, client: &Client, namespace: &str, task_queue: &str, max_cached_workflows: u32, no_remote_activity: bool) -> Result<Worker, WorkerError> {
         let config = WorkerConfigBuilder::default()
             .namespace(namespace)
             .task_queue(task_queue)
             .worker_build_id("test-worker-build") // TODO: replace this with an actual build id
+            .max_cached_workflows(usize::try_from(max_cached_workflows).unwrap())
+            .no_remote_activities(no_remote_activity)
             .build()?;
 
         let core_worker = runtime.tokio_runtime.block_on(async move {
@@ -56,14 +71,14 @@ impl Worker {
         }).expect("Failed to initialize Core Worker");
 
         Ok(Worker {
-            core_worker: Arc::new(core_worker),
+            core_worker: Some(Arc::new(core_worker)),
             tokio_runtime: runtime.tokio_runtime.clone(),
             callback_tx: runtime.callback_tx.clone(),
         })
     }
 
     pub fn poll_activity_task<F>(&self, callback: F) -> Result<(), WorkerError> where F: FnOnce(WorkerResult) + Send + 'static {
-        let core_worker = self.core_worker.clone();
+        let core_worker = Arc::clone(self.core_worker.as_ref().ok_or(WorkerError::WorkerHasBeenShutdown())?);
         let callback_tx = self.callback_tx.clone();
 
         self.tokio_runtime.spawn(async move {
@@ -85,7 +100,7 @@ impl Worker {
     }
 
     pub fn complete_activity_task<F>(&self, bytes: Vec<u8>, callback: F) -> Result<(), WorkerError> where F: FnOnce(WorkerResult) + Send + 'static {
-        let core_worker = self.core_worker.clone();
+        let core_worker = Arc::clone(self.core_worker.as_ref().ok_or(WorkerError::WorkerHasBeenShutdown())?);
         let callback_tx = self.callback_tx.clone();
         let proto = ActivityTaskCompletion::decode(&*bytes)?;
 
@@ -104,21 +119,79 @@ impl Worker {
     }
 
     pub fn record_activity_heartbeat(&self, bytes: Vec<u8>) -> Result<(), WorkerError> {
+        let core_worker = Arc::clone(self.core_worker.as_ref().ok_or(WorkerError::WorkerHasBeenShutdown())?);
         let proto = ActivityHeartbeat::decode(&*bytes)?;
-        self.core_worker.record_activity_heartbeat(proto);
+
+        core_worker.record_activity_heartbeat(proto);
+
+        Ok(())
+    }
+
+    pub fn poll_workflow_activation<F>(&self, callback: F) -> Result<(), WorkerError> where F: FnOnce(WorkerResult) + Send + 'static {
+        let core_worker = Arc::clone(self.core_worker.as_ref().ok_or(WorkerError::WorkerHasBeenShutdown())?);
+        let callback_tx = self.callback_tx.clone();
+
+        self.tokio_runtime.spawn(async move {
+            let result = core_worker.poll_workflow_activation().await;
+
+            let callback: Callback = match result {
+                Ok(task) => {
+                    let bytes = task.encode_to_vec();
+                    Box::new(move || callback(Ok(bytes)))
+                },
+                Err(PollWfError::ShutDown) => Box::new(move || callback(Err(WorkerError::Shutdown()))),
+                Err(e) => Box::new(move || callback(Err(WorkerError::UnableToPollWorkflowActivation(e))))
+            };
+
+            callback_tx.send(Command::RunCallback(callback)).expect("Unable to send a callback");
+        });
 
         Ok(())
     }
 
-    pub fn initiate_shutdown(&self) {
-        self.core_worker.initiate_shutdown();
+    pub fn complete_workflow_activation<F>(&self, bytes: Vec<u8>, callback: F) -> Result<(), WorkerError> where F: FnOnce(WorkerResult) + Send + 'static {
+        let core_worker = Arc::clone(self.core_worker.as_ref().ok_or(WorkerError::WorkerHasBeenShutdown())?);
+        let callback_tx = self.callback_tx.clone();
+        let proto = WorkflowActivationCompletion::decode(&*bytes)?;
+
+        self.tokio_runtime.spawn(async move {
+            let result = core_worker.complete_workflow_activation(proto).await;
+
+            let callback: Callback = match result {
+                Ok(()) => Box::new(move || callback(Ok(vec!()))),
+                Err(e) => Box::new(move || callback(Err(WorkerError::UnableToCompleteWorkflowActivation(e))))
+            };
+
+            callback_tx.send(Command::RunCallback(callback)).expect("Unable to send a callback");
+        });
+
+        Ok(())
     }
 
-    pub fn shutdown(&self) {
-        let core_worker = self.core_worker.clone();
+    pub fn initiate_shutdown(&self) -> Result<(), WorkerError> {
+        let core_worker = Arc::clone(self.core_worker.as_ref().ok_or(WorkerError::WorkerHasBeenShutdown())?);
+
+        core_worker.initiate_shutdown();
+
+        Ok(())
+    }
+
+    pub fn finalize_shutdown(&mut self) -> Result<(), WorkerError> {
+        // Take the worker out of the option and leave None
+        let core_worker = self.core_worker.take().ok_or(WorkerError::WorkerHasBeenShutdown())?;
+
+        // By design, there should be no more task using the worker by the time the Ruby code calls finalize_shutdown.
+        // This should therefore be the very last reference remaining to the worker. This is a condition for try_unwrap to work.
+        let core_worker = Arc::try_unwrap(core_worker).map_err(|arc| {
+            WorkerError::UnexpectedError(format!(
+                "Can't finalize worker's shutdown because there are still active references to it (expected exactly 1 reference, but got {})",
+                Arc::strong_count(&arc)
+            ))
+        })?;
 
         self.tokio_runtime.block_on(async move {
-            core_worker.shutdown().await;
-        });
+            core_worker.finalize_shutdown().await;
+            Ok(())
+        })
     }
 }

data/lib/gen/temporal/api/command/v1/message_pb.rb

@@ -9,7 +9,6 @@ require 'temporal/api/enums/v1/workflow_pb'
 require 'temporal/api/enums/v1/command_type_pb'
 require 'temporal/api/common/v1/message_pb'
 require 'temporal/api/failure/v1/message_pb'
-require 'temporal/api/interaction/v1/message_pb'
 require 'temporal/api/taskqueue/v1/message_pb'
 
 Google::Protobuf::DescriptorPool.generated_pool.build do
@@ -109,17 +108,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :memo, :message, 15, "temporal.api.common.v1.Memo"
       optional :search_attributes, :message, 16, "temporal.api.common.v1.SearchAttributes"
     end
-    add_message "temporal.api.command.v1.AcceptWorkflowUpdateCommandAttributes" do
-      optional :meta, :message, 1, "temporal.api.interaction.v1.Meta"
-      optional :input, :message, 2, "temporal.api.interaction.v1.Input"
-    end
-    add_message "temporal.api.command.v1.CompleteWorkflowUpdateCommandAttributes" do
-      optional :meta, :message, 1, "temporal.api.interaction.v1.Meta"
-      optional :output, :message, 2, "temporal.api.interaction.v1.Output"
-    end
-    add_message "temporal.api.command.v1.RejectWorkflowUpdateCommandAttributes" do
-      optional :meta, :message, 1, "temporal.api.interaction.v1.Meta"
-      optional :failure, :message, 2, "temporal.api.failure.v1.Failure"
+    add_message "temporal.api.command.v1.ProtocolMessageCommandAttributes" do
+      optional :message_id, :string, 1
     end
     add_message "temporal.api.command.v1.Command" do
       optional :command_type, :enum, 1, "temporal.api.enums.v1.CommandType"
@@ -137,10 +127,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
         optional :start_child_workflow_execution_command_attributes, :message, 12, "temporal.api.command.v1.StartChildWorkflowExecutionCommandAttributes"
         optional :signal_external_workflow_execution_command_attributes, :message, 13, "temporal.api.command.v1.SignalExternalWorkflowExecutionCommandAttributes"
         optional :upsert_workflow_search_attributes_command_attributes, :message, 14, "temporal.api.command.v1.UpsertWorkflowSearchAttributesCommandAttributes"
-        optional :accept_workflow_update_command_attributes, :message, 15, "temporal.api.command.v1.AcceptWorkflowUpdateCommandAttributes"
-        optional :complete_workflow_update_command_attributes, :message, 16, "temporal.api.command.v1.CompleteWorkflowUpdateCommandAttributes"
+        optional :protocol_message_command_attributes, :message, 15, "temporal.api.command.v1.ProtocolMessageCommandAttributes"
         optional :modify_workflow_properties_command_attributes, :message, 17, "temporal.api.command.v1.ModifyWorkflowPropertiesCommandAttributes"
-        optional :reject_workflow_update_command_attributes, :message, 18, "temporal.api.command.v1.RejectWorkflowUpdateCommandAttributes"
       end
     end
   end
@@ -164,9 +152,7 @@ module Temporalio
         RecordMarkerCommandAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.RecordMarkerCommandAttributes").msgclass
         ContinueAsNewWorkflowExecutionCommandAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.ContinueAsNewWorkflowExecutionCommandAttributes").msgclass
         StartChildWorkflowExecutionCommandAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.StartChildWorkflowExecutionCommandAttributes").msgclass
-        AcceptWorkflowUpdateCommandAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.AcceptWorkflowUpdateCommandAttributes").msgclass
-        CompleteWorkflowUpdateCommandAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.CompleteWorkflowUpdateCommandAttributes").msgclass
-        RejectWorkflowUpdateCommandAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.RejectWorkflowUpdateCommandAttributes").msgclass
+        ProtocolMessageCommandAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.ProtocolMessageCommandAttributes").msgclass
         Command = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.command.v1.Command").msgclass
       end
     end

data/lib/gen/temporal/api/common/v1/message_pb.rb

@@ -46,6 +46,9 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :maximum_attempts, :int32, 4
       repeated :non_retryable_error_types, :string, 5
     end
+    add_message "temporal.api.common.v1.MeteringMetadata" do
+      optional :nonfirst_local_activity_execution_attempts, :uint32, 13
+    end
   end
 end
 
@@ -63,6 +66,7 @@ module Temporalio
         WorkflowType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.common.v1.WorkflowType").msgclass
         ActivityType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.common.v1.ActivityType").msgclass
         RetryPolicy = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.common.v1.RetryPolicy").msgclass
+        MeteringMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.common.v1.MeteringMetadata").msgclass
       end
     end
   end

data/lib/gen/temporal/api/enums/v1/command_type_pb.rb

@@ -20,10 +20,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :COMMAND_TYPE_START_CHILD_WORKFLOW_EXECUTION, 11
       value :COMMAND_TYPE_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION, 12
       value :COMMAND_TYPE_UPSERT_WORKFLOW_SEARCH_ATTRIBUTES, 13
-      value :COMMAND_TYPE_ACCEPT_WORKFLOW_UPDATE, 14
-      value :COMMAND_TYPE_COMPLETE_WORKFLOW_UPDATE, 15
+      value :COMMAND_TYPE_PROTOCOL_MESSAGE, 14
       value :COMMAND_TYPE_MODIFY_WORKFLOW_PROPERTIES, 16
-      value :COMMAND_TYPE_REJECT_WORKFLOW_UPDATE, 17
     end
   end
 end

data/lib/gen/temporal/api/enums/v1/event_type_pb.rb

@@ -47,9 +47,9 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :EVENT_TYPE_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED, 38
       value :EVENT_TYPE_EXTERNAL_WORKFLOW_EXECUTION_SIGNALED, 39
       value :EVENT_TYPE_UPSERT_WORKFLOW_SEARCH_ATTRIBUTES, 40
-      value :EVENT_TYPE_WORKFLOW_UPDATE_REJECTED, 41
-      value :EVENT_TYPE_WORKFLOW_UPDATE_ACCEPTED, 42
-      value :EVENT_TYPE_WORKFLOW_UPDATE_COMPLETED, 43
+      value :EVENT_TYPE_WORKFLOW_EXECUTION_UPDATE_ACCEPTED, 41
+      value :EVENT_TYPE_WORKFLOW_EXECUTION_UPDATE_REJECTED, 42
+      value :EVENT_TYPE_WORKFLOW_EXECUTION_UPDATE_COMPLETED, 43
       value :EVENT_TYPE_WORKFLOW_PROPERTIES_MODIFIED_EXTERNALLY, 44
       value :EVENT_TYPE_ACTIVITY_PROPERTIES_MODIFIED_EXTERNALLY, 45
       value :EVENT_TYPE_WORKFLOW_PROPERTIES_MODIFIED, 46

data/lib/gen/temporal/api/enums/v1/failed_cause_pb.rb

@@ -36,6 +36,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :WORKFLOW_TASK_FAILED_CAUSE_PENDING_ACTIVITIES_LIMIT_EXCEEDED, 27
       value :WORKFLOW_TASK_FAILED_CAUSE_PENDING_SIGNALS_LIMIT_EXCEEDED, 28
       value :WORKFLOW_TASK_FAILED_CAUSE_PENDING_REQUEST_CANCEL_LIMIT_EXCEEDED, 29
+      value :WORKFLOW_TASK_FAILED_CAUSE_BAD_UPDATE_WORKFLOW_EXECUTION_MESSAGE, 30
+      value :WORKFLOW_TASK_FAILED_CAUSE_UNHANDLED_UPDATE, 31
     end
     add_enum "temporal.api.enums.v1.StartChildWorkflowExecutionFailedCause" do
       value :START_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_UNSPECIFIED, 0

data/lib/gen/temporal/api/enums/v1/update_pb.rb

@@ -5,9 +5,11 @@ require 'google/protobuf'
 
 Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("temporal/api/enums/v1/update.proto", :syntax => :proto3) do
-    add_enum "temporal.api.enums.v1.WorkflowUpdateResultAccessStyle" do
-      value :WORKFLOW_UPDATE_RESULT_ACCESS_STYLE_UNSPECIFIED, 0
-      value :WORKFLOW_UPDATE_RESULT_ACCESS_STYLE_REQUIRE_INLINE, 1
+    add_enum "temporal.api.enums.v1.UpdateWorkflowExecutionLifecycleStage" do
+      value :UPDATE_WORKFLOW_EXECUTION_LIFECYCLE_STAGE_UNSPECIFIED, 0
+      value :UPDATE_WORKFLOW_EXECUTION_LIFECYCLE_STAGE_ADMITTED, 1
+      value :UPDATE_WORKFLOW_EXECUTION_LIFECYCLE_STAGE_ACCEPTED, 2
+      value :UPDATE_WORKFLOW_EXECUTION_LIFECYCLE_STAGE_COMPLETED, 3
     end
   end
 end
@@ -16,7 +18,7 @@ module Temporalio
   module Api
     module Enums
       module V1
-        WorkflowUpdateResultAccessStyle = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.enums.v1.WorkflowUpdateResultAccessStyle").enummodule
+        UpdateWorkflowExecutionLifecycleStage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.enums.v1.UpdateWorkflowExecutionLifecycleStage").enummodule
       end
     end
   end

data/lib/gen/temporal/api/history/v1/message_pb.rb

@@ -11,9 +11,10 @@ require 'temporal/api/enums/v1/failed_cause_pb'
 require 'temporal/api/enums/v1/workflow_pb'
 require 'temporal/api/common/v1/message_pb'
 require 'temporal/api/failure/v1/message_pb'
-require 'temporal/api/interaction/v1/message_pb'
 require 'temporal/api/taskqueue/v1/message_pb'
+require 'temporal/api/update/v1/message_pb'
 require 'temporal/api/workflow/v1/message_pb'
+require 'temporal/api/sdk/v1/task_complete_metadata_pb'
 
 Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("temporal/api/history/v1/message.proto", :syntax => :proto3) do
@@ -95,6 +96,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :identity, :string, 3
       optional :binary_checksum, :string, 4
       optional :worker_versioning_id, :message, 5, "temporal.api.taskqueue.v1.VersionId"
+      optional :sdk_metadata, :message, 6, "temporal.api.sdk.v1.WorkflowTaskCompletedMetadata"
+      optional :metering_metadata, :message, 13, "temporal.api.common.v1.MeteringMetadata"
     end
     add_message "temporal.api.history.v1.WorkflowTaskTimedOutEventAttributes" do
       optional :scheduled_event_id, :int64, 1
@@ -347,18 +350,6 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :initiated_event_id, :int64, 4
       optional :started_event_id, :int64, 5
     end
-    add_message "temporal.api.history.v1.WorkflowUpdateAcceptedEventAttributes" do
-      optional :meta, :message, 1, "temporal.api.interaction.v1.Meta"
-      optional :input, :message, 2, "temporal.api.interaction.v1.Input"
-    end
-    add_message "temporal.api.history.v1.WorkflowUpdateCompletedEventAttributes" do
-      optional :meta, :message, 1, "temporal.api.interaction.v1.Meta"
-      optional :output, :message, 2, "temporal.api.interaction.v1.Output"
-    end
-    add_message "temporal.api.history.v1.WorkflowUpdateRejectedEventAttributes" do
-      optional :meta, :message, 1, "temporal.api.interaction.v1.Meta"
-      optional :failure, :message, 2, "temporal.api.failure.v1.Failure"
-    end
     add_message "temporal.api.history.v1.WorkflowPropertiesModifiedExternallyEventAttributes" do
       optional :new_task_queue, :string, 1
       optional :new_workflow_task_timeout, :message, 2, "google.protobuf.Duration"
@@ -370,6 +361,23 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :scheduled_event_id, :int64, 1
       optional :new_retry_policy, :message, 2, "temporal.api.common.v1.RetryPolicy"
     end
+    add_message "temporal.api.history.v1.WorkflowExecutionUpdateAcceptedEventAttributes" do
+      optional :protocol_instance_id, :string, 1
+      optional :accepted_request_message_id, :string, 2
+      optional :accepted_request_sequencing_event_id, :int64, 3
+      optional :accepted_request, :message, 4, "temporal.api.update.v1.Request"
+    end
+    add_message "temporal.api.history.v1.WorkflowExecutionUpdateCompletedEventAttributes" do
+      optional :meta, :message, 1, "temporal.api.update.v1.Meta"
+      optional :outcome, :message, 2, "temporal.api.update.v1.Outcome"
+    end
+    add_message "temporal.api.history.v1.WorkflowExecutionUpdateRejectedEventAttributes" do
+      optional :protocol_instance_id, :string, 1
+      optional :rejected_request_message_id, :string, 2
+      optional :rejected_request_sequencing_event_id, :int64, 3
+      optional :rejected_request, :message, 4, "temporal.api.update.v1.Request"
+      optional :failure, :message, 5, "temporal.api.failure.v1.Failure"
+    end
     add_message "temporal.api.history.v1.HistoryEvent" do
       optional :event_id, :int64, 1
       optional :event_time, :message, 2, "google.protobuf.Timestamp"
@@ -418,9 +426,9 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
         optional :signal_external_workflow_execution_failed_event_attributes, :message, 43, "temporal.api.history.v1.SignalExternalWorkflowExecutionFailedEventAttributes"
         optional :external_workflow_execution_signaled_event_attributes, :message, 44, "temporal.api.history.v1.ExternalWorkflowExecutionSignaledEventAttributes"
         optional :upsert_workflow_search_attributes_event_attributes, :message, 45, "temporal.api.history.v1.UpsertWorkflowSearchAttributesEventAttributes"
-        optional :workflow_update_rejected_event_attributes, :message, 46, "temporal.api.history.v1.WorkflowUpdateRejectedEventAttributes"
-        optional :workflow_update_accepted_event_attributes, :message, 47, "temporal.api.history.v1.WorkflowUpdateAcceptedEventAttributes"
-        optional :workflow_update_completed_event_attributes, :message, 48, "temporal.api.history.v1.WorkflowUpdateCompletedEventAttributes"
+        optional :workflow_execution_update_accepted_event_attributes, :message, 46, "temporal.api.history.v1.WorkflowExecutionUpdateAcceptedEventAttributes"
+        optional :workflow_execution_update_rejected_event_attributes, :message, 47, "temporal.api.history.v1.WorkflowExecutionUpdateRejectedEventAttributes"
+        optional :workflow_execution_update_completed_event_attributes, :message, 48, "temporal.api.history.v1.WorkflowExecutionUpdateCompletedEventAttributes"
         optional :workflow_properties_modified_externally_event_attributes, :message, 49, "temporal.api.history.v1.WorkflowPropertiesModifiedExternallyEventAttributes"
         optional :activity_properties_modified_externally_event_attributes, :message, 50, "temporal.api.history.v1.ActivityPropertiesModifiedExternallyEventAttributes"
         optional :workflow_properties_modified_event_attributes, :message, 51, "temporal.api.history.v1.WorkflowPropertiesModifiedEventAttributes"
@@ -477,11 +485,11 @@ module Temporalio
         ChildWorkflowExecutionCanceledEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.ChildWorkflowExecutionCanceledEventAttributes").msgclass
         ChildWorkflowExecutionTimedOutEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.ChildWorkflowExecutionTimedOutEventAttributes").msgclass
         ChildWorkflowExecutionTerminatedEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.ChildWorkflowExecutionTerminatedEventAttributes").msgclass
-        WorkflowUpdateAcceptedEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.WorkflowUpdateAcceptedEventAttributes").msgclass
-        WorkflowUpdateCompletedEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.WorkflowUpdateCompletedEventAttributes").msgclass
-        WorkflowUpdateRejectedEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.WorkflowUpdateRejectedEventAttributes").msgclass
         WorkflowPropertiesModifiedExternallyEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.WorkflowPropertiesModifiedExternallyEventAttributes").msgclass
         ActivityPropertiesModifiedExternallyEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.ActivityPropertiesModifiedExternallyEventAttributes").msgclass
+        WorkflowExecutionUpdateAcceptedEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.WorkflowExecutionUpdateAcceptedEventAttributes").msgclass
+        WorkflowExecutionUpdateCompletedEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.WorkflowExecutionUpdateCompletedEventAttributes").msgclass
+        WorkflowExecutionUpdateRejectedEventAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.WorkflowExecutionUpdateRejectedEventAttributes").msgclass
         HistoryEvent = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.HistoryEvent").msgclass
         History = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.history.v1.History").msgclass
       end

data/lib/gen/temporal/api/namespace/v1/message_pb.rb

@@ -26,6 +26,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :history_archival_uri, :string, 4
       optional :visibility_archival_state, :enum, 5, "temporal.api.enums.v1.ArchivalState"
       optional :visibility_archival_uri, :string, 6
+      map :custom_search_attribute_aliases, :string, :string, 7
     end
     add_message "temporal.api.namespace.v1.BadBinaries" do
       map :binaries, :string, :message, 1, "temporal.api.namespace.v1.BadBinaryInfo"

data/lib/gen/temporal/api/operatorservice/v1/request_response_pb.rb

@@ -9,15 +9,18 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("temporal/api/operatorservice/v1/request_response.proto", :syntax => :proto3) do
     add_message "temporal.api.operatorservice.v1.AddSearchAttributesRequest" do
       map :search_attributes, :string, :enum, 1, "temporal.api.enums.v1.IndexedValueType"
+      optional :namespace, :string, 2
     end
     add_message "temporal.api.operatorservice.v1.AddSearchAttributesResponse" do
     end
     add_message "temporal.api.operatorservice.v1.RemoveSearchAttributesRequest" do
       repeated :search_attributes, :string, 1
+      optional :namespace, :string, 2
     end
     add_message "temporal.api.operatorservice.v1.RemoveSearchAttributesResponse" do
     end
     add_message "temporal.api.operatorservice.v1.ListSearchAttributesRequest" do
+      optional :namespace, :string, 1
     end
     add_message "temporal.api.operatorservice.v1.ListSearchAttributesResponse" do
       map :custom_attributes, :string, :enum, 1, "temporal.api.enums.v1.IndexedValueType"

data/lib/gen/temporal/api/protocol/v1/message_pb.rb

@@ -0,0 +1,30 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: temporal/api/protocol/v1/message.proto
+
+require 'google/protobuf'
+
+require 'google/protobuf/any_pb'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("temporal/api/protocol/v1/message.proto", :syntax => :proto3) do
+    add_message "temporal.api.protocol.v1.Message" do
+      optional :id, :string, 1
+      optional :protocol_instance_id, :string, 2
+      optional :body, :message, 5, "google.protobuf.Any"
+      oneof :sequencing_id do
+        optional :event_id, :int64, 3
+        optional :command_index, :int64, 4
+      end
+    end
+  end
+end
+
+module Temporalio
+  module Api
+    module Protocol
+      module V1
+        Message = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.protocol.v1.Message").msgclass
+      end
+    end
+  end
+end

data/lib/gen/temporal/api/sdk/v1/task_complete_metadata_pb.rb

@@ -0,0 +1,23 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: temporal/api/sdk/v1/task_complete_metadata.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("temporal/api/sdk/v1/task_complete_metadata.proto", :syntax => :proto3) do
+    add_message "temporal.api.sdk.v1.WorkflowTaskCompletedMetadata" do
+      repeated :core_used_flags, :uint32, 1
+      repeated :lang_used_flags, :uint32, 2
+    end
+  end
+end
+
+module Temporalio
+  module Api
+    module Sdk
+      module V1
+        WorkflowTaskCompletedMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.sdk.v1.WorkflowTaskCompletedMetadata").msgclass
+      end
+    end
+  end
+end

data/lib/gen/temporal/api/testservice/v1/request_response_pb.rb

@@ -0,0 +1,49 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: temporal/api/testservice/v1/request_response.proto
+
+require 'google/protobuf'
+
+require 'google/protobuf/duration_pb'
+require 'google/protobuf/timestamp_pb'
+require 'dependencies/gogoproto/gogo_pb'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("temporal/api/testservice/v1/request_response.proto", :syntax => :proto3) do
+    add_message "temporal.api.testservice.v1.LockTimeSkippingRequest" do
+    end
+    add_message "temporal.api.testservice.v1.LockTimeSkippingResponse" do
+    end
+    add_message "temporal.api.testservice.v1.UnlockTimeSkippingRequest" do
+    end
+    add_message "temporal.api.testservice.v1.UnlockTimeSkippingResponse" do
+    end
+    add_message "temporal.api.testservice.v1.SleepUntilRequest" do
+      optional :timestamp, :message, 1, "google.protobuf.Timestamp"
+    end
+    add_message "temporal.api.testservice.v1.SleepRequest" do
+      optional :duration, :message, 1, "google.protobuf.Duration"
+    end
+    add_message "temporal.api.testservice.v1.SleepResponse" do
+    end
+    add_message "temporal.api.testservice.v1.GetCurrentTimeResponse" do
+      optional :time, :message, 1, "google.protobuf.Timestamp"
+    end
+  end
+end
+
+module Temporalio
+  module Api
+    module TestService
+      module V1
+        LockTimeSkippingRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.LockTimeSkippingRequest").msgclass
+        LockTimeSkippingResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.LockTimeSkippingResponse").msgclass
+        UnlockTimeSkippingRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.UnlockTimeSkippingRequest").msgclass
+        UnlockTimeSkippingResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.UnlockTimeSkippingResponse").msgclass
+        SleepUntilRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.SleepUntilRequest").msgclass
+        SleepRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.SleepRequest").msgclass
+        SleepResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.SleepResponse").msgclass
+        GetCurrentTimeResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.testservice.v1.GetCurrentTimeResponse").msgclass
+      end
+    end
+  end
+end

data/lib/gen/temporal/api/testservice/v1/service_pb.rb

@@ -0,0 +1,21 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: temporal/api/testservice/v1/service.proto
+
+require 'google/protobuf'
+
+require 'temporal/api/testservice/v1/request_response_pb'
+require 'google/protobuf/empty_pb'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("temporal/api/testservice/v1/service.proto", :syntax => :proto3) do
+  end
+end
+
+module Temporalio
+  module Api
+    module TestService
+      module V1
+      end
+    end
+  end
+end

data/lib/gen/temporal/api/update/v1/message_pb.rb

@@ -0,0 +1,72 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: temporal/api/update/v1/message.proto
+
+require 'google/protobuf'
+
+require 'temporal/api/common/v1/message_pb'
+require 'temporal/api/enums/v1/update_pb'
+require 'temporal/api/failure/v1/message_pb'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("temporal/api/update/v1/message.proto", :syntax => :proto3) do
+    add_message "temporal.api.update.v1.WaitPolicy" do
+      optional :lifecycle_stage, :enum, 1, "temporal.api.enums.v1.UpdateWorkflowExecutionLifecycleStage"
+    end
+    add_message "temporal.api.update.v1.UpdateRef" do
+      optional :workflow_execution, :message, 1, "temporal.api.common.v1.WorkflowExecution"
+      optional :update_id, :string, 2
+    end
+    add_message "temporal.api.update.v1.Outcome" do
+      oneof :value do
+        optional :success, :message, 1, "temporal.api.common.v1.Payloads"
+        optional :failure, :message, 2, "temporal.api.failure.v1.Failure"
+      end
+    end
+    add_message "temporal.api.update.v1.Meta" do
+      optional :update_id, :string, 1
+      optional :identity, :string, 2
+    end
+    add_message "temporal.api.update.v1.Input" do
+      optional :header, :message, 1, "temporal.api.common.v1.Header"
+      optional :name, :string, 2
+      optional :args, :message, 3, "temporal.api.common.v1.Payloads"
+    end
+    add_message "temporal.api.update.v1.Request" do
+      optional :meta, :message, 1, "temporal.api.update.v1.Meta"
+      optional :input, :message, 2, "temporal.api.update.v1.Input"
+    end
+    add_message "temporal.api.update.v1.Rejection" do
+      optional :rejected_request_message_id, :string, 1
+      optional :rejected_request_sequencing_event_id, :int64, 2
+      optional :rejected_request, :message, 3, "temporal.api.update.v1.Request"
+      optional :failure, :message, 4, "temporal.api.failure.v1.Failure"
+    end
+    add_message "temporal.api.update.v1.Acceptance" do
+      optional :accepted_request_message_id, :string, 1
+      optional :accepted_request_sequencing_event_id, :int64, 2
+      optional :accepted_request, :message, 3, "temporal.api.update.v1.Request"
+    end
+    add_message "temporal.api.update.v1.Response" do
+      optional :meta, :message, 1, "temporal.api.update.v1.Meta"
+      optional :outcome, :message, 2, "temporal.api.update.v1.Outcome"
+    end
+  end
+end
+
+module Temporalio
+  module Api
+    module Update
+      module V1
+        WaitPolicy = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.WaitPolicy").msgclass
+        UpdateRef = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.UpdateRef").msgclass
+        Outcome = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.Outcome").msgclass
+        Meta = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.Meta").msgclass
+        Input = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.Input").msgclass
+        Request = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.Request").msgclass
+        Rejection = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.Rejection").msgclass
+        Acceptance = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.Acceptance").msgclass
+        Response = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("temporal.api.update.v1.Response").msgclass
+      end
+    end
+  end
+end