Nathan.fooo 7831d8d4ab
refactor: database row and cell notification (#5237)
* refactor: database row and cell notification

* chore: clippy

* chore: fix test
2024-05-02 11:42:33 +08:00

40 lines
1.3 KiB
Rust

use std::sync::RwLock;
use crate::entities::SubscribeObject;
use lazy_static::lazy_static;
mod builder;
pub use builder::*;
mod debounce;
pub use debounce::*;
pub mod entities;
mod protobuf;
lazy_static! {
static ref NOTIFICATION_SENDER: RwLock<Vec<Box<dyn NotificationSender>>> = RwLock::new(vec![]);
}
/// Register a notification sender. The sender will be alive until the process exits.
/// Flutter integration test or Tauri hot reload might cause register multiple times.
/// So before register a new sender, you might need to unregister the old one. Currently,
/// Just remove all senders by calling `unregister_all_notification_sender`.
pub fn register_notification_sender<T: NotificationSender>(sender: T) {
let box_sender = Box::new(sender);
match NOTIFICATION_SENDER.write() {
Ok(mut write_guard) => write_guard.push(box_sender),
Err(err) => tracing::error!("Failed to push notification sender: {:?}", err),
}
}
pub fn unregister_all_notification_sender() {
match NOTIFICATION_SENDER.write() {
Ok(mut write_guard) => write_guard.clear(),
Err(err) => tracing::error!("Failed to remove all notification senders: {:?}", err),
}
}
pub trait NotificationSender: Send + Sync + 'static {
fn send_subject(&self, subject: SubscribeObject) -> Result<(), String>;
}