|
1 | 1 | using System.Collections.Generic;
|
2 | 2 | using System.Threading;
|
3 | 3 | using System.Threading.Tasks;
|
| 4 | +using System.Linq; |
4 | 5 | using IndicoV2.StrawberryShake;
|
5 |
| -using IndicoV2.V1Adapters.Workflows; |
6 | 6 | using IndicoV2.Workflows.Models;
|
7 | 7 |
|
8 | 8 | namespace IndicoV2.Workflows
|
9 | 9 | {
|
10 | 10 | public class WorkflowsClient : IWorkflowsClient
|
11 | 11 | {
|
12 |
| - private readonly WorkflowsV1ClientAdapter _legacy; |
13 | 12 | private readonly IndicoStrawberryShakeClient _strawberryShake;
|
| 13 | + private readonly IndicoClient _indicoClient; |
14 | 14 |
|
15 | 15 | public WorkflowsClient(IndicoClient indicoClient)
|
16 | 16 | {
|
17 |
| - _legacy = new WorkflowsV1ClientAdapter(indicoClient.LegacyClient); |
| 17 | + _indicoClient = indicoClient; |
18 | 18 | _strawberryShake = indicoClient.IndicoStrawberryShakeClient;
|
19 | 19 | }
|
20 | 20 |
|
21 |
| - public Task<IEnumerable<IWorkflow>> ListAsync(int dataSetId, CancellationToken cancellationToken = default) => |
22 |
| - _legacy.ListAsync(dataSetId, cancellationToken); |
| 21 | + public async Task<IEnumerable<IWorkflow>> ListAsync(int dataSetId, CancellationToken cancellationToken = default) |
| 22 | + { |
| 23 | + var result = await _strawberryShake.Workflows().ListAsync(dataSetId, cancellationToken); |
| 24 | + return result.Workflows.Select(x => ToWorkflowFromSs(x)).ToList(); |
| 25 | + } |
23 | 26 |
|
24 |
| - public Task<IEnumerable<IWorkflow>> |
25 |
| - ListAsync(int[] dataSetIds, CancellationToken cancellationToken = default) => |
26 |
| - _legacy.ListAsync(dataSetIds, cancellationToken); |
| 27 | + public async Task<IEnumerable<IWorkflow>>ListAsync(int[] dataSetIds, CancellationToken cancellationToken = default) |
| 28 | + { |
| 29 | + var result = await _strawberryShake.Workflows().ListAsync(dataSetIds, cancellationToken); |
| 30 | + return result.Workflows.Select(x => ToWorkflowFromSs(x)).ToList(); |
| 31 | + } |
27 | 32 |
|
28 | 33 | public Task<IWorkflowAddDataResult> AddDataAsync(int workflowId, CancellationToken cancellationToken) =>
|
29 | 34 | _strawberryShake.Workflows().AddData(workflowId, cancellationToken);
|
30 | 35 |
|
31 | 36 | public Task<WorkflowStatus> GetStatusAsync(int workflowId, CancellationToken cancellationToken) =>
|
32 | 37 | _strawberryShake.Workflows().GetStatus(workflowId, cancellationToken);
|
| 38 | + |
| 39 | + private IWorkflow ToWorkflowFromSs(IListWorkflows_Workflows_Workflows workflow) => new WorkflowSs(workflow); |
33 | 40 | }
|
34 | 41 | }
|
0 commit comments