aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-02-19 23:15:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix/btree: balance and seek after overwritten cell overflows

Browse Source
This commit is contained in:
Jussi Saurio
2025-06-24 11:08:22 +03:00
parent a99833ba8d
commit 5878724d0e
1 changed files with 32 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
core/storage/btree.rs
View File

@@ -2128,12 +2128,22 @@ impl BTreeCursor {
match cell {
BTreeCell::TableLeafCell(tbl_leaf) => {
if tbl_leaf._rowid == bkey.to_rowid() {
tracing::debug!("found exact match with cell_idx={cell_idx}, overwriting");
tracing::debug!("TableLeafCell: found exact match with cell_idx={cell_idx}, overwriting");
self.overwrite_cell(page.clone(), cell_idx, record)?;
self.state
let write_info = self
.state
.mut_write_info()
.expect("expected write info")
.state = WriteState::Finish;
.expect("expected write info");
if page.get().get_contents().overflow_cells.is_empty() {
write_info.state = WriteState::Finish;
return Ok(CursorResult::Ok(()));
} else {
write_info.state = WriteState::BalanceStart;
// If we balance, we must save the cursor position and seek to it later.
// FIXME: we shouldn't have both DeleteState::SeekAfterBalancing and
// save_context()/restore/context(), they are practically the same thing.
self.save_context(CursorContext::TableRowId(bkey.to_rowid()));
}
continue;
}
}
@@ -2149,13 +2159,23 @@ impl BTreeCursor {
&self.collations,
);
if cmp == Ordering::Equal {
tracing::debug!("found exact match with cell_idx={cell_idx}, overwriting");
tracing::debug!("IndexLeafCell: found exact match with cell_idx={cell_idx}, overwriting");
self.has_record.set(true);
self.overwrite_cell(page.clone(), cell_idx, record)?;
self.state
let write_info = self
.state
.mut_write_info()
.expect("expected write info")
.state = WriteState::Finish;
.expect("expected write info");
if page.get().get_contents().overflow_cells.is_empty() {
write_info.state = WriteState::Finish;
return Ok(CursorResult::Ok(()));
} else {
write_info.state = WriteState::BalanceStart;
// If we balance, we must save the cursor position and seek to it later.
// FIXME: we shouldn't have both DeleteState::SeekAfterBalancing and
// save_context()/restore/context(), they are practically the same thing.
self.save_context(CursorContext::IndexKeyRowId((*record).clone()));
}
continue;
}
}
@@ -4023,7 +4043,10 @@ impl BTreeCursor {
_rowid, _payload, ..
}) = cell
else {
unreachable!("unexpected page_type");
unreachable!(
"BTreeCursor::rowid(): unexpected page_type: {:?}",
page_type
);
};
Ok(CursorResult::Ok(Some(_rowid)))
} else {