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

core/mvcc/cursor: add next rowid lock

Browse Source
This commit is contained in:
Pere Diaz Bou
2025-10-21 13:41:10 +02:00
parent 0fee588bca
commit 3f41a092f2
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
core/mvcc/cursor.rs
View File

@@ -29,6 +29,7 @@ pub struct MvccLazyCursor<Clock: LogicalClock> {
_btree_cursor: Box<dyn CursorTrait>,
null_flag: bool,
record_cursor: RefCell<RecordCursor>,
next_rowid_lock: Arc<RwLock<()>>,
}
impl<Clock: LogicalClock + 'static> MvccLazyCursor<Clock> {
@@ -54,6 +55,8 @@ impl<Clock: LogicalClock + 'static> MvccLazyCursor<Clock> {
_btree_cursor: btree_cursor,
null_flag: false,
record_cursor: RefCell::new(RecordCursor::new()),
next_rowid_lock: Arc::new(RwLock::new(())),
})
}
pub fn current_row(&self) -> Result<Option<Row>> {
@@ -80,6 +83,9 @@ impl<Clock: LogicalClock + 'static> MvccLazyCursor<Clock> {
}
pub fn get_next_rowid(&mut self) -> i64 {
// lock so we don't get same two rowids
let lock = self.next_rowid_lock.clone();
let _lock = lock.write();
let _ = self.last();
match *self.current_pos.borrow() {
CursorPosition::Loaded(id) => id.row_id + 1,
@@ -147,6 +153,7 @@ impl<Clock: LogicalClock + 'static> CursorTrait for MvccLazyCursor<Clock> {
}
};
self.current_pos.replace(new_position);
self.invalidate_record();
Ok(IOResult::Done(matches!(
self.get_current_pos(),