From 19261d447cda072a49961089fe2c5719e2514451 Mon Sep 17 00:00:00 2001
From: Leonid Stryuk <leo.stryuk@gmail.com>
Date: Fri, 20 Dec 2024 15:55:27 +0100
Subject: [PATCH] Fixed contains in multple entitites iterable

---
 .../iterate/link/OMultipleEntitiesIterable.kt | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/entity-store/src/main/kotlin/jetbrains/exodus/entitystore/orientdb/iterate/link/OMultipleEntitiesIterable.kt b/entity-store/src/main/kotlin/jetbrains/exodus/entitystore/orientdb/iterate/link/OMultipleEntitiesIterable.kt
index 6dbf0e4c8..42135c6d9 100644
--- a/entity-store/src/main/kotlin/jetbrains/exodus/entitystore/orientdb/iterate/link/OMultipleEntitiesIterable.kt
+++ b/entity-store/src/main/kotlin/jetbrains/exodus/entitystore/orientdb/iterate/link/OMultipleEntitiesIterable.kt
@@ -16,6 +16,7 @@
 package jetbrains.exodus.entitystore.orientdb.iterate.link
 
 import jetbrains.exodus.entitystore.Entity
+import jetbrains.exodus.entitystore.EntityIterable
 import jetbrains.exodus.entitystore.orientdb.OEntityId
 import jetbrains.exodus.entitystore.orientdb.OStoreTransaction
 import jetbrains.exodus.entitystore.orientdb.iterate.OEntityIterableBase
@@ -26,4 +27,32 @@ class OMultipleEntitiesIterable(tx: OStoreTransaction, val entities: List<Entity
     override fun query(): OSelect {
         return ORecordIdSelect(entities.map { (it.id as OEntityId).asOId() })
     }
+
+    override fun contains(entity: Entity): Boolean {
+        return entities.contains(entity)
+    }
+
+    override fun count(): Long {
+        return entities.size.toLong()
+    }
+
+    override fun getRoughCount(): Long {
+        return entities.size.toLong()
+    }
+
+    override fun getRoughSize(): Long {
+        return entities.size.toLong()
+    }
+
+    override fun size(): Long {
+        return entities.size.toLong()
+    }
+
+    override fun skip(number: Int): EntityIterable {
+        return OMultipleEntitiesIterable(transaction as OStoreTransaction, entities.drop(number) )
+    }
+
+    override fun take(number: Int): EntityIterable {
+        return OMultipleEntitiesIterable(transaction as OStoreTransaction, entities.take(number) )
+    }
 }