event_scanner/
types.rs

1use std::error::Error;
2
3use tokio::sync::mpsc;
4use tracing::{info, warn};
5
6#[derive(Copy, Debug, Clone)]
7pub enum ScannerMessage<T: Clone, E: Error + Clone> {
8    Data(T),
9    Error(E),
10    Status(ScannerStatus),
11}
12
13#[derive(Copy, Debug, Clone, PartialEq)]
14pub enum ScannerStatus {
15    SwitchingToLive,
16    ReorgDetected,
17}
18
19impl<T: Clone, E: Error + Clone> From<ScannerStatus> for ScannerMessage<T, E> {
20    fn from(value: ScannerStatus) -> Self {
21        ScannerMessage::Status(value)
22    }
23}
24
25impl<T: Clone, E: Error + Clone> PartialEq<ScannerStatus> for ScannerMessage<T, E> {
26    fn eq(&self, other: &ScannerStatus) -> bool {
27        if let ScannerMessage::Status(status) = self { status == other } else { false }
28    }
29}
30
31pub(crate) trait TryStream<T: Clone, E: Error + Clone> {
32    async fn try_stream<M: Into<ScannerMessage<T, E>>>(&self, msg: M) -> bool;
33}
34
35impl<T: Clone, E: Error + Clone> TryStream<T, E> for mpsc::Sender<ScannerMessage<T, E>> {
36    async fn try_stream<M: Into<ScannerMessage<T, E>>>(&self, msg: M) -> bool {
37        if let Err(err) = self.send(msg.into()).await {
38            warn!(error = %err, "Downstream channel closed, stopping stream");
39            return false;
40        }
41        info!("Message sent");
42        true
43    }
44}
event_scanner/types.rs

event_scanner/
types.rs
