mirrors/limbo
1
0
Fork 0
mirror of https://github.com/tursodatabase/limbo.git synced 2025-08-04 01:58:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

faster validate

Browse source
This commit is contained in:
Jussi Saurio 2025-04-24 16:05:12 +03:00
parent af6a783f4d
commit 04adf8242a
2 changed files with 29 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

30
core/storage/sqlite3_ondisk.rs
View file

@ -985,8 +985,12 @@ fn read_payload(unread: &'static [u8], payload_size: usize) -> (&'static [u8], O
}
}
pub fn validate_serial_type(value: u64) -> Result<SerialType> {
value.try_into()
#[inline(always)]
pub fn validate_serial_type(value: u64) -> Result<()> {
if !SerialType::u64_is_valid_serial_type(value) {
crate::bail_corrupt_error!("Invalid serial type: {}", value);
}
Ok(())
}
pub struct SmallVec<T, const N: usize = 64> {
@ -1081,7 +1085,7 @@ pub fn read_record(payload: &[u8], reuse_immutable: &mut ImmutableRecord) -> Res
let mut serial_types = SmallVec::<u64, 64>::new();
while header_size > 0 {
let (serial_type, nr) = read_varint(&reuse_immutable.get_payload()[pos..])?;
let _ = validate_serial_type(serial_type)?;
validate_serial_type(serial_type)?;
serial_types.push(serial_type);
pos += nr;
assert!(header_size >= nr);
@ -1641,15 +1645,25 @@ mod tests {
#[case(13, SerialType::Text { content_size: 0 })]
#[case(14, SerialType::Blob { content_size: 1 })]
#[case(15, SerialType::Text { content_size: 1 })]
fn test_validate_serial_type(#[case] input: u64, #[case] expected: SerialType) {
let result = validate_serial_type(input).unwrap();
fn test_parse_serial_type(#[case] input: u64, #[case] expected: SerialType) {
let result = SerialType::try_from(input).unwrap();
assert_eq!(result, expected);
}
#[test]
fn test_invalid_serial_type() {
let result = validate_serial_type(10);
assert!(result.is_err());
fn test_validate_serial_type() {
for i in 0..=9 {
let result = validate_serial_type(i);
assert!(result.is_ok());
}
for i in 10..=11 {
let result = validate_serial_type(i);
assert!(result.is_err());
}
for i in 12..=1000 {
let result = validate_serial_type(i);
assert!(result.is_ok());
}
}
#[test]