Add file_position to EntryDetailed

This commit is contained in:
Yuriy Dupyn 2024-02-03 22:54:55 +01:00
parent ff378b1dd8
commit 0f98903759
2 changed files with 6 additions and 4 deletions

View file

@ -121,12 +121,13 @@ pub trait CursorWithStoreHeader<T>: PrimitiveCursor<T> {
async fn next(&mut self) -> Result<Option<EntryDetailed<T>>> async fn next(&mut self) -> Result<Option<EntryDetailed<T>>>
where T: Decode where T: Decode
{ {
let file_position = self.current_file_position().await?;
let Some(entry_header) = self.next_entry_header().await? else { return Ok(None) }; let Some(entry_header) = self.next_entry_header().await? else { return Ok(None) };
let mut data_bytes: Vec<u8> = vec![0; entry_header.size_of_data()]; let mut data_bytes: Vec<u8> = vec![0; entry_header.size_of_data()];
self.read_bytes(&mut data_bytes).await?; self.read_bytes(&mut data_bytes).await?;
let entry: EntryDetailed<T> = let entry: EntryDetailed<T> =
EntryDetailed::decode(entry_header, self.header().number_of_columns, &mut data_bytes)?; EntryDetailed::decode(entry_header, file_position, self.header().number_of_columns, &mut data_bytes)?;
Ok(Some(entry)) Ok(Some(entry))
} }

View file

@ -1,7 +1,7 @@
use bincode::{Decode, Encode}; use bincode::{Decode, Encode};
use crate::binary_coding::{encode_sequence, encode_sequence_with_sizes, decode_sequence}; use crate::binary_coding::{encode_sequence, encode_sequence_with_sizes, decode_sequence};
use crate::storage_engine::Result; use crate::storage_engine::{Result, FilePosition};
use crate::error::{Error, DecodeErrorKind}; use crate::error::{Error, DecodeErrorKind};
use crate::entry_header::{EntryHeader, EntryHeaderWithDataSize}; use crate::entry_header::{EntryHeader, EntryHeaderWithDataSize};
@ -14,6 +14,7 @@ pub struct Entry<T> {
#[derive(Debug)] #[derive(Debug)]
pub struct EntryDetailed<T> { pub struct EntryDetailed<T> {
pub header: EntryHeaderWithDataSize, pub header: EntryHeaderWithDataSize,
pub file_position: FilePosition,
pub data: Vec<T>, pub data: Vec<T>,
} }
@ -40,11 +41,11 @@ impl <T>Entry<T> {
} }
impl <T>EntryDetailed<T> { impl <T>EntryDetailed<T> {
pub fn decode(header: EntryHeaderWithDataSize, number_of_columns: usize, bytes: &[u8]) -> Result<Self> pub fn decode(header: EntryHeaderWithDataSize, file_position: FilePosition, number_of_columns: usize, bytes: &[u8]) -> Result<Self>
where T: Decode where T: Decode
{ {
let data = decode_sequence::<T>(number_of_columns, bytes) let data = decode_sequence::<T>(number_of_columns, bytes)
.map_err(|e| Error::DecodeError(DecodeErrorKind::EntryData, e))?; .map_err(|e| Error::DecodeError(DecodeErrorKind::EntryData, e))?;
Ok(EntryDetailed { header, data }) Ok(EntryDetailed { header, file_position, data })
} }
} }