Lines
98.82 %
Functions
43.33 %
Branches
100 %
use std::sync::Arc;
use async_trait::async_trait;
use bonsaidb_core::{
arc_bytes::serde::Bytes,
networking::{CreateSubscriber, Publish, PublishToAll, SubscribeTo, UnsubscribeFrom},
pubsub::{AsyncPubSub, AsyncSubscriber, Receiver},
};
use crate::Client;
#[async_trait]
impl AsyncPubSub for super::RemoteDatabase {
type Subscriber = RemoteSubscriber;
async fn create_subscriber(&self) -> Result<Self::Subscriber, bonsaidb_core::Error> {
let subscriber_id = self
.client
.send_api_request_async(&CreateSubscriber {
database: self.name.to_string(),
})
.await?;
let (sender, receiver) = flume::unbounded();
self.client.register_subscriber(subscriber_id, sender);
Ok(RemoteSubscriber {
client: self.client.clone(),
database: self.name.clone(),
id: subscriber_id,
receiver: Receiver::new(receiver),
#[cfg(not(target_arch = "wasm32"))]
tokio: tokio::runtime::Handle::try_current().ok().map(Arc::new),
}
async fn publish_bytes(
&self,
topic: Vec<u8>,
payload: Vec<u8>,
) -> Result<(), bonsaidb_core::Error> {
self.client
.send_api_request_async(&Publish {
topic: Bytes::from(topic),
payload: Bytes::from(payload),
Ok(())
async fn publish_bytes_to_all(
topics: impl IntoIterator<Item = Vec<u8>> + Send + 'async_trait,
let topics = topics.into_iter().map(Bytes::from).collect();
.send_api_request_async(&PublishToAll {
topics,
/// A `PubSub` subscriber from a remote server.
#[derive(Debug)]
pub struct RemoteSubscriber {
pub(crate) client: Client,
pub(crate) database: Arc<String>,
pub(crate) id: u64,
pub(crate) receiver: Receiver,
pub(crate) tokio: Option<Arc<tokio::runtime::Handle>>,
impl AsyncSubscriber for RemoteSubscriber {
async fn subscribe_to_bytes(&self, topic: Vec<u8>) -> Result<(), bonsaidb_core::Error> {
.send_api_request_async(&SubscribeTo {
database: self.database.to_string(),
subscriber_id: self.id,
async fn unsubscribe_from_bytes(&self, topic: &[u8]) -> Result<(), bonsaidb_core::Error> {
.send_api_request_async(&UnsubscribeFrom {
fn receiver(&self) -> &Receiver {
&self.receiver
#[cfg(target_arch = "wasm32")]
impl Drop for RemoteSubscriber {
fn drop(&mut self) {
let client = self.client.clone();
let database = self.database.to_string();
let subscriber_id = self.id;
let drop_future = async move {
client
.unregister_subscriber_async(database, subscriber_id)
.await;
wasm_bindgen_futures::spawn_local(drop_future);
if let Some(tokio) = &self.tokio {
tokio.spawn(async move {
});
} else {
.unregister_subscriber(self.database.to_string(), self.id);