87 lines
2.6 KiB
Rust
87 lines
2.6 KiB
Rust
mod data;
|
|
use std::{
|
|
collections::{BTreeMap, HashMap},
|
|
sync::Arc,
|
|
};
|
|
|
|
pub use data::Data;
|
|
use ruma::{events::StateEventType, EventId, RoomId};
|
|
|
|
use crate::{PduEvent, Result};
|
|
|
|
pub struct Service {
|
|
db: Arc<dyn Data>,
|
|
}
|
|
|
|
impl Service {
|
|
/// Builds a StateMap by iterating over all keys that start
|
|
/// with state_hash, this gives the full state for the given state_hash.
|
|
pub async fn state_full_ids(&self, shortstatehash: u64) -> Result<BTreeMap<u64, Arc<EventId>>> {
|
|
self.db.state_full_ids(shortstatehash).await
|
|
}
|
|
|
|
pub async fn state_full(
|
|
&self,
|
|
shortstatehash: u64,
|
|
) -> Result<HashMap<(StateEventType, String), Arc<PduEvent>>> {
|
|
self.db.state_full(shortstatehash).await
|
|
}
|
|
|
|
/// Returns a single PDU from `room_id` with key (`event_type`, `state_key`).
|
|
#[tracing::instrument(skip(self))]
|
|
pub fn state_get_id(
|
|
&self,
|
|
shortstatehash: u64,
|
|
event_type: &StateEventType,
|
|
state_key: &str,
|
|
) -> Result<Option<Arc<EventId>>> {
|
|
self.db.state_get_id(shortstatehash, event_type, state_key)
|
|
}
|
|
|
|
/// Returns a single PDU from `room_id` with key (`event_type`, `state_key`).
|
|
#[tracing::instrument(skip(self))]
|
|
pub fn state_get(
|
|
&self,
|
|
shortstatehash: u64,
|
|
event_type: &StateEventType,
|
|
state_key: &str,
|
|
) -> Result<Option<Arc<PduEvent>>> {
|
|
self.db.state_get(shortstatehash, event_type, state_key)
|
|
}
|
|
|
|
/// Returns the state hash for this pdu.
|
|
pub fn pdu_shortstatehash(&self, event_id: &EventId) -> Result<Option<u64>> {
|
|
self.db.pdu_shortstatehash(event_id)
|
|
}
|
|
|
|
/// Returns the full room state.
|
|
#[tracing::instrument(skip(self))]
|
|
pub async fn room_state_full(
|
|
&self,
|
|
room_id: &RoomId,
|
|
) -> Result<HashMap<(StateEventType, String), Arc<PduEvent>>> {
|
|
self.db.room_state_full(room_id).await
|
|
}
|
|
|
|
/// Returns a single PDU from `room_id` with key (`event_type`, `state_key`).
|
|
#[tracing::instrument(skip(self))]
|
|
pub fn room_state_get_id(
|
|
&self,
|
|
room_id: &RoomId,
|
|
event_type: &StateEventType,
|
|
state_key: &str,
|
|
) -> Result<Option<Arc<EventId>>> {
|
|
self.db.room_state_get_id(room_id, event_type, state_key)
|
|
}
|
|
|
|
/// Returns a single PDU from `room_id` with key (`event_type`, `state_key`).
|
|
#[tracing::instrument(skip(self))]
|
|
pub fn room_state_get(
|
|
&self,
|
|
room_id: &RoomId,
|
|
event_type: &StateEventType,
|
|
state_key: &str,
|
|
) -> Result<Option<Arc<PduEvent>>> {
|
|
self.db.room_state_get(room_id, event_type, state_key)
|
|
}
|
|
}
|