Added special 'empty' retriever that just looks up a retrievable base…

…d on an id
vitrivr · Dec 9, 2023 · 7af00d5 · 7af00d5
1 parent 8ec62fd
commit 7af00d5
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 2 deletions.
diff --git a/vitrivr-engine-query/src/main/kotlin/org/vitrivr/engine/query/execution/RetrievalRuntime.kt b/vitrivr-engine-query/src/main/kotlin/org/vitrivr/engine/query/execution/RetrievalRuntime.kt
@@ -33,11 +33,24 @@ class RetrievalRuntime {
                 OperatorType.RETRIEVER -> {
                     operationDescription as RetrieverDescription
 
+                    val inputDescription = informationNeed.inputs[operationDescription.input]
+                        ?: throw IllegalArgumentException("Input '${operationDescription.input}' for operation '$operationName' not found")
+
+                    if (operationDescription.field.isEmpty()) { //special case, handle pass-through
+
+                        if(inputDescription.type != InputType.ID) {
+                            throw IllegalArgumentException("Only inputs of type ID are supported for direct retrievable lookup")
+                        }
+
+                        operators[operationName] = RetrievedLookup(
+                            schema.connection.getRetrievableReader(), listOf(UUID.fromString((inputDescription as RetrievableIdInputData).id))
+                        )
+
+                    }
+
                     val field = schema[operationDescription.field]
                         ?: throw IllegalArgumentException("Retriever '${operationDescription.field}' not defined in schema")
 
-                    val inputDescription = informationNeed.inputs[operationDescription.input]
-                        ?: throw IllegalArgumentException("Input '${operationDescription.input}' for operation '$operationName' not found")
 
                     val retriever = when (inputDescription.type) {
                         InputType.VECTOR -> {

diff --git a/vitrivr-engine-query/src/main/kotlin/org/vitrivr/engine/query/execution/RetrievedLookup.kt b/vitrivr-engine-query/src/main/kotlin/org/vitrivr/engine/query/execution/RetrievedLookup.kt
@@ -0,0 +1,22 @@
+package org.vitrivr.engine.query.execution
+
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.flow.asFlow
+import org.vitrivr.engine.core.database.retrievable.RetrievableReader
+import org.vitrivr.engine.core.model.retrievable.RetrievableId
+import org.vitrivr.engine.core.model.retrievable.Retrieved
+import org.vitrivr.engine.core.operators.Operator
+
+class RetrievedLookup(
+    private val retrievableReader: RetrievableReader,
+    private val ids: Collection<RetrievableId>
+) : Operator.Nullary<Retrieved> {
+    override fun toFlow(scope: CoroutineScope): Flow<Retrieved> {
+
+        return this.retrievableReader.getAll(ids).map {
+            Retrieved.Default(it.id, it.type, false)
+        }.asFlow()
+
+    }
+}