Fix type of exists and count

DATP4G6 · Apr 25, 2024 · 136f5b9 · 136f5b9
1 parent abac73b
commit 136f5b9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/SocietalConstructionTool/Runtime/IQueryHandler.cs b/SocietalConstructionTool/Runtime/IQueryHandler.cs
@@ -2,7 +2,7 @@ namespace Sct.Runtime
 {
     public interface IQueryHandler
     {
-        public int Count(IRuntimeContext ctx, IQueryPredicate predicate);
-        public bool Exists(IRuntimeContext ctx, IQueryPredicate predicate);
+        public long Count(IRuntimeContext ctx, IQueryPredicate predicate);
+        public long Exists(IRuntimeContext ctx, IQueryPredicate predicate);
     }
 }
diff --git a/SocietalConstructionTool/Runtime/QueryHandler.cs b/SocietalConstructionTool/Runtime/QueryHandler.cs
@@ -4,7 +4,7 @@ public class QueryHandler(IEnumerable<BaseAgent> agents) : IQueryHandler
     {
         private IEnumerable<BaseAgent> Filter(IQueryPredicate predicate) => agents.Where(predicate.Test);
 
-        public int Count(IRuntimeContext ctx, IQueryPredicate predicate) => Filter(predicate).Count();
-        public bool Exists(IRuntimeContext ctx, IQueryPredicate predicate) => Filter(predicate).Any();
+        public long Count(IRuntimeContext ctx, IQueryPredicate predicate) => Filter(predicate).Count();
+        public long Exists(IRuntimeContext ctx, IQueryPredicate predicate) => Filter(predicate).Any() ? 1 : 0;
     }
 }