github/darkfi
1
1
Fork 0
mirror of https://github.com/darkrenaissance/darkfi.git synced 2026-04-28 03:00:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

event_graph: gracefully stop pruning task

Browse Source
This commit is contained in:
dasman
2024-07-30 19:00:26 +03:00
parent e701c72888
commit 19f70bd58f
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
bin/darkirc/src/main.rs
View File

@@ -260,6 +260,8 @@ async fn realmain(args: Args, ex: Arc<Executor<'static>>) -> Result<()> {
)
.await?;
let prune_task = event_graph.prune_task.get().unwrap();
info!("Registering EventGraph P2P protocol");
let event_graph_ = Arc::clone(&event_graph);
let registry = p2p.protocol_registry();
@@ -413,6 +415,7 @@ async fn realmain(args: Args, ex: Arc<Executor<'static>>) -> Result<()> {
info!("Stopping IRC server");
irc_task.stop().await;
prune_task.stop().await;
info!("Flushing sled database...");
let flushed_bytes = sled_db.flush_async().await?;

10
src/event_graph/mod.rs
View File

@@ -101,7 +101,7 @@ pub struct EventGraph {
/// `TipRep` message telling peers about our unreferenced tips.
broadcasted_ids: RwLock<HashSet<blake3::Hash>>,
/// DAG Pruning Task
prune_task: OnceCell<StoppableTaskPtr>,
pub prune_task: OnceCell<StoppableTaskPtr>,
/// Event publisher, this notifies whenever an event is
/// inserted into the DAG
pub event_pub: PublisherPtr<Event>,
@@ -167,14 +167,13 @@ impl EventGraph {
// Spawn the DAG pruning task
if days_rotation > 0 {
let self__ = self_.clone();
let prune_task = StoppableTask::new();
let _ = self_.prune_task.set(prune_task.clone()).await;
prune_task.clone().start(
self_.clone().dag_prune_task(days_rotation),
|_| async move {
self__.clone()._handle_stop(sled_db).await;
info!(target: "event_graph::_handle_stop()", "[EVENTGRAPH] Prune task stopped, flushing sled")
},
Error::DetachedTaskStopped,
ex.clone(),
@@ -188,11 +187,6 @@ impl EventGraph {
self.days_rotation
}
async fn _handle_stop(&self, sled_db: sled::Db) {
info!(target: "event_graph::_handle_stop()", "[EVENTGRAPH] Prune task stopped, flushing sled");
sled_db.flush_async().await.unwrap();
}
/// Sync the DAG from connected peers
pub async fn dag_sync(&self) -> Result<()> {
// We do an optimistic sync where we ask all our connected peers for