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

Add a bit of abstraction for creating EQ cursorcontexts

Browse Source
This commit is contained in:
Jussi Saurio
2025-08-18 13:12:40 +03:00
parent 3eb89982ba
commit 9abc63d853
1 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
core/storage/btree.rs
View File

@@ -405,6 +405,24 @@ pub struct CursorContext {
pub seek_op: SeekOp,
}
impl CursorContext {
fn seek_eq_only(key: &BTreeKey<'_>) -> Self {
Self {
key: key.into(),
seek_op: SeekOp::GE { eq_only: true },
}
}
}
impl From<&BTreeKey<'_>> for CursorContextKey {
fn from(key: &BTreeKey<'_>) -> Self {
match key {
BTreeKey::TableRowId((rowid, _)) => CursorContextKey::TableRowId(*rowid),
BTreeKey::IndexKey(record) => CursorContextKey::IndexKeyRowId((*record).clone()),
}
}
}
/// In the future, we may expand these general validity states
#[derive(Debug, PartialEq, Eq)]
pub enum CursorValidState {
@@ -2235,16 +2253,7 @@ impl BTreeCursor {
*write_state = WriteState::Balancing;
assert!(self.balance_state.sub_state == BalanceSubState::Start, "There should be no balancing operation in progress when insert state is {:?}, got: {:?}", self.state, self.balance_state.sub_state);
// If we balance, we must save the cursor position and seek to it later.
self.save_context(match bkey {
BTreeKey::TableRowId(rowid) => CursorContext {
key: CursorContextKey::TableRowId(rowid.0),
seek_op: SeekOp::GE { eq_only: true },
},
BTreeKey::IndexKey(record) => CursorContext {
key: CursorContextKey::IndexKeyRowId((*record).clone()),
seek_op: SeekOp::GE { eq_only: true },
},
});
self.save_context(CursorContext::seek_eq_only(bkey));
} else {
*write_state = WriteState::Finish;
}
@@ -2292,16 +2301,7 @@ impl BTreeCursor {
*write_state = WriteState::Balancing;
assert!(self.balance_state.sub_state == BalanceSubState::Start, "There should be no balancing operation in progress when overwrite state is {:?}, got: {:?}", self.state, self.balance_state.sub_state);
// If we balance, we must save the cursor position and seek to it later.
self.save_context(match bkey {
BTreeKey::TableRowId(rowid) => CursorContext {
key: CursorContextKey::TableRowId(rowid.0),
seek_op: SeekOp::GE { eq_only: true },
},
BTreeKey::IndexKey(record) => CursorContext {
key: CursorContextKey::IndexKeyRowId((*record).clone()),
seek_op: SeekOp::GE { eq_only: true },
},
});
self.save_context(CursorContext::seek_eq_only(bkey));
} else {
*write_state = WriteState::Finish;
}