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

update rate_limiter api (#153)

parent c4e5096c
......@@ -168,7 +168,7 @@ impl RateLimiter {
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 {
crocksdb_ffi::crocksdb_ratelimiter_set_bytes_per_second(self.inner, bytes_per_sec);
}
......@@ -178,7 +178,7 @@ impl RateLimiter {
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 {
crocksdb_ffi::crocksdb_ratelimiter_request(self.inner, bytes, pri);
}
......
......@@ -16,7 +16,7 @@ use rocksdb::RateLimiter;
#[test]
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);
rate_limiter.set_bytes_per_second(20 * 1024 * 1024);
......@@ -41,7 +41,7 @@ fn test_rate_limiter() {
#[test]
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 || {
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