use super::{storage_sql_error, Engine, ExecutionContext, SQLError};
impl Engine {
pub(crate) fn snapshot_context(
&self,
table: &str,
) -> Result<Option<ExecutionContext>, SQLError> {
self.snapshot_context_with_documents(table, None)
}
pub(crate) fn snapshot_context_with_document_store(
&self,
table: &str,
documents: std::sync::Arc<dyn uqa_storage::DocumentStore>,
) -> Result<Option<ExecutionContext>, SQLError> {
self.snapshot_context_with_documents(table, Some(documents))
}
fn snapshot_context_with_documents(
&self,
table: &str,
documents: Option<std::sync::Arc<dyn uqa_storage::DocumentStore>>,
) -> Result<Option<ExecutionContext>, SQLError> {
let Some(t) = self
.try_query_table(table)
.map_err(|error| storage_sql_error("resolve snapshot table", error))?
else {
return Ok(None);
};
let control = self.query_retention_control()?;
let inv = t
.inverted_index
.read()
.snapshot_with_control(&control)
.map_err(|error| storage_sql_error("snapshot inverted index", error))?;
let documents = match documents {
Some(documents) => documents,
None => t
.document_store
.read()
.snapshot()
.map_err(|error| storage_sql_error("snapshot document store", error))?,
};
let read = self.serializable_table_read(table)?;
let columns = t.columns.snapshot();
let mut ctx = ExecutionContext::new()
.with_inverted_index(uqa_execution::serializable::text::observe_snapshot(
read.as_ref(),
columns.clone(),
inv,
))
.with_document_store(documents);
for (field, idx) in t.vector_indexes.read().iter() {
ctx = ctx.with_vector_index(
field.clone(),
uqa_execution::serializable::vector::observe_snapshot(
read.as_ref(),
&columns,
field,
idx.snapshot_with_control(&control)
.map_err(|error| storage_sql_error("snapshot vector index", error))?,
)
.map_err(|error| storage_sql_error("retain vector search participant", error))?,
);
}
Ok(Some(ctx))
}
}