Skip to content

Commit fc1e49e

Browse files
committed
Fix missing client methods
1 parent fb6ad3a commit fc1e49e

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

src/Temporalio/Bridge/src/client.rs

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,9 @@ use std::str::FromStr;
99
use std::time::Duration;
1010
use 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
};
1516
use tonic::metadata::MetadataKey;
1617
use 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

Comments
 (0)