042e80570c
* Add logging on shutdown * Replace tokio::spawn with handle.spawn * Upgrade tokio * Add a task executor * Beacon chain tasks use task executor * Validator client tasks use task executor * Rename runtime_handle to executor * Add duration histograms; minor fixes * Cleanup * Fix logs * Fix tests * Remove random file * Get enr dependency instead of libp2p * Address some review comments * Libp2p takes a TaskExecutor * Ugly fix libp2p tests * Move TaskExecutor to own file * Upgrade Dockerfile rust version * Minor fixes * Revert "Ugly fix libp2p tests" This reverts commit 58d4bb690f52de28d893943b7504d2d0c6621429. * Pretty fix libp2p tests * Add spawn_without_exit; change Counter to Gauge * Tidy * Move log from RuntimeContext to TaskExecutor * Fix errors * Replace histogram with int_gauge for async tasks * Fix todo * Fix memory leak in test by exiting all spawned tasks at the end
39 lines
1.2 KiB
Rust
39 lines
1.2 KiB
Rust
//! A timer service for the beacon node.
|
|
//!
|
|
//! This service allows task execution on the beacon node for various functionality.
|
|
|
|
use beacon_chain::{BeaconChain, BeaconChainTypes};
|
|
use futures::stream::StreamExt;
|
|
use slog::info;
|
|
use slot_clock::SlotClock;
|
|
use std::sync::Arc;
|
|
use std::time::Duration;
|
|
use tokio::time::{interval_at, Instant};
|
|
|
|
/// Spawns a timer service which periodically executes tasks for the beacon chain
|
|
pub fn spawn_timer<T: BeaconChainTypes>(
|
|
executor: environment::TaskExecutor,
|
|
beacon_chain: Arc<BeaconChain<T>>,
|
|
milliseconds_per_slot: u64,
|
|
) -> Result<(), &'static str> {
|
|
let log = executor.log();
|
|
let start_instant = Instant::now()
|
|
+ beacon_chain
|
|
.slot_clock
|
|
.duration_to_next_slot()
|
|
.ok_or_else(|| "slot_notifier unable to determine time to next slot")?;
|
|
|
|
// Warning: `interval_at` panics if `milliseconds_per_slot` = 0.
|
|
let mut interval = interval_at(start_instant, Duration::from_millis(milliseconds_per_slot));
|
|
let timer_future = async move {
|
|
while interval.next().await.is_some() {
|
|
beacon_chain.per_slot_task();
|
|
}
|
|
};
|
|
|
|
executor.spawn(timer_future, "timer");
|
|
info!(log, "Timer service started");
|
|
|
|
Ok(())
|
|
}
|