ref: async trait functions

Cargo.toml

@@ -5,6 +5,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+async-trait = "0.1.73"
 clap = { version = "4.4.0", features = ["string"] }
 ethers = "1"
 eyre = "0.6.8"

src/main.rs

@@ -73,7 +73,10 @@ async fn main() -> Result<()> {
                 let fetcher = L1Fetcher::new(http_url)?;
                 let processor = TreeProcessor::new(&db_dir)?;
                 let (tx, rx) = mpsc::channel::<Vec<CommitBlockInfoV1>>(5);
-                processor.run(rx);
+
+                tokio::spawn(async move {
+                    processor.run(rx).await;
+                });
 
                 fetcher.fetch(tx, Some(U64([*start_block])), None).await?;
             }

src/processor/mod.rs

@@ -1,9 +1,11 @@
-use tokio::{sync::mpsc, task::JoinHandle};
+use async_trait::async_trait;
+use tokio::sync::mpsc;
 
 use crate::types::CommitBlockInfoV1;
 
 pub mod tree;
 
+#[async_trait]
 pub trait Processor {
-    fn run(self, rx: mpsc::Receiver<Vec<CommitBlockInfoV1>>) -> JoinHandle<()>;
+    async fn run(self, rx: mpsc::Receiver<Vec<CommitBlockInfoV1>>);
 }

src/processor/tree/mod.rs

@@ -3,6 +3,7 @@ mod tree_wrapper;
 
 use std::path::Path;
 
+use async_trait::async_trait;
 use eyre::Result;
 use tokio::{sync::mpsc, task::JoinHandle};
 
@@ -44,9 +45,9 @@ impl TreeProcessor<'static> {
     }
 }
 
+#[async_trait]
 impl Processor for TreeProcessor<'static> {
-    fn run(mut self, mut rx: mpsc::Receiver<Vec<CommitBlockInfoV1>>) -> JoinHandle<()> {
-        tokio::spawn(async move {
+    async fn run(mut self, mut rx: mpsc::Receiver<Vec<CommitBlockInfoV1>>) {
             while let Some(blocks) = rx.recv().await {
                 for block in blocks {
                     // Check if we've already processed this block.