bader-db 0.1.5

Key-value cache RESP server with support for key expirations
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

mod server;
mod cache;
mod resp;

use std::time::Duration;
use std::sync::Arc;
use tokio::{net::TcpListener, sync::broadcast, signal};

use crate::server::Server;
use crate::cache::Cache;

pub async fn run_server(socket_addr: &str,
                        sample: usize,
                        threshold: f64,
                        frequency: Duration) {

    // Bind a tcp listener
    let listener = TcpListener::bind(socket_addr).await.unwrap();

    // Create the shutdown signal which will shutdown when we hit ctrl_c
    let shutdown = signal::ctrl_c();

    // Channels used for a graceful shutdown
    let (notify_shutdown, _) = broadcast::channel(1);

    // Create the main_cache arc that we clone in every connection. We only clone a ref to the store
    // which makes it inexpensive
    let main_cache = Arc::new(Cache::new(
        sample,
        threshold,
        frequency,
    ));

    // Create a a cache clone to spawn the cache monitor_for_expiry
    let clone = main_cache.clone();

    let task = tokio::spawn(async move {
        clone.monitor_for_expiry().await
    });

    // Create the server instance
    let server = Server::new(socket_addr,
                                 main_cache,
                                 listener,
                                 notify_shutdown);

    log::info!("{:?}", "Server is created");

    tokio::select! {
        res = server.run() => {
            if let Err(err) = res {
                log::error!("failed to run the server, error {:?}", err);
            }
        }
        _ = shutdown => {
            log::info!("shutting down");
        }
    }

    let Server {
        notify_shutdown,
        ..
    } = server;

    // closing background monitor
    task.abort();

    // gracefully exit from all spinning connections
    drop(notify_shutdown);

    log::info!("{:?}", "Server is closed");
}