moloch_storage/
traits.rs

1//! Storage traits.
2
3use moloch_core::{AuditEvent, Block, BlockHash, BlockHeader, EventId, Hash, Result};
4
5/// Store for audit events.
6pub trait EventStore {
7    /// Get an event by ID.
8    fn get_event(&self, id: &EventId) -> Result<Option<AuditEvent>>;
9
10    /// Store an event.
11    fn put_event(&self, event: &AuditEvent) -> Result<()>;
12
13    /// Check if an event exists.
14    fn event_exists(&self, id: &EventId) -> Result<bool>;
15
16    /// Get events by block height.
17    fn get_events_by_block(&self, height: u64) -> Result<Vec<AuditEvent>>;
18}
19
20/// Store for blocks.
21pub trait BlockStore {
22    /// Get a block by height.
23    fn get_block(&self, height: u64) -> Result<Option<Block>>;
24
25    /// Get a block by hash.
26    fn get_block_by_hash(&self, hash: &BlockHash) -> Result<Option<Block>>;
27
28    /// Get a block header by height.
29    fn get_header(&self, height: u64) -> Result<Option<BlockHeader>>;
30
31    /// Store a block.
32    fn put_block(&self, block: &Block) -> Result<()>;
33
34    /// Get the latest block height.
35    fn latest_height(&self) -> Result<Option<u64>>;
36
37    /// Get the latest block.
38    fn latest_block(&self) -> Result<Option<Block>>;
39}
40
41/// Combined chain store with MMR support.
42pub trait ChainStore: EventStore + BlockStore {
43    /// Get an MMR node by position.
44    fn get_mmr_node(&self, pos: u64) -> Result<Option<Hash>>;
45
46    /// Store an MMR node.
47    fn put_mmr_node(&self, pos: u64, hash: Hash) -> Result<()>;
48
49    /// Get the MMR size.
50    fn mmr_size(&self) -> Result<u64>;
51
52    /// Get the MMR leaf count.
53    fn mmr_leaf_count(&self) -> Result<u64>;
54
55    /// Set MMR metadata.
56    fn set_mmr_meta(&self, size: u64, leaf_count: u64) -> Result<()>;
57
58    /// Flush all pending writes.
59    fn flush(&self) -> Result<()>;
60}
moloch_storage/traits.rs

moloch_storage/
traits.rs
