aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-02-09 10:14:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

add new Cursor type

Browse Source
This commit is contained in:
Nikita Sivukhin
2025-10-27 17:59:14 +04:00
parent 5af10e6ccb
commit b994e2cbd8
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
core/vdbe/builder.rs
View File

@@ -7,6 +7,7 @@ use tracing::{instrument, Level};
use turso_parser::ast::{self, TableInternalId};
use crate::{
index_method::IndexMethodAttachment,
numeric::Numeric,
parameters::Parameters,
schema::{BTreeTable, Index, PseudoCursorType, Schema, Table},
@@ -134,6 +135,7 @@ pub struct ProgramBuilder {
pub enum CursorType {
BTreeTable(Arc<BTreeTable>),
BTreeIndex(Arc<Index>),
IndexMethod(Arc<dyn IndexMethodAttachment>),
Pseudo(PseudoCursorType),
Sorter,
VirtualTable(Arc<VirtualTable>),
@@ -332,6 +334,20 @@ impl ProgramBuilder {
}
}
/// allocate proper cursor for the given index (either [CursorType::BTreeIndex] or [CursorType::IndexMethod])
pub fn alloc_cursor_index(
&mut self,
key: Option<CursorKey>,
index: &Arc<Index>,
) -> crate::Result<usize> {
let module = index.index_method.as_ref();
if module.is_some_and(|m| !m.definition().backing_btree) {
let module = module.unwrap().clone();
return Ok(self._alloc_cursor_id(key, CursorType::IndexMethod(module)));
}
Ok(self._alloc_cursor_id(key, CursorType::BTreeIndex(index.clone())))
}
pub fn alloc_cursor_id(&mut self, cursor_type: CursorType) -> usize {
self._alloc_cursor_id(None, cursor_type)
}