daemon: incorporate update monitor in daemon activities
This commit is contained in:
@@ -30,6 +30,9 @@ use kordophone::api::{
|
||||
AuthenticationStore,
|
||||
};
|
||||
|
||||
mod update_monitor;
|
||||
use update_monitor::UpdateMonitor;
|
||||
|
||||
#[derive(Debug, Error)]
|
||||
pub enum DaemonError {
|
||||
#[error("Client Not Configured")]
|
||||
@@ -87,11 +90,11 @@ impl AuthenticationStore for DatabaseAuthenticationStore {
|
||||
}
|
||||
}
|
||||
|
||||
mod target {
|
||||
pub mod target {
|
||||
pub static SYNC: &str = "sync";
|
||||
pub static EVENT: &str = "event";
|
||||
pub static UPDATES: &str = "updates";
|
||||
}
|
||||
|
||||
pub struct Daemon {
|
||||
pub event_sender: Sender<Event>,
|
||||
event_receiver: Receiver<Event>,
|
||||
@@ -139,6 +142,13 @@ impl Daemon {
|
||||
log::info!("Starting daemon version {}", self.version);
|
||||
log::debug!("Debug logging enabled.");
|
||||
|
||||
let mut update_monitor = UpdateMonitor::new(self.database.clone(), self.event_sender.clone());
|
||||
|
||||
tokio::spawn(async move {
|
||||
log::info!(target: target::UPDATES, "Starting update monitor");
|
||||
update_monitor.run().await; // should run indefinitely
|
||||
});
|
||||
|
||||
while let Some(event) = self.event_receiver.recv().await {
|
||||
log::debug!(target: target::EVENT, "Received event: {:?}", event);
|
||||
self.handle_event(event).await;
|
||||
@@ -151,6 +161,20 @@ impl Daemon {
|
||||
reply.send(self.version.clone()).unwrap();
|
||||
},
|
||||
|
||||
Event::SyncConversationList(reply) => {
|
||||
let mut db_clone = self.database.clone();
|
||||
let signal_sender = self.signal_sender.clone();
|
||||
self.runtime.spawn(async move {
|
||||
let result = Self::sync_conversation_list(&mut db_clone, &signal_sender).await;
|
||||
if let Err(e) = result {
|
||||
log::error!("Error handling sync event: {}", e);
|
||||
}
|
||||
});
|
||||
|
||||
// This is a background operation, so return right away.
|
||||
reply.send(()).unwrap();
|
||||
},
|
||||
|
||||
Event::SyncAllConversations(reply) => {
|
||||
let mut db_clone = self.database.clone();
|
||||
let signal_sender = self.signal_sender.clone();
|
||||
@@ -231,8 +255,32 @@ impl Daemon {
|
||||
self.database.lock().await.with_repository(|r| r.get_messages_for_conversation(&conversation_id).unwrap()).await
|
||||
}
|
||||
|
||||
async fn sync_conversation_list(database: &mut Arc<Mutex<Database>>, signal_sender: &Sender<Signal>) -> Result<()> {
|
||||
log::info!(target: target::SYNC, "Starting list conversation sync");
|
||||
|
||||
let mut client = Self::get_client_impl(database).await?;
|
||||
|
||||
// Fetch conversations from server
|
||||
let fetched_conversations = client.get_conversations().await?;
|
||||
let db_conversations: Vec<kordophone_db::models::Conversation> = fetched_conversations.into_iter()
|
||||
.map(kordophone_db::models::Conversation::from)
|
||||
.collect();
|
||||
|
||||
// Insert each conversation
|
||||
let num_conversations = db_conversations.len();
|
||||
for conversation in db_conversations {
|
||||
database.with_repository(|r| r.insert_conversation(conversation)).await?;
|
||||
}
|
||||
|
||||
// Send conversations updated signal
|
||||
signal_sender.send(Signal::ConversationsUpdated).await?;
|
||||
|
||||
log::info!(target: target::SYNC, "Synchronized {} conversations", num_conversations);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn sync_all_conversations_impl(database: &mut Arc<Mutex<Database>>, signal_sender: &Sender<Signal>) -> Result<()> {
|
||||
log::info!(target: target::SYNC, "Starting conversation sync");
|
||||
log::info!(target: target::SYNC, "Starting full conversation sync");
|
||||
|
||||
let mut client = Self::get_client_impl(database).await?;
|
||||
|
||||
@@ -266,6 +314,13 @@ impl Daemon {
|
||||
|
||||
let mut client = Self::get_client_impl(database).await?;
|
||||
|
||||
// Check if conversation exists in database.
|
||||
let conversation = database.with_repository(|r| r.get_conversation_by_guid(&conversation_id)).await?;
|
||||
if conversation.is_none() {
|
||||
// If the conversation doesn't exist, first do a conversation list sync.
|
||||
Self::sync_conversation_list(database, signal_sender).await?;
|
||||
}
|
||||
|
||||
// Fetch and sync messages for this conversation
|
||||
let last_message_id = database.with_repository(|r| -> Option<String> {
|
||||
r.get_last_message_for_conversation(&conversation_id)
|
||||
|
||||
Reference in New Issue
Block a user