From 46259b30141a7047a673c9909255e89c16239ee5 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipkeschoorstra@outlook.com>
Date: Mon, 17 May 2021 19:21:15 +0200
Subject: [PATCH] Update YesSQL provider with support for
 WorkflowInstanceIdsSpecification

---
 .../Stores/YesSqlWorkflowInstanceStore.cs                       | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs b/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs
index 6ccef8304..2b25ab829 100644
--- a/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs
+++ b/src/persistence/Elsa.Persistence.YesSql/Stores/YesSqlWorkflowInstanceStore.cs
@@ -11,6 +11,7 @@ using Elsa.Persistence.YesSql.Services;
 using Microsoft.Extensions.Logging;
 using NodaTime;
 using YesSql;
+using YesSql.Services;
 using IIdGenerator = Elsa.Services.IIdGenerator;
 
 namespace Elsa.Persistence.YesSql.Stores
@@ -31,6 +32,7 @@ namespace Elsa.Persistence.YesSql.Stores
             {
                 EntityIdSpecification<WorkflowInstance> spec => Query<WorkflowInstanceIndex>(session, x => x.InstanceId == spec.Id),
                 WorkflowInstanceIdSpecification spec => Query<WorkflowInstanceIndex>(session, x => x.InstanceId == spec.Id),
+                WorkflowInstanceIdsSpecification spec => Query<WorkflowInstanceIndex>(session, x => x.InstanceId.IsIn(spec.WorkflowInstanceIds)),
                 BlockingActivityTypeSpecification spec => Query<WorkflowInstanceBlockingActivitiesIndex>(session, x => x.ActivityType == spec.ActivityType),
                 _ => AutoMapSpecification<WorkflowInstanceIndex>(session, specification)
             };