use TQueryData for single()

andreakarasho · Jan 27, 2025 · 6b8b3d8 · 6b8b3d8
1 parent 0c509dc
commit 6b8b3d8
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/Bevy.cs b/src/Bevy.cs
@@ -417,11 +417,14 @@ public static ISystemParam<World> Generate(World arg)
 
 	public TQueryData Iter(EcsID id) => TQueryData.CreateIterator(_query.Iter(id));
 
-	public ref T Single<T>() where T : struct
-		=> ref _query.Single<T>();
-
-	public EntityView Single()
-		=> _query.Single();
+	public TQueryData Single()
+	{
+		EcsAssert.Panic(_query.Count() == 1, "'Single' must match one and only one entity.");
+		var enumerator = GetEnumerator();
+		var ok = enumerator.MoveNext();
+		EcsAssert.Panic(ok, "'Single' is not matching any entity.");
+		return enumerator;
+	}
 
 	public int Count()
 		=> _query.Count();