geph5-client 0.2.1

Geph5 client
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

use std::sync::atomic::Ordering;

use anyctx::AnyCtx;
use async_trait::async_trait;
use atomic_float::AtomicF64;
use dashmap::DashMap;
use nanorpc::nanorpc_derive;

use smol_str::SmolStr;

use crate::{client::CtxField, Config};

static NUM_STATS: CtxField<DashMap<SmolStr, AtomicF64>> = |_| DashMap::new();

// pub fn stat_set_num(ctx: &AnyCtx<Config>, stat: &str, num: f64) {
//     ctx.get(NUM_STATS)
//         .entry(stat.into())
//         .or_default()
//         .store(num, Ordering::Relaxed);
// }

pub fn stat_incr_num(ctx: &AnyCtx<Config>, stat: &str, num: f64) {
    ctx.get(NUM_STATS)
        .entry(stat.into())
        .or_default()
        .fetch_add(num, Ordering::Relaxed);
}

pub fn stat_get_num(ctx: &AnyCtx<Config>, stat: &str) -> f64 {
    ctx.get(NUM_STATS)
        .get(stat)
        .map(|v| v.load(Ordering::Relaxed))
        .unwrap_or(0.0)
}

pub struct ClientControlImpl(pub AnyCtx<Config>);

#[async_trait]
impl ClientControlProtocol for ClientControlImpl {
    async fn stat_num(&self, stat: SmolStr) -> f64 {
        stat_get_num(&self.0, &stat)
    }
}

/// The RPC protocol exposed by a single client.
#[nanorpc_derive]
#[async_trait]
pub trait ClientControlProtocol {
    /// Get the current statistics.
    async fn stat_num(&self, stat: SmolStr) -> f64;
}

// #[tracing::instrument(skip_all)]
// pub async fn stat_serve_loop(ctx: &AnyCtx<Config>) -> anyhow::Result<()> {
//     if let Some(listen) = ctx.init().stats_listen {
//         nanorpc_sillad::rpc_serve(
//             sillad::tcp::TcpListener::bind(listen).await?,
//             ClientControlService(ClientControlImpl(ctx.clone())),
//         )
//         .await?;
//         Ok(())
//     } else {
//         smol::future::pending().await
//     }
// }