Commit 325476aa authored by Tyler Neely's avatar Tyler Neely

spacing

parent 8b24b630
...@@ -7,49 +7,49 @@ use std::io::process::InheritFd; ...@@ -7,49 +7,49 @@ use std::io::process::InheritFd;
//TODO windows support //TODO windows support
fn main() { fn main() {
// Next, fall back and try to use pkg-config if its available. // Next, fall back and try to use pkg-config if its available.
match pkg_config::find_library("librocksdb") { match pkg_config::find_library("librocksdb") {
Ok(()) => return, Ok(()) => return,
Err(..) => {} Err(..) => {}
} }
let src = os::getcwd().unwrap(); let src = os::getcwd().unwrap();
let dst = Path::new(os::getenv("OUT_DIR").unwrap()); let dst = Path::new(os::getenv("OUT_DIR").unwrap());
let _ = fs::mkdir(&dst.join("build"), io::USER_DIR); let _ = fs::mkdir(&dst.join("build"), io::USER_DIR);
println!("cwd: {}", src.join("rocksdb").as_str()); println!("cwd: {}", src.join("rocksdb").as_str());
run(Command::new(make()) run(Command::new(make())
.arg("shared_lib") .arg("shared_lib")
.arg(format!("-j{}", os::getenv("NUM_JOBS").unwrap())) .arg(format!("-j{}", os::getenv("NUM_JOBS").unwrap()))
.cwd(&src.join("rocksdb"))); .cwd(&src.join("rocksdb")));
// Don't run `make install` because apparently it's a little buggy on mingw // Don't run `make install` because apparently it's a little buggy on mingw
// for windows. // for windows.
fs::mkdir_recursive(&dst.join("lib/pkgconfig"), io::USER_DIR).unwrap(); fs::mkdir_recursive(&dst.join("lib/pkgconfig"), io::USER_DIR).unwrap();
let target = os::getenv("TARGET").unwrap(); let target = os::getenv("TARGET").unwrap();
if target.contains("apple") { if target.contains("apple") {
fs::rename(&src.join("rocksdb/librocksdb.dylib"), &dst.join("lib/librocksdb.dylib")).unwrap(); fs::rename(&src.join("rocksdb/librocksdb.dylib"), &dst.join("lib/librocksdb.dylib")).unwrap();
} else { } else {
fs::rename(&src.join("rocksdb/librocksdb.so"), &dst.join("lib/librocksdb.so")).unwrap(); fs::rename(&src.join("rocksdb/librocksdb.so"), &dst.join("lib/librocksdb.so")).unwrap();
} }
println!("cargo:rustc-flags=-L {}/lib -l rocksdb:dylib", dst.display()); println!("cargo:rustc-flags=-L {}/lib -l rocksdb:dylib", dst.display());
println!("cargo:root={}", dst.display()); println!("cargo:root={}", dst.display());
println!("cargo:include={}/include", src.join("rocksdb").display()); println!("cargo:include={}/include", src.join("rocksdb").display());
} }
fn run(cmd: &mut Command) { fn run(cmd: &mut Command) {
println!("running: {}", cmd); println!("running: {}", cmd);
assert!(cmd.stdout(InheritFd(1)) assert!(cmd.stdout(InheritFd(1))
.stderr(InheritFd(2)) .stderr(InheritFd(2))
.status() .status()
.unwrap() .unwrap()
.success()); .success());
} }
fn make() -> &'static str { fn make() -> &'static str {
if cfg!(target_os = "freebsd") {"gmake"} else {"make"} if cfg!(target_os = "freebsd") {"gmake"} else {"make"}
} }
This diff is collapsed.
...@@ -5,9 +5,9 @@ ...@@ -5,9 +5,9 @@
extern crate "rocksdb-sys" as rocksdb_ffi; extern crate "rocksdb-sys" as rocksdb_ffi;
pub use rocksdb::{ pub use rocksdb::{
create_or_open, create_or_open,
open, open,
Rocksdb, Rocksdb,
RocksdbResult, RocksdbResult,
}; };
pub mod rocksdb; pub mod rocksdb;
...@@ -5,50 +5,50 @@ use test::Bencher; ...@@ -5,50 +5,50 @@ use test::Bencher;
#[allow(dead_code)] #[allow(dead_code)]
fn main() { fn main() {
match rocksdb::create_or_open("/tmp/rust-rocksdb".to_string()) { match rocksdb::create_or_open("/tmp/rust-rocksdb".to_string()) {
Ok(db) => { Ok(db) => {
db.put(b"my key", b"my value"); db.put(b"my key", b"my value");
db.get(b"my key").map( |value| { db.get(b"my key").map( |value| {
match value.to_utf8() { match value.to_utf8() {
Some(v) => Some(v) =>
println!("retrieved utf8 value {}", v), println!("retrieved utf8 value {}", v),
None => None =>
println!("did not read valid utf-8 out of the db"), println!("did not read valid utf-8 out of the db"),
}}); }});
db.get(b"NOT my key").on_absent(|| { println!("value not found") }); db.get(b"NOT my key").on_absent(|| { println!("value not found") });
db.close(); db.close();
}, },
Err(e) => panic!(e), Err(e) => panic!(e),
} }
} }
#[allow(dead_code)] #[allow(dead_code)]
#[bench] #[bench]
fn writes(b: &mut Bencher) { fn writes(b: &mut Bencher) {
let db = open("testdb".to_string(), true).unwrap(); let db = open("testdb".to_string(), true).unwrap();
let mut i = 0 as u64; let mut i = 0 as u64;
b.iter(|| { b.iter(|| {
db.put(i.to_string().as_bytes(), b"v1111"); db.put(i.to_string().as_bytes(), b"v1111");
i += 1; i += 1;
}); });
db.close(); db.close();
} }
#[allow(dead_code)] #[allow(dead_code)]
#[bench] #[bench]
fn reads(b: &mut Bencher) { fn reads(b: &mut Bencher) {
let db = open("testdb".to_string(), true).unwrap(); let db = open("testdb".to_string(), true).unwrap();
let mut i = 0 as u64; let mut i = 0 as u64;
b.iter(|| { b.iter(|| {
db.get(i.to_string().as_bytes()).on_error( db.get(i.to_string().as_bytes()).on_error(
|e| { |e| {
println!("error: {}", e); println!("error: {}", e);
e e
}); });
i += 1; i += 1;
}); });
db.close(); db.close();
} }
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment