#![warn(missing_docs)]
use async_trait::async_trait;
use crate::dht::Did;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum MeasureCounter {
Sent,
FailedToSend,
Received,
FailedToReceive,
Connect,
Disconnected,
}
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait Measure {
async fn incr(&self, did: Did, counter: MeasureCounter);
async fn get_count(&self, did: Did, counter: MeasureCounter) -> u64;
}
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait BehaviourJudgement: Measure {
async fn good(&self, did: Did) -> bool;
}
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait ConnectBehaviour<const THRESHOLD: i16>: Measure {
async fn good(&self, did: Did) -> bool {
let conn = self.get_count(did, MeasureCounter::Connect).await;
let disconn = self.get_count(did, MeasureCounter::Disconnected).await;
tracing::debug!(
"[ConnectBehaviour] in Threadhold: {:}, connect: {:}, disconn: {:}, delta: {:}",
THRESHOLD,
conn,
disconn,
conn - disconn
);
((conn - disconn) as i16) < THRESHOLD
}
}
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait MessageSendBehaviour<const THRESHOLD: i16>: Measure {
async fn good(&self, did: Did) -> bool {
let failed = self.get_count(did, MeasureCounter::FailedToSend).await;
(failed as i16) < THRESHOLD
}
}
#[cfg_attr(feature = "wasm", async_trait(?Send))]
#[cfg_attr(not(feature = "wasm"), async_trait)]
pub trait MessageRecvBehaviour<const THRESHOLD: i16>: Measure {
async fn good(&self, did: Did) -> bool {
let failed = self.get_count(did, MeasureCounter::FailedToReceive).await;
(failed as i16) < THRESHOLD
}
}