relay-core-runtime 0.1.0

High-performance Rust traffic interception engine and proxy platform
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use tokio::sync::broadcast;
use relay_core_api::flow::FlowUpdate;
use crate::CoreState;

pub trait FlowEventHub: Send + Sync {
    fn subscribe_flow_updates(&self) -> broadcast::Receiver<FlowUpdate>;
    fn redact_flow_update_for_output(&self, update: FlowUpdate) -> FlowUpdate;
    fn record_flow_events_lagged(&self, skipped: u64);
}

impl FlowEventHub for CoreState {
    fn subscribe_flow_updates(&self) -> broadcast::Receiver<FlowUpdate> {
        CoreState::subscribe_flow_updates(self)
    }

    fn redact_flow_update_for_output(&self, update: FlowUpdate) -> FlowUpdate {
        CoreState::redact_flow_update_for_output(self, update)
    }

    fn record_flow_events_lagged(&self, skipped: u64) {
        CoreState::record_flow_events_lagged(self, skipped)
    }
}