mithril_aggregator/event_store/
event.rs

1use chrono::{DateTime, Utc};
2use mithril_common::entities::SignerWithStake;
3use serde::Serialize;
4
5use std::collections::HashMap;
6
7/// Event that is sent from a thread to be persisted.
8#[derive(Debug, Clone, PartialEq)]
9pub struct EventMessage {
10    /// The source of the message shall be composed of the name of the thread
11    /// that sends the message, the name of the method can be added to it,
12    /// separated by `:`. Example: `Runtime::update_beacon` or
13    /// `HTTP::register_signer`.
14    pub source: String,
15
16    /// The action represent the action that is going to be declared and as such
17    /// represents the type of the JSON content.
18    pub action: String,
19
20    /// JSON content of the message, its type is declared in the action property.
21    pub content: serde_json::Value,
22
23    /// Headers
24    pub headers: HashMap<String, String>,
25}
26
27impl EventMessage {
28    /// Instantiate a new EventMessage.
29    pub fn new<T>(source: &str, action: &str, content: &T, headers: Vec<(&str, &str)>) -> Self
30    where
31        T: Serialize,
32    {
33        let content = serde_json::json!(content);
34
35        EventMessage {
36            source: source.to_string(),
37            action: action.to_string(),
38            content,
39            headers: headers
40                .into_iter()
41                .map(|(h, v)| (h.to_string(), v.to_string()))
42                .collect(),
43        }
44    }
45
46    /// Create a signer registration event message.
47    pub fn signer_registration(
48        source: &str,
49        signer_with_stake: &SignerWithStake,
50        signer_node_version: Option<String>,
51        epoch_str: &str,
52    ) -> Self {
53        let mut headers: Vec<(&str, &str)> = match signer_node_version.as_ref() {
54            Some(version) => vec![("signer-node-version", version)],
55            None => Vec::new(),
56        };
57
58        if !epoch_str.is_empty() {
59            headers.push(("epoch", epoch_str));
60        }
61
62        Self::new::<SignerWithStake>(source, "register_signer", signer_with_stake, headers)
63    }
64}
65
66/// Event persisted in the Event Store.
67pub struct Event {
68    /// Sequential number of the event, this is set by the database.
69    pub event_id: i64,
70
71    /// timestamp of event creation in the database.
72    pub created_at: DateTime<Utc>,
73
74    /// the `source` of the original [EventMessage] this Event originates from.
75    pub source: String,
76
77    /// the `action` of the original [EventMessage] this Event originates from.
78    pub action: String,
79
80    /// the `content` of the original [EventMessage] this Event originates from.
81    pub content: String,
82}
mithril_aggregator/event_store/event.rs

mithril_aggregator/event_store/
event.rs
