wp-source-types 0.1.0

use smol_str::SmolStr;
use std::net::IpAddr;
use std::sync::Arc;
use wp_model_core::raw::RawData;

use crate::tags::Tags;

/// Parse 侧预处理钩子
pub type EventPreHook = Arc<dyn Fn(&mut SourceEvent) + Send + Sync + 'static>;

#[derive(Clone)]
pub struct SourceEvent {
    pub event_id: u64,
    pub src_key: SmolStr,
    pub payload: RawData,
    pub tags: Arc<Tags>,
    pub ups_ip: Option<IpAddr>,
    /// 可选：parse 线程在进入 WPL 前调用
    pub preproc: Option<EventPreHook>,
}

/// 一批源事件，便于批量传输；允许返回空 Vec 代表暂时无数据。
/// A batch of events for bulk delivery; empty Vec means "no data for now".
pub type SourceBatch = Vec<SourceEvent>;

impl SourceEvent {
    /// 构造一个最小帧
    pub fn new(
        event_id: u64,
        src_key: impl Into<SmolStr>,
        payload: RawData,
        tags: Arc<Tags>,
    ) -> Self {
        Self {
            event_id,
            src_key: src_key.into(),
            payload,
            tags,
            ups_ip: None,
            preproc: None,
        }
    }
}

impl std::fmt::Debug for SourceEvent {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("SourceEvent")
            .field("id", &self.event_id)
            .field("src_key", &self.src_key)
            .field(
                "payload",
                &match &self.payload {
                    RawData::String(s) => format!("String(len={})", s.len()),
                    RawData::Bytes(b) => format!("Bytes(len={})", b.len()),
                    RawData::ArcBytes(arc) => format!("ArcBytes(len={}, zcp ={})", arc.len(), true),
                },
            )
            .field("tags", &format!("{} tags", self.tags.len()))
            .field("ups_ip", &self.ups_ip)
            .finish()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::sync::Arc;

    #[test]
    fn source_event_new_sets_defaults() {
        let tags = Arc::new(Tags::default());
        let event = SourceEvent::new(7, "main", RawData::from_string("payload"), tags.clone());

        assert_eq!(event.event_id, 7);
        assert_eq!(event.src_key.as_str(), "main");
        assert!(matches!(event.payload, RawData::String(_)));
        assert!(Arc::ptr_eq(&event.tags, &tags));
        assert!(event.ups_ip.is_none());
        assert!(event.preproc.is_none());
    }

    #[test]
    fn debug_impl_reports_summary() {
        let event = SourceEvent::new(
            1,
            "key",
            RawData::from_string("hello"),
            Arc::new(Tags::default()),
        );
        let debug = format!("{event:?}");
        assert!(debug.contains("SourceEvent"));
        assert!(debug.contains("len=5"));
    }
}