From edb0859029dce8abf3d53b4801d6925aa3798230 Mon Sep 17 00:00:00 2001
From: siddontang <siddontang@gmail.com>
Date: Thu, 20 Oct 2016 09:25:36 +0800
Subject: [PATCH] support get statistics

---
 src/ffi.rs             |  1 +
 src/rocksdb_options.rs | 24 ++++++++++++++++++++++--
 2 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/src/ffi.rs b/src/ffi.rs
index 15c7ac1..a9f7e4e 100644
--- a/src/ffi.rs
+++ b/src/ffi.rs
@@ -207,6 +207,7 @@ extern "C" {
     pub fn rocksdb_options_set_report_bg_io_stats(options: DBOptions, v: c_int);
     pub fn rocksdb_options_set_wal_recovery_mode(options: DBOptions, mode: DBRecoveryMode);
     pub fn rocksdb_options_enable_statistics(options: DBOptions);
+    pub fn rocksdb_options_statistics_get_string(options: DBOptions) -> *const c_char;
     pub fn rocksdb_options_set_stats_dump_period_sec(options: DBOptions, v: usize);
     pub fn rocksdb_options_set_num_levels(options: DBOptions, v: c_int);
     pub fn rocksdb_filterpolicy_create_bloom_full(bits_per_key: c_int)
diff --git a/src/rocksdb_options.rs b/src/rocksdb_options.rs
index 568d5d1..5f0c67e 100644
--- a/src/rocksdb_options.rs
+++ b/src/rocksdb_options.rs
@@ -13,8 +13,8 @@
 // limitations under the License.
 //
 extern crate libc;
-use self::libc::{c_int,size_t};
-use std::ffi::CString;
+use self::libc::{c_int, size_t, c_void};
+use std::ffi::{CStr, CString};
 use std::mem;
 
 use rocksdb_ffi::{self, DBCompressionType, DBRecoveryMode};
@@ -379,6 +379,22 @@ impl Options {
         }
     }
 
+    pub fn get_statistics(&self) -> Option<String> {
+        unsafe {
+            let value = rocksdb_ffi::rocksdb_options_statistics_get_string(self.inner);
+
+
+            if value.is_null() {
+                return None;
+            }
+
+            // Must valid UTF-8 format.
+            let s = CStr::from_ptr(value).to_str().unwrap().to_owned();
+            libc::free(value as *mut c_void);
+            Some(s)
+        }
+    }
+
     pub fn set_stats_dump_period_sec(&mut self, period: usize) {
         unsafe {
             rocksdb_ffi::rocksdb_options_set_stats_dump_period_sec(self.inner,
@@ -454,5 +470,9 @@ mod tests {
         let mut opts = Options::default();
         opts.enable_statistics();
         opts.set_stats_dump_period_sec(60);
+        assert!(opts.get_statistics().is_some());
+
+        let opts = Options::default();
+        assert!(opts.get_statistics().is_none());
     }
 }