Commit ec72a896 authored by UncP's avatar UncP Committed by zhangjinpeng1987

update rate_limiter api (#153)

parent c4e5096c
...@@ -168,7 +168,7 @@ impl RateLimiter { ...@@ -168,7 +168,7 @@ impl RateLimiter {
RateLimiter { inner: limiter } RateLimiter { inner: limiter }
} }
pub fn set_bytes_per_second(&mut self, bytes_per_sec: i64) { pub fn set_bytes_per_second(&self, bytes_per_sec: i64) {
unsafe { unsafe {
crocksdb_ffi::crocksdb_ratelimiter_set_bytes_per_second(self.inner, bytes_per_sec); crocksdb_ffi::crocksdb_ratelimiter_set_bytes_per_second(self.inner, bytes_per_sec);
} }
...@@ -178,7 +178,7 @@ impl RateLimiter { ...@@ -178,7 +178,7 @@ impl RateLimiter {
unsafe { crocksdb_ffi::crocksdb_ratelimiter_get_singleburst_bytes(self.inner) } unsafe { crocksdb_ffi::crocksdb_ratelimiter_get_singleburst_bytes(self.inner) }
} }
pub fn request(&mut self, bytes: i64, pri: c_uchar) { pub fn request(&self, bytes: i64, pri: c_uchar) {
unsafe { unsafe {
crocksdb_ffi::crocksdb_ratelimiter_request(self.inner, bytes, pri); crocksdb_ffi::crocksdb_ratelimiter_request(self.inner, bytes, pri);
} }
......
...@@ -16,7 +16,7 @@ use rocksdb::RateLimiter; ...@@ -16,7 +16,7 @@ use rocksdb::RateLimiter;
#[test] #[test]
fn test_rate_limiter() { fn test_rate_limiter() {
let mut rate_limiter = RateLimiter::new(10 * 1024 * 1024, 100 * 1000, 10); let rate_limiter = RateLimiter::new(10 * 1024 * 1024, 100 * 1000, 10);
assert_eq!(rate_limiter.get_singleburst_bytes(), 1 * 1024 * 1024); assert_eq!(rate_limiter.get_singleburst_bytes(), 1 * 1024 * 1024);
rate_limiter.set_bytes_per_second(20 * 1024 * 1024); rate_limiter.set_bytes_per_second(20 * 1024 * 1024);
...@@ -41,7 +41,7 @@ fn test_rate_limiter() { ...@@ -41,7 +41,7 @@ fn test_rate_limiter() {
#[test] #[test]
fn test_rate_limiter_sendable() { fn test_rate_limiter_sendable() {
let mut rate_limiter = RateLimiter::new(10 * 1024 * 1024, 100 * 1000, 10); let rate_limiter = RateLimiter::new(10 * 1024 * 1024, 100 * 1000, 10);
let handle = thread::spawn(move || { let handle = thread::spawn(move || {
rate_limiter.request(1024, 0); rate_limiter.request(1024, 0);
......
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