From e673fc4d88bdd7d15745489256f468b1ae46ace7 Mon Sep 17 00:00:00 2001
From: Nan Jiang <njiang028@gmail.com>
Date: Fri, 8 Jun 2018 15:54:13 -0400
Subject: [PATCH] add an example for iterators

---
 examples/README.md   |  2 ++
 examples/iterator.rs | 81 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 83 insertions(+)
 create mode 100644 examples/iterator.rs

diff --git a/examples/README.md b/examples/README.md
index d132b3d..0e5a3e6 100644
--- a/examples/README.md
+++ b/examples/README.md
@@ -7,3 +7,5 @@ cargo run --example $name
 ```
 
 * [`simple-store`](simple-store.rs) - a simple key/value store that showcases the basic usage of rkv.
+
+* [`iterator`](iterator.rs) - a demo that showcases the basic usage of iterators in rkv.
diff --git a/examples/iterator.rs b/examples/iterator.rs
new file mode 100644
index 0000000..6ef26a5
--- /dev/null
+++ b/examples/iterator.rs
@@ -0,0 +1,81 @@
+// Any copyright is dedicated to the Public Domain.
+// http://creativecommons.org/publicdomain/zero/1.0/
+
+//! A demo that showcases the basic usage of iterators in rkv.
+//!
+//! You can test this out by running:
+//!
+//!     cargo run --example iterator
+
+extern crate rkv;
+extern crate tempfile;
+
+use rkv::{
+    Manager,
+    Rkv,
+    Store,
+    StoreError,
+    Value,
+};
+use tempfile::Builder;
+
+use std::fs;
+use std::str;
+
+fn main() {
+    let root = Builder::new().prefix("iterator").tempdir().unwrap();
+    fs::create_dir_all(root.path()).unwrap();
+    let p = root.path();
+
+    let created_arc = Manager::singleton()
+        .write()
+        .unwrap()
+        .get_or_create(p, Rkv::new)
+        .unwrap();
+    let k = created_arc.read().unwrap();
+    let store: Store<&str> = k.create_or_open("store").unwrap();
+
+    populate_store(&k, &store).unwrap();
+
+    let reader = store.read(&k).unwrap();
+
+    println!("Iterating from the beginning...");
+    // Reader::iter_start() iterates from the first item in the store, and
+    // returns the (key, value) tuples in order.
+    let mut iter = reader.iter_start().unwrap();
+    while let Some((country, city)) = iter.next() {
+        println!("{}, {:?}", str::from_utf8(country).unwrap(), city);
+    }
+
+    println!("");
+    println!("Iterating from the given key...");
+    // Reader::iter_from() iterates from the first key equal to or greater
+    // than the given key.
+    let mut iter = reader.iter_from("Japan").unwrap();
+    while let Some((country, city)) = iter.next() {
+        println!("{}, {:?}", str::from_utf8(country).unwrap(), city);
+    }
+
+    println!("");
+    println!("Iterating from the given prefix...");
+    let mut iter = reader.iter_from("Un").unwrap();
+    while let Some((country, city)) = iter.next() {
+        println!("{}, {:?}", str::from_utf8(country).unwrap(), city);
+    }
+}
+
+fn populate_store(k: &Rkv, store: &Store<&str>) -> Result<(), StoreError> {
+    let mut writer = store.write(k)?;
+    for (country, city) in &vec![
+        ("Canada", Value::Str("Ottawa")),
+        ("United States of America", Value::Str("Washington")),
+        ("Germany", Value::Str("Berlin")),
+        ("France", Value::Str("Paris")),
+        ("Italy", Value::Str("Rome")),
+        ("United Kingdom", Value::Str("London")),
+        ("Japan", Value::Str("Tokyo")),
+    ] {
+        writer.put(country, city)?;
+    }
+    writer.commit()
+}