From d2642dfe0cc130de04f1344a793bffe474af91bb Mon Sep 17 00:00:00 2001 From: Pere Diaz Bou Date: Tue, 1 Apr 2025 00:58:01 +0200 Subject: [PATCH] skip repeated keys --- core/storage/btree.rs | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/core/storage/btree.rs b/core/storage/btree.rs index d475e384f..21c74574a 100644 --- a/core/storage/btree.rs +++ b/core/storage/btree.rs @@ -4007,8 +4007,20 @@ mod tests { let mut rng = ChaCha8Rng::seed_from_u64(seed); for insert_id in 0..inserts { let size = size(&mut rng); - let key = (rng.next_u64() % (1 << 30)) as i64; - assert!(seen.insert(key)); + let key = { + let mut result = None; + loop { + let key = (rng.next_u64() % (1 << 30)) as i64; + if seen.contains(&key) { + continue; + } else { + seen.insert(key); + } + result = Some(key); + break; + } + result.unwrap() + }; keys.push(key); tracing::info!( "INSERT INTO t VALUES ({}, randomblob({})); -- {}",