view examples/write-a-lot/src/main.rs @ 609:68887f0a3e83

Add Compressor
author kaiyohugo <41114603+KAIYOHUGO@users.noreply.github.com>
date Tue, 04 Jul 2023 21:50:32 +0800
parents 63b1f80e40d5
children
line wrap: on
line source

extern crate rand;
extern crate rusty_leveldb;

use rand::Rng;
use rusty_leveldb::{compressor, CompressorId, Options, DB};

use std::error::Error;
use std::iter::FromIterator;

const KEY_LEN: usize = 16;
const VAL_LEN: usize = 48;

fn gen_string(len: usize) -> String {
    let mut rng = rand::thread_rng();
    String::from_iter(rng.gen_ascii_chars().take(len))
}

fn fill_db(db: &mut DB, entries: usize) -> Result<(), Box<dyn Error>> {
    for i in 0..entries {
        let (k, v) = (gen_string(KEY_LEN), gen_string(VAL_LEN));
        db.put(k.as_bytes(), v.as_bytes())?;

        if i % 100 == 0 {
            db.flush()?;
        }
    }
    Ok(())
}

fn main() {
    let mut opt = Options::default();
    opt.compressor = compressor::SnappyCompressor::ID;
    let mut db = DB::open("test1", opt).unwrap();

    fill_db(&mut db, 32768).unwrap();

    db.close().unwrap();
}