rust-rocksdb/src/main.rs

extern crate rocksdb;
extern crate test;
use rocksdb::open;
use test::Bencher;

#[allow(dead_code)]
fn main() {
  match rocksdb::create_or_open("/tmp/rust-rocksdb".to_string()) {
    Ok(db) => {
      db.put(b"my key", b"my value");

      db.get(b"my key").map( |value| {
        match value.to_utf8() {
          Some(v) =>
            println!("retrieved utf8 value {}", v),
          None =>
            println!("did not read valid utf-8 out of the db"),
        }});

      db.get(b"NOT my key").on_absent(|| { println!("value not found") });

      db.close();
    },
    Err(e) => panic!(e),
  }
}

#[allow(dead_code)]
#[bench]
fn writes(b: &mut Bencher) {
  let db = open("testdb".to_string(), true).unwrap();
  let mut i = 0 as u64;
  b.iter(|| {
    db.put(i.to_string().as_bytes(), b"v1111");
    i += 1;
  });
  db.close();
}

#[allow(dead_code)]
#[bench]
fn reads(b: &mut Bencher) {
  let db = open("testdb".to_string(), true).unwrap();
  let mut i = 0 as u64;
  b.iter(|| {
    db.get(i.to_string().as_bytes()).on_error(
      |e| {
        println!("error: {}", e);
        e
      });
    i += 1;
  });
  db.close();
}