brakes 0.1.1

A distributed rate limiting library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use brakes::backend::{local::Memory, Backend};

#[test]
fn memory() {
    let backend = Memory::new();
    test_backend(backend);
}

#[cfg(feature = "memcache")]
#[test]
fn memcache() {
    use brakes::backend::memcache::MemCache;

    let cache = memcache::connect("memcache://127.0.0.1:11211").unwrap();
    let backend = MemCache::new(cache);
    test_backend(backend);
}

#[cfg(feature = "redis")]
#[test]
fn redis() {
    use std::time::Duration;

    use brakes::backend::redis::RedisBackend;

    let client = redis::Client::open("redis://127.0.0.1/").unwrap();
    let pool = r2d2::Pool::builder()
        .connection_timeout(Duration::from_secs(1))
        .build(client)
        .unwrap();

    let backend = RedisBackend::new(pool);
    test_backend(backend);
}

fn test_backend(backend: impl Backend) {
    let key = "key";

    backend.set(key, &[], None).unwrap();

    let value = backend.get(key);
    assert!(value.is_ok());
    assert_eq!(value.unwrap().0, Vec::<u8>::new());

    assert!(backend.delete(key).is_ok());

    let value = backend.get(key);
    assert!(value.is_err());

    assert!(backend.delete(key).is_ok());
}