blob: 9f20e3c3f17e3161fd9a8c7e62b74f5ae3e3ae33 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
use log_db::*;
use rand::distributions::Alphanumeric;
use rand::Rng;
use std::fmt::Debug;
use std::io;
// Function to generate a random integer
pub fn random_int() -> i64 {
let mut rng = rand::thread_rng();
rng.gen_range(0..1000) // Random number between 0 and 1000
}
// Function to generate a random string
pub fn random_string(len: usize) -> String {
let mut rng = rand::thread_rng();
(0..len).map(|_| rng.sample(Alphanumeric) as char).collect()
}
// Function to generate random bytes
pub fn random_bytes(len: usize) -> Vec<u8> {
let mut rng = rand::thread_rng();
(0..len).map(|_| rng.gen()).collect()
}
// Function to generate a random record
pub fn random_record() -> Record {
Record {
values: vec![
RecordValue::Int(random_int()), // Random int value
RecordValue::String(random_string(5)), // Random string of length 5
RecordValue::Bytes(random_bytes(10)), // Random bytes of length 10
],
}
}
pub fn prefill_db_with_n_records<T: Eq + Clone + Debug>(
db: &mut DB<T>,
n: usize,
) -> Result<(), io::Error> {
for _ in 0..n {
let record = random_record();
db.upsert(&record)?;
}
Ok(())
}
|
