filip/Stalwart
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Registry crate implementation

Browse Source
This commit is contained in:
mdecimus
2026-01-23 18:44:11 +01:00
parent 2266634f36
commit acecd32c8e
43 changed files with 2258 additions and 912 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
crates/store/src/dispatch/mod.rs
View File

@@ -4,13 +4,12 @@
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
*/
use roaring::RoaringBitmap;
use crate::Store;
use roaring::RoaringBitmap;
pub mod blob;
pub mod lookup;
pub mod pubsub;
pub mod registry;
pub mod search;
pub mod store;

121
crates/store/src/dispatch/pubsub.rs
View File

@@ -1,121 +0,0 @@
/*
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <hello@stalw.art>
*
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
*/
use crate::PubSubStore;
pub enum PubSubStream {
#[cfg(feature = "redis")]
Redis(crate::backend::redis::pubsub::RedisPubSubStream),
#[cfg(feature = "redis")]
RedisCluster(crate::backend::redis::pubsub::RedisClusterPubSubStream),
#[cfg(feature = "nats")]
Nats(crate::backend::nats::pubsub::NatsPubSubStream),
#[cfg(feature = "zenoh")]
Zenoh(crate::backend::zenoh::pubsub::ZenohPubSubStream),
#[cfg(feature = "kafka")]
Kafka(crate::backend::kafka::pubsub::KafkaPubSubStream),
#[cfg(not(any(feature = "redis", feature = "nats")))]
Unimplemented,
}
pub enum Msg {
#[cfg(feature = "redis")]
Redis(redis::Msg),
#[cfg(feature = "nats")]
Nats(async_nats::Message),
#[cfg(feature = "zenoh")]
Zenoh(Vec<u8>),
#[cfg(feature = "kafka")]
Kafka(Vec<u8>),
#[cfg(not(any(feature = "redis", feature = "nats")))]
Unimplemented,
}
#[allow(unused_variables)]
impl PubSubStore {
pub async fn publish(&self, topic: &'static str, message: Vec<u8>) -> trc::Result<()> {
match self {
#[cfg(feature = "redis")]
PubSubStore::Redis(store) => store.publish(topic, message).await,
#[cfg(feature = "nats")]
PubSubStore::Nats(store) => store.publish(topic, message).await,
#[cfg(feature = "zenoh")]
PubSubStore::Zenoh(store) => store.publish(topic, message).await,
#[cfg(feature = "kafka")]
PubSubStore::Kafka(store) => store.publish(topic, message).await,
PubSubStore::None => Err(trc::StoreEvent::NotSupported.into_err()),
}
}
pub async fn subscribe(&self, topic: &'static str) -> trc::Result<PubSubStream> {
match self {
#[cfg(feature = "redis")]
PubSubStore::Redis(store) => store.subscribe(topic).await,
#[cfg(feature = "nats")]
PubSubStore::Nats(store) => store.subscribe(topic).await,
#[cfg(feature = "zenoh")]
PubSubStore::Zenoh(store) => store.subscribe(topic).await,
#[cfg(feature = "kafka")]
PubSubStore::Kafka(store) => store.subscribe(topic).await,
PubSubStore::None => Err(trc::StoreEvent::NotSupported.into_err()),
}
}
pub fn is_none(&self) -> bool {
matches!(self, PubSubStore::None)
}
}
impl PubSubStream {
pub async fn next(&mut self) -> Option<Msg> {
match self {
#[cfg(feature = "redis")]
PubSubStream::Redis(stream) => stream.next().await,
#[cfg(feature = "redis")]
PubSubStream::RedisCluster(stream) => stream.next().await,
#[cfg(feature = "nats")]
PubSubStream::Nats(stream) => stream.next().await,
#[cfg(feature = "zenoh")]
PubSubStream::Zenoh(stream) => stream.next().await,
#[cfg(feature = "kafka")]
PubSubStream::Kafka(stream) => stream.next().await,
#[cfg(not(any(feature = "redis", feature = "nats")))]
PubSubStream::Unimplemented => None,
}
}
}
impl Msg {
pub fn payload(&self) -> &[u8] {
match self {
#[cfg(feature = "redis")]
Msg::Redis(msg) => msg.get_payload_bytes(),
#[cfg(feature = "nats")]
Msg::Nats(msg) => msg.payload.as_ref(),
#[cfg(feature = "zenoh")]
Msg::Zenoh(msg) => msg.as_slice(),
#[cfg(feature = "kafka")]
Msg::Kafka(msg) => msg.as_slice(),
#[cfg(not(any(feature = "redis", feature = "nats")))]
Msg::Unimplemented => &[],
}
}
pub fn topic(&self) -> &str {
match self {
#[cfg(feature = "redis")]
Msg::Redis(msg) => msg.get_channel_name(),
#[cfg(feature = "nats")]
Msg::Nats(msg) => msg.subject.as_str(),
#[cfg(feature = "zenoh")]
Msg::Zenoh(_) => "",
#[cfg(feature = "kafka")]
Msg::Kafka(_) => "",
#[cfg(not(any(feature = "redis", feature = "nats")))]
Msg::Unimplemented => "",
}
}
}

23
crates/store/src/dispatch/registry.rs Normal file
View File

@@ -0,0 +1,23 @@
/*
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <hello@stalw.art>
*
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
*/
use registry::{schema::prelude::Registry, types::id::Id};
use crate::RegistryStore;
impl RegistryStore {
pub async fn get(&self, id: Id) -> trc::Result<Option<Registry>> {
todo!()
}
pub async fn get_or_default(&self, id: Id) -> trc::Result<Registry> {
todo!()
}
pub async fn delete(&self, id: Id) -> trc::Result<()> {
todo!()
}
}