@@ -9,8 +9,9 @@ use std::str::FromStr;
99use std:: time:: Duration ;
1010use temporal_client:: {
1111 ClientKeepAliveConfig , ClientOptions as CoreClientOptions , ClientOptionsBuilder ,
12- ClientTlsConfig , CloudService , ConfiguredClient , HealthService , HttpConnectProxyOptions , OperatorService , RetryClient ,
13- RetryConfig , TemporalServiceClientWithMetrics , TestService , TlsConfig , WorkflowService ,
12+ ClientTlsConfig , CloudService , ConfiguredClient , HealthService , HttpConnectProxyOptions ,
13+ OperatorService , RetryClient , RetryConfig , TemporalServiceClientWithMetrics , TestService ,
14+ TlsConfig , WorkflowService ,
1415} ;
1516use tonic:: metadata:: MetadataKey ;
1617use url:: Url ;
@@ -335,6 +336,7 @@ async fn call_workflow_service(
335336 "ListTaskQueuePartitions" => rpc_call ! ( client, call, list_task_queue_partitions) ,
336337 "ListWorkflowExecutions" => rpc_call ! ( client, call, list_workflow_executions) ,
337338 "PatchSchedule" => rpc_call ! ( client, call, patch_schedule) ,
339+ "PauseActivityById" => rpc_call ! ( client, call, pause_activity_by_id) ,
338340 "PollActivityTaskQueue" => rpc_call ! ( client, call, poll_activity_task_queue) ,
339341 "PollNexusTaskQueue" => rpc_call ! ( client, call, poll_nexus_task_queue) ,
340342 "PollWorkflowExecutionUpdate" => rpc_call ! ( client, call, poll_workflow_execution_update) ,
@@ -348,6 +350,7 @@ async fn call_workflow_service(
348350 "RequestCancelWorkflowExecution" => {
349351 rpc_call ! ( client, call, request_cancel_workflow_execution)
350352 }
353+ "ResetActivityById" => rpc_call ! ( client, call, reset_activity_by_id) ,
351354 "ResetStickyTaskQueue" => rpc_call ! ( client, call, reset_sticky_task_queue) ,
352355 "ResetWorkflowExecution" => rpc_call ! ( client, call, reset_workflow_execution) ,
353356 "RespondActivityTaskCanceled" => rpc_call ! ( client, call, respond_activity_task_canceled) ,
@@ -368,6 +371,7 @@ async fn call_workflow_service(
368371 "RespondWorkflowTaskCompleted" => rpc_call ! ( client, call, respond_workflow_task_completed) ,
369372 "RespondWorkflowTaskFailed" => rpc_call ! ( client, call, respond_workflow_task_failed) ,
370373 "ScanWorkflowExecutions" => rpc_call ! ( client, call, scan_workflow_executions) ,
374+ "ShutdownWorker" => rpc_call ! ( client, call, shutdown_worker) ,
371375 "SignalWithStartWorkflowExecution" => {
372376 rpc_call ! ( client, call, signal_with_start_workflow_execution)
373377 }
@@ -376,6 +380,12 @@ async fn call_workflow_service(
376380 "StartBatchOperation" => rpc_call ! ( client, call, start_batch_operation) ,
377381 "StopBatchOperation" => rpc_call ! ( client, call, stop_batch_operation) ,
378382 "TerminateWorkflowExecution" => rpc_call ! ( client, call, terminate_workflow_execution) ,
383+ "UnpauseActivityById" => {
384+ rpc_call_on_trait ! ( client, call, WorkflowService , unpause_activity_by_id)
385+ }
386+ "UpdateActivityOptionsById" => {
387+ rpc_call_on_trait ! ( client, call, WorkflowService , update_activity_options_by_id)
388+ }
379389 "UpdateNamespace" => rpc_call_on_trait ! ( client, call, WorkflowService , update_namespace) ,
380390 "UpdateSchedule" => rpc_call ! ( client, call, update_schedule) ,
381391 "UpdateWorkerVersioningRules" => rpc_call ! ( client, call, update_worker_versioning_rules) ,
@@ -526,7 +536,9 @@ impl TryFrom<&ClientOptions> for CoreClientOptions {
526536 Some ( opts. metadata . to_string_map_on_newlines ( ) )
527537 } )
528538 . api_key ( opts. api_key . to_option_string ( ) )
529- . http_connect_proxy ( unsafe { opts. http_connect_proxy_options . as_ref ( ) } . map ( Into :: into) ) ;
539+ . http_connect_proxy (
540+ unsafe { opts. http_connect_proxy_options . as_ref ( ) } . map ( Into :: into) ,
541+ ) ;
530542 if let Some ( tls_config) = unsafe { opts. tls_options . as_ref ( ) } {
531543 opts_builder. tls_cfg ( tls_config. try_into ( ) ?) ;
532544 }
0 commit comments