eventdbx 3.19.4

Immutable, event-sourced, nosql, write-side database system.
Documentation
use serde_json::to_string;
use tracing::{Level, debug, error, info, trace, warn};

use crate::{
    config::LogPluginConfig,
    error::{EventError, Result},
    store::EventRecord,
};

use super::{Plugin, PluginDelivery};

pub(super) struct LogPlugin {
    config: LogPluginConfig,
}

impl LogPlugin {
    pub(super) fn new(config: LogPluginConfig) -> Self {
        Self { config }
    }

    pub(super) fn ensure_ready(&self) -> Result<()> {
        self.level().map(|_| ())
    }

    fn level(&self) -> Result<Level> {
        match self.config.level.to_ascii_lowercase().as_str() {
            "trace" => Ok(Level::TRACE),
            "debug" => Ok(Level::DEBUG),
            "info" => Ok(Level::INFO),
            "warn" => Ok(Level::WARN),
            "error" => Ok(Level::ERROR),
            other => Err(EventError::Config(format!(
                "unsupported log level '{}'",
                other
            ))),
        }
    }

    fn format_message(&self, record: &EventRecord) -> String {
        if let Some(template) = &self.config.template {
            template
                .replace("{aggregate}", &record.aggregate_type)
                .replace("{id}", &record.aggregate_id)
                .replace("{event}", &record.event_type)
        } else {
            format!(
                "aggregate={} id={} event={} payload={}",
                record.aggregate_type,
                record.aggregate_id,
                record.event_type,
                serde_json::to_string(&record.payload).unwrap_or_default()
            )
        }
    }

    fn log(&self, level: Level, message: String) {
        match level {
            Level::TRACE => trace!(target: "eventdbx.plugin.log", "{}", message),
            Level::DEBUG => debug!(target: "eventdbx.plugin.log", "{}", message),
            Level::INFO => info!(target: "eventdbx.plugin.log", "{}", message),
            Level::WARN => warn!(target: "eventdbx.plugin.log", "{}", message),
            Level::ERROR => error!(target: "eventdbx.plugin.log", "{}", message),
        }
    }
}

impl Plugin for LogPlugin {
    fn name(&self) -> &'static str {
        "log"
    }

    fn notify_event(&self, delivery: PluginDelivery<'_>) -> Result<()> {
        let level = self.level()?;
        let mut emitted = false;
        if let Some(record) = delivery.record {
            let message = self.format_message(record);
            self.log(level, message);
            emitted = true;
        }
        if let Some(state) = delivery.state {
            let message = format!(
                "aggregate={} id={} state={}",
                state.aggregate_type,
                state.aggregate_id,
                to_string(&state.state).unwrap_or_default()
            );
            self.log(level, message);
            emitted = true;
        }
        if let Some(schema) = delivery.schema {
            let message = format!(
                "schema={} payload={}",
                schema.aggregate,
                to_string(schema).unwrap_or_default()
            );
            self.log(level, message);
            emitted = true;
        }
        if !emitted {
            self.log(level, "plugin job contained no data".to_string());
        }
        Ok(())
    }
}