add count with predicate to family

Quillraven · Dec 22, 2024 · c694660 · c694660
1 parent deb414e
commit c694660
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/commonMain/kotlin/com/github/quillraven/fleks/family.kt b/src/commonMain/kotlin/com/github/quillraven/fleks/family.kt
@@ -254,6 +254,11 @@ data class Family(
      */
     fun none(predicate: (Entity) -> Boolean): Boolean = mutableEntities.none(predicate)
 
+    /**
+     * Returns the number of [entities][Entity] matching the given [predicate].
+     */
+    fun count(predicate: (Entity) -> Boolean): Int = mutableEntities.count(predicate)
+
     /**
      * Returns a [Map] containing key-value pairs provided by the [transform] function applied to
      * each [entity][Entity] of the family.

diff --git a/src/commonTest/kotlin/com/github/quillraven/fleks/FamilyBagFunctionsTest.kt b/src/commonTest/kotlin/com/github/quillraven/fleks/FamilyBagFunctionsTest.kt
@@ -48,6 +48,11 @@ class FamilyBagFunctionsTest {
         assertFalse(testFamily.none { it.id == 0 })
     }
 
+    @Test
+    fun testCount() {
+        assertEquals(1, testFamily.count { it.id == 0 })
+    }
+
     @Test
     fun testAssociate() {
         val expected = mapOf(testEntity1 to 0, testEntity2 to 1)