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

simplify cursor.exists() by using seek()

Browse Source
This commit is contained in:
Jussi Saurio
2025-06-06 13:29:09 +03:00
parent 844461d20b
commit 843eb18daf
1 changed files with 5 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
core/storage/btree.rs
View File

@@ -4510,38 +4510,11 @@ impl BTreeCursor {
Value::Integer(i) => i,
_ => unreachable!("btree tables are indexed by integers!"),
};
let _ = return_if_io!(
self.move_to(SeekKey::TableRowId(*int_key), SeekOp::GE { eq_only: true })
);
self.seek_state = CursorSeekState::Start;
let page = self.stack.top();
// TODO(pere): request load
return_if_locked_maybe_load!(self.pager, page);
let page = page.get();
let contents = page.get().contents.as_ref().unwrap();
// find cell
let int_key = match key {
Value::Integer(i) => *i,
_ => unreachable!("btree tables are indexed by integers!"),
};
let cell_idx =
return_if_io!(self.find_cell(contents, &BTreeKey::new_table_rowid(int_key, None)));
if cell_idx >= contents.cell_count() {
Ok(CursorResult::Ok(false))
} else {
let equals = match &contents.cell_get(
cell_idx,
payload_overflow_threshold_max(contents.page_type(), self.usable_space() as u16),
payload_overflow_threshold_min(contents.page_type(), self.usable_space() as u16),
self.usable_space(),
)? {
BTreeCell::TableLeafCell(l) => l._rowid == int_key,
_ => unreachable!(),
};
Ok(CursorResult::Ok(equals))
}
let has_record =
return_if_io!(self.seek(SeekKey::TableRowId(*int_key), SeekOp::GE { eq_only: true }));
self.has_record.set(has_record);
self.invalidate_record();
Ok(CursorResult::Ok(has_record))
}
/// Clear the overflow pages linked to a specific page provided by the leaf cell