Skip to content

Commit 3fa86b8

Browse files
committed
update to align with latest core api
1 parent 655a941 commit 3fa86b8

File tree

1 file changed

+19
-5
lines changed

1 file changed

+19
-5
lines changed

src/Worker/Core/Shims/TaskOrchestrationContextWrapper.cs

Lines changed: 19 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -150,24 +150,38 @@ public override async Task<T> CallActivityAsync<T>(
150150
// TODO: Cancellation (https://github.com/microsoft/durabletask-dotnet/issues/7)
151151
if (options?.Retry?.Policy is RetryPolicy policy)
152152
{
153-
return await this.innerContext.ScheduleWithRetry<T>(
153+
return await this.innerContext.ScheduleTask<T>(
154154
name.Name,
155155
name.Version,
156-
policy.ToDurableTaskCoreRetryOptions(),
157-
tags: tags,
156+
options: ScheduleTaskOptions.CreateBuilder()
157+
.WithRetryOptions(policy.ToDurableTaskCoreRetryOptions())
158+
.WithTags(tags)
159+
.Build(),
158160
parameters: input);
159161
}
160162
else if (options?.Retry?.Handler is AsyncRetryHandler handler)
161163
{
162164
return await this.InvokeWithCustomRetryHandler(
163-
() => this.innerContext.ScheduleTask<T>(name.Name, name.Version, tags: tags, parameters: input),
165+
() => this.innerContext.ScheduleTask<T>(
166+
name.Name,
167+
name.Version,
168+
options: ScheduleTaskOptions.CreateBuilder()
169+
.WithTags(tags)
170+
.Build(),
171+
parameters: input),
164172
name.Name,
165173
handler,
166174
default);
167175
}
168176
else
169177
{
170-
return await this.innerContext.ScheduleTask<T>(name.Name, name.Version, tags: tags, parameters: input);
178+
return await this.innerContext.ScheduleTask<T>(
179+
name.Name,
180+
name.Version,
181+
options: ScheduleTaskOptions.CreateBuilder()
182+
.WithTags(tags)
183+
.Build(),
184+
parameters: input);
171185
}
172186
}
173187
catch (global::DurableTask.Core.Exceptions.TaskFailedException e)

0 commit comments

Comments
 (0)