wip

Author: kariy

Cargo.lock

@@ -27,3 +27,6 @@ parking_lot.workspace = true
 tokio.workspace = true
 tracing.workspace = true
 url.workspace = true
+
+[dev-dependencies]
+jsonrpsee.workspace = true

crates/sharding/Cargo.toml

@@ -101,6 +101,11 @@ impl Scheduler {
         self.inner.time_quantum
     }
 
+    /// Returns the current number of scheduled shard tasks in the queue.
+    pub fn queue_len(&self) -> usize {
+        self.inner.queue.lock().len()
+    }
+
     /// Signal shutdown and wake all waiting workers.
     pub fn shutdown(&self) {
         self.inner.shutdown.store(true, Ordering::SeqCst);

crates/sharding/src/scheduler.rs

@@ -2,14 +2,10 @@ use std::time::Instant;
 
 use anyhow::Result;
 use katana_pool::TransactionPool;
-use katana_primitives::env::BlockEnv;
-use katana_primitives::transaction::ExecutableTxWithHash;
-use katana_provider::api::state::StateFactoryProvider;
-use katana_provider::ProviderFactory;
 use tracing::{error, trace};
 
 use crate::scheduler::Scheduler;
-use crate::shard::{Shard, ShardState};
+use crate::shard::ShardState;
 
 /// A worker that picks shards from the scheduler and executes their pending transactions.
 ///