Merge pull request #16 from spacejam/tyler_fmt

Run rustfmt on the code.

Cargo.toml

@@ -2,7 +2,7 @@
 
 name = "rocksdb"
 description = "A Rust wrapper for Facebook's RocksDB embeddable database."
-version = "0.0.6"
+version = "0.0.7"
 authors = ["Tyler Neely <t@jujit.su>"]
 license = "Apache-2.0"
 

README.md

@@ -2,7 +2,7 @@ rust-rocksdb
 ============
 [![Build Status](https://travis-ci.org/spacejam/rust-rocksdb.svg?branch=master)](https://travis-ci.org/spacejam/rust-rocksdb)
 
-This library has been tested against RocksDB 3.8.1 on linux and OSX.  The 0.0.6 crate should work with the Rust nightly release as of 7/12/15.
+This library has been tested against RocksDB 3.8.1 on linux and OSX.  The 0.0.7 crate should work with the Rust nightly release as of 7/16/15.
 
 ### status
   - [x] basic open/put/get/delete/close
@@ -12,7 +12,7 @@ This library has been tested against RocksDB 3.8.1 on linux and OSX.  The 0.0.6
   - [x] LRU cache
   - [x] destroy/repair
   - [x] iterator
-  - [ ] comparator
+  - [x] comparator
   - [x] snapshot
   - [ ] column family operations
   - [ ] slicetransform
@@ -31,7 +31,7 @@ sudo make install
 ###### Cargo.toml
 ```rust
 [dependencies]
-rocksdb = "~0.0.6"
+rocksdb = "~0.0.7"
 ```
 ###### Code
 ```rust

src/comparator.rs

@@ -42,10 +42,12 @@ pub extern "C" fn name_callback(raw_cb: *mut c_void) -> *const c_char {
     }
 }
 
-pub extern "C" fn compare_callback(
+pub extern "C" fn compare_callback(raw_cb: *mut c_void,
-    raw_cb: *mut c_void,
+                                   a_raw: *const c_char,
-    a_raw: *const c_char, a_len: size_t,
+                                   a_len: size_t,
-    b_raw: *const c_char, b_len: size_t) -> c_int {
+                                   b_raw: *const c_char,
+                                   b_len: size_t)
+                                   -> c_int {
     unsafe {
         let cb: &mut ComparatorCallback =
             &mut *(raw_cb as *mut ComparatorCallback);

src/ffi.rs

@@ -76,20 +76,20 @@ pub enum RocksDBCompressionType {
     RocksDBZlibCompression = 2,
     RocksDBBz2Compression = 3,
     RocksDBLz4Compression = 4,
-    RocksDBLz4hcCompression  = 5
+    RocksDBLz4hcCompression = 5,
 }
 
 #[repr(C)]
 pub enum RocksDBCompactionStyle {
     RocksDBLevelCompaction = 0,
     RocksDBUniversalCompaction = 1,
-    RocksDBFifoCompaction      = 2
+    RocksDBFifoCompaction = 2,
 }
 
 #[repr(C)]
 pub enum RocksDBUniversalCompactionStyle {
     rocksdb_similar_size_compaction_stop_style = 0,
-    rocksdb_total_size_compaction_stop_style   = 1
+    rocksdb_total_size_compaction_stop_style = 1,
 }
 
 //TODO audit the use of boolean arguments, b/c I think they need to be u8 instead...

src/lib.rs

@@ -21,26 +21,10 @@
 #![feature(raw)]
 
 pub use ffi as rocksdb_ffi;
-pub use ffi::{
+pub use ffi::{new_bloom_filter, RocksDBCompactionStyle, RocksDBComparator};
-    new_bloom_filter,
+pub use rocksdb::{RocksDB, RocksDBResult, RocksDBVector, WriteBatch, Writable, Direction};
-    RocksDBCompactionStyle,
+pub use rocksdb_options::{Options, BlockBasedOptions};
-    RocksDBComparator,
+pub use merge_operator::MergeOperands;
-};
-pub use rocksdb::{
-    RocksDB,
-    RocksDBResult,
-    RocksDBVector,
-    WriteBatch,
-    Writable,
-    Direction,
-};
-pub use rocksdb_options::{
-    Options,
-    BlockBasedOptions,
-};
-pub use merge_operator::{
-    MergeOperands,
-};
 pub mod rocksdb;
 pub mod ffi;
 pub mod rocksdb_options;

src/merge_operator.rs

@@ -20,7 +20,7 @@ use std::mem;
 use std::ptr;
 use std::slice;
 
-use rocksdb_options::{Options};
+use rocksdb_options::Options;
 use rocksdb::{RocksDB, RocksDBResult, RocksDBVector, Writable};
 
 pub struct MergeOperatorCallback {
@@ -43,12 +43,17 @@ pub extern "C" fn name_callback(raw_cb: *mut c_void) -> *const c_char {
     }
 }
 
-pub extern "C" fn full_merge_callback(
+pub extern "C" fn full_merge_callback(raw_cb: *mut c_void,
-    raw_cb: *mut c_void, raw_key: *const c_char, key_len: size_t,
+                                      raw_key: *const c_char,
-    existing_value: *const c_char, existing_value_len: size_t,
+                                      key_len: size_t,
-    operands_list: *const *const c_char, operands_list_len: *const size_t,
+                                      existing_value: *const c_char,
+                                      existing_value_len: size_t,
+                                      operands_list: *const *const c_char,
+                                      operands_list_len: *const size_t,
                                       num_operands: c_int,
-    success: *mut u8, new_value_length: *mut size_t) -> *const c_char {
+                                      success: *mut u8,
+                                      new_value_length: *mut size_t)
+                                      -> *const c_char {
     unsafe {
         let cb: &mut MergeOperatorCallback =
             &mut *(raw_cb as *mut MergeOperatorCallback);
@@ -72,11 +77,15 @@ pub extern "C" fn full_merge_callback(
     }
 }
 
-pub extern "C" fn partial_merge_callback(
+pub extern "C" fn partial_merge_callback(raw_cb: *mut c_void,
-    raw_cb: *mut c_void, raw_key: *const c_char, key_len: size_t,
+                                         raw_key: *const c_char,
-    operands_list: *const *const c_char, operands_list_len: *const size_t,
+                                         key_len: size_t,
+                                         operands_list: *const *const c_char,
+                                         operands_list_len: *const size_t,
                                          num_operands: c_int,
-    success: *mut u8, new_value_length: *mut size_t) -> *const c_char {
+                                         success: *mut u8,
+                                         new_value_length: *mut size_t)
+                                         -> *const c_char {
     unsafe {
         let cb: &mut MergeOperatorCallback =
             &mut *(raw_cb as *mut MergeOperatorCallback);
@@ -107,7 +116,8 @@ pub struct MergeOperands {
 impl MergeOperands {
     fn new(operands_list: *const *const c_char,
            operands_list_len: *const size_t,
-           num_operands: c_int) -> MergeOperands {
+           num_operands: c_int)
+           -> MergeOperands {
         assert!(num_operands >= 0);
         MergeOperands {
             operands_list: operands_list,
@@ -148,8 +158,10 @@ impl<'a> Iterator for &'a mut MergeOperands {
     }
 }
 
-fn test_provided_merge(new_key: &[u8], existing_val: Option<&[u8]>,
+fn test_provided_merge(new_key: &[u8],
-    mut operands: &mut MergeOperands) -> Vec<u8> {
+                       existing_val: Option<&[u8]>,
+                       mut operands: &mut MergeOperands)
+                       -> Vec<u8> {
     let nops = operands.size_hint().0;
     let mut result: Vec<u8> = Vec::with_capacity(nops);
     match existing_val {

src/rocksdb.rs

@@ -46,14 +46,16 @@ pub struct Snapshot<'a> {
 }
 
 pub struct DBIterator {
-//TODO: should have a reference to DB to enforce scope, but it's trickier than I thought to add
+    // TODO: should have a reference to DB to enforce scope, but it's trickier than I
+    // thought to add
     inner: rocksdb_ffi::RocksDBIterator,
     direction: Direction,
-    just_seeked: bool
+    just_seeked: bool,
 }
 
 pub enum Direction {
-  forward, reverse
+    forward,
+    reverse,
 }
 
 pub struct SubDBIterator<'a> {
@@ -105,7 +107,7 @@ impl DBIterator {
         unsafe {
             rocksdb_ffi::rocksdb_iter_seek_to_first(self.inner);
         };
-        SubDBIterator{ iter: self, direction: Direction::forward, }
+        SubDBIterator { iter: self, direction: Direction::forward }
     }
 
     pub fn from_end(&mut self) -> SubDBIterator {
@@ -113,7 +115,7 @@ impl DBIterator {
         unsafe {
             rocksdb_ffi::rocksdb_iter_seek_to_last(self.inner);
         };
-        SubDBIterator{ iter: self, direction: Direction::reverse, }
+        SubDBIterator { iter: self, direction: Direction::reverse }
     }
 
     pub fn from(&mut self, key: &[u8], dir: Direction) -> SubDBIterator {
@@ -121,7 +123,7 @@ impl DBIterator {
         unsafe {
             rocksdb_ffi::rocksdb_iter_seek(self.inner, key.as_ptr(), key.len() as size_t);
         }
-        SubDBIterator{ iter: self, direction: dir, }
+        SubDBIterator { iter: self, direction: dir }
     }
 }
 
@@ -180,7 +182,8 @@ impl RocksDB {
     pub fn open(opts: &Options, path: &str) -> Result<RocksDB, String> {
         let cpath = match CString::new(path.as_bytes()) {
             Ok(c) => c,
-                        Err(_) => return Err("Failed to convert path to CString when opening rocksdb".to_string()),
+            Err(_) =>
+                return Err("Failed to convert path to CString when opening rocksdb".to_string()),
         };
         let cpath_ptr = cpath.as_ptr();
 
@@ -357,7 +360,7 @@ impl WriteBatch {
         WriteBatch {
             inner: unsafe {
                        rocksdb_ffi::rocksdb_writebatch_create()
-                   }
+                   },
         }
     }
 }
@@ -419,7 +422,8 @@ impl ReadOptions {
         }
     }
 // TODO add snapshot setting here
-//TODO add snapshot wrapper structs with proper destructors; that struct needs an "iterator" impl too.
+// TODO add snapshot wrapper structs with proper destructors;
+// that struct needs an "iterator" impl too.
     fn fill_cache(&mut self, v: bool) {
         unsafe {
             rocksdb_ffi::rocksdb_readoptions_set_fill_cache(self.inner, v);

src/rocksdb_options.rs

@@ -19,8 +19,8 @@ use std::ffi::CString;
 use std::mem;
 
 use rocksdb_ffi;
-use merge_operator::{self, MergeOperatorCallback, MergeOperands, full_merge_callback,
+use merge_operator::{self, MergeOperatorCallback, MergeOperands,
-              partial_merge_callback};
+                     full_merge_callback, partial_merge_callback};
 use comparator::{self, ComparatorCallback, compare_callback};
 
 pub struct BlockBasedOptions {
@@ -54,7 +54,7 @@ impl BlockBasedOptions {
         if opt_ptr.is_null() {
             panic!("Could not create rocksdb block based options".to_string());
         }
-        BlockBasedOptions{ inner: block_opts, }
+        BlockBasedOptions { inner: block_opts }
     }
 
     pub fn set_block_size(&mut self, size: u64) {
@@ -107,8 +107,7 @@ impl Options {
         }
     }
 
-    pub fn optimize_level_style_compaction(&mut self,
+    pub fn optimize_level_style_compaction(&mut self, memtable_memory_budget: i32) {
-        memtable_memory_budget: i32) {
         unsafe {
             rocksdb_ffi::rocksdb_options_optimize_level_style_compaction(
                 self.inner, memtable_memory_budget);
@@ -122,7 +121,8 @@ impl Options {
         }
     }
 
-    pub fn add_merge_operator<'a>(&mut self, name: &str,
+    pub fn add_merge_operator<'a>(&mut self,
+                                  name: &str,
                                   merge_fn: fn(&[u8], Option<&[u8]>, &mut MergeOperands) -> Vec<u8>) {
         let cb = Box::new(MergeOperatorCallback {
             name: CString::new(name.as_bytes()).unwrap(),
@@ -258,8 +258,7 @@ impl Options {
         }
     }
 
-    pub fn set_compaction_style(&mut self, style:
+    pub fn set_compaction_style(&mut self, style: rocksdb_ffi::RocksDBCompactionStyle) {
-                                rocksdb_ffi::RocksDBCompactionStyle) {
         unsafe {
             rocksdb_ffi::rocksdb_options_set_compaction_style(
                 self.inner, style);
@@ -306,5 +305,3 @@ impl Options {
         }
     }
 }
-
-