mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-07-22 14:55:41 +00:00
Runtime diagnostics for leaked messages in unbounded channels (#12971)
This commit is contained in:
Generated
+2
@@ -7948,6 +7948,7 @@ dependencies = [
|
|||||||
"pin-project",
|
"pin-project",
|
||||||
"sc-network-common",
|
"sc-network-common",
|
||||||
"sc-peerset",
|
"sc-peerset",
|
||||||
|
"sc-utils",
|
||||||
"sp-consensus",
|
"sp-consensus",
|
||||||
"sp-runtime",
|
"sp-runtime",
|
||||||
"substrate-prometheus-endpoint",
|
"substrate-prometheus-endpoint",
|
||||||
@@ -8284,6 +8285,7 @@ dependencies = [
|
|||||||
"parking_lot 0.12.1",
|
"parking_lot 0.12.1",
|
||||||
"pin-project",
|
"pin-project",
|
||||||
"rand 0.8.5",
|
"rand 0.8.5",
|
||||||
|
"sc-utils",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"thiserror",
|
"thiserror",
|
||||||
|
|||||||
@@ -144,7 +144,9 @@ impl<Block: BlockT> StorageNotifications<Block> {
|
|||||||
filter_keys: Option<&[StorageKey]>,
|
filter_keys: Option<&[StorageKey]>,
|
||||||
filter_child_keys: Option<&[(StorageKey, Option<Vec<StorageKey>>)]>,
|
filter_child_keys: Option<&[(StorageKey, Option<Vec<StorageKey>>)]>,
|
||||||
) -> StorageEventStream<Block::Hash> {
|
) -> StorageEventStream<Block::Hash> {
|
||||||
let receiver = self.0.subscribe(registry::SubscribeOp { filter_keys, filter_child_keys });
|
let receiver = self
|
||||||
|
.0
|
||||||
|
.subscribe(registry::SubscribeOp { filter_keys, filter_child_keys }, 100_000);
|
||||||
|
|
||||||
StorageEventStream(receiver)
|
StorageEventStream(receiver)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -120,7 +120,7 @@ where
|
|||||||
) -> Result<Self, Error> {
|
) -> Result<Self, Error> {
|
||||||
let beefy_best_block = Arc::new(RwLock::new(None));
|
let beefy_best_block = Arc::new(RwLock::new(None));
|
||||||
|
|
||||||
let stream = best_block_stream.subscribe();
|
let stream = best_block_stream.subscribe(100_000);
|
||||||
let closure_clone = beefy_best_block.clone();
|
let closure_clone = beefy_best_block.clone();
|
||||||
let future = stream.for_each(move |best_beefy| {
|
let future = stream.for_each(move |best_beefy| {
|
||||||
let async_clone = closure_clone.clone();
|
let async_clone = closure_clone.clone();
|
||||||
@@ -141,7 +141,7 @@ where
|
|||||||
fn subscribe_justifications(&self, mut sink: SubscriptionSink) -> SubscriptionResult {
|
fn subscribe_justifications(&self, mut sink: SubscriptionSink) -> SubscriptionResult {
|
||||||
let stream = self
|
let stream = self
|
||||||
.finality_proof_stream
|
.finality_proof_stream
|
||||||
.subscribe()
|
.subscribe(100_000)
|
||||||
.map(|vfp| notification::EncodedVersionedFinalityProof::new::<Block>(vfp));
|
.map(|vfp| notification::EncodedVersionedFinalityProof::new::<Block>(vfp));
|
||||||
|
|
||||||
let fut = async move {
|
let fut = async move {
|
||||||
|
|||||||
@@ -265,7 +265,7 @@ where
|
|||||||
// Subscribe to finality notifications and justifications before waiting for runtime pallet and
|
// Subscribe to finality notifications and justifications before waiting for runtime pallet and
|
||||||
// reuse the streams, so we don't miss notifications while waiting for pallet to be available.
|
// reuse the streams, so we don't miss notifications while waiting for pallet to be available.
|
||||||
let mut finality_notifications = client.finality_notification_stream().fuse();
|
let mut finality_notifications = client.finality_notification_stream().fuse();
|
||||||
let block_import_justif = links.from_block_import_justif_stream.subscribe().fuse();
|
let block_import_justif = links.from_block_import_justif_stream.subscribe(100_000).fuse();
|
||||||
|
|
||||||
// Wait for BEEFY pallet to be active before starting voter.
|
// Wait for BEEFY pallet to be active before starting voter.
|
||||||
let persisted_state =
|
let persisted_state =
|
||||||
|
|||||||
@@ -430,8 +430,8 @@ pub(crate) fn get_beefy_streams(
|
|||||||
let beefy_rpc_links = net.peer(index).data.beefy_rpc_links.lock().clone().unwrap();
|
let beefy_rpc_links = net.peer(index).data.beefy_rpc_links.lock().clone().unwrap();
|
||||||
let BeefyRPCLinks { from_voter_justif_stream, from_voter_best_beefy_stream } =
|
let BeefyRPCLinks { from_voter_justif_stream, from_voter_best_beefy_stream } =
|
||||||
beefy_rpc_links;
|
beefy_rpc_links;
|
||||||
best_block_streams.push(from_voter_best_beefy_stream.subscribe());
|
best_block_streams.push(from_voter_best_beefy_stream.subscribe(100_000));
|
||||||
versioned_finality_proof_streams.push(from_voter_justif_stream.subscribe());
|
versioned_finality_proof_streams.push(from_voter_justif_stream.subscribe(100_000));
|
||||||
});
|
});
|
||||||
(best_block_streams, versioned_finality_proof_streams)
|
(best_block_streams, versioned_finality_proof_streams)
|
||||||
}
|
}
|
||||||
@@ -736,7 +736,7 @@ async fn beefy_importing_blocks() {
|
|||||||
let hashof1 = block.header.hash();
|
let hashof1 = block.header.hash();
|
||||||
|
|
||||||
// Import without justifications.
|
// Import without justifications.
|
||||||
let mut justif_recv = justif_stream.subscribe();
|
let mut justif_recv = justif_stream.subscribe(100_000);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
block_import
|
block_import
|
||||||
.import_block(params(block.clone(), None), HashMap::new())
|
.import_block(params(block.clone(), None), HashMap::new())
|
||||||
@@ -779,7 +779,7 @@ async fn beefy_importing_blocks() {
|
|||||||
let builder = full_client.new_block_at(&parent_id, Default::default(), false).unwrap();
|
let builder = full_client.new_block_at(&parent_id, Default::default(), false).unwrap();
|
||||||
let block = builder.build().unwrap().block;
|
let block = builder.build().unwrap().block;
|
||||||
let hashof2 = block.header.hash();
|
let hashof2 = block.header.hash();
|
||||||
let mut justif_recv = justif_stream.subscribe();
|
let mut justif_recv = justif_stream.subscribe(100_000);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
block_import.import_block(params(block, justif), HashMap::new()).await.unwrap(),
|
block_import.import_block(params(block, justif), HashMap::new()).await.unwrap(),
|
||||||
ImportResult::Imported(ImportedAux {
|
ImportResult::Imported(ImportedAux {
|
||||||
@@ -823,7 +823,7 @@ async fn beefy_importing_blocks() {
|
|||||||
let builder = full_client.new_block_at(&parent_id, Default::default(), false).unwrap();
|
let builder = full_client.new_block_at(&parent_id, Default::default(), false).unwrap();
|
||||||
let block = builder.build().unwrap().block;
|
let block = builder.build().unwrap().block;
|
||||||
let hashof3 = block.header.hash();
|
let hashof3 = block.header.hash();
|
||||||
let mut justif_recv = justif_stream.subscribe();
|
let mut justif_recv = justif_stream.subscribe(100_000);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
block_import.import_block(params(block, justif), HashMap::new()).await.unwrap(),
|
block_import.import_block(params(block, justif), HashMap::new()).await.unwrap(),
|
||||||
ImportResult::Imported(ImportedAux {
|
ImportResult::Imported(ImportedAux {
|
||||||
|
|||||||
@@ -69,7 +69,7 @@ impl<B: BlockT, Transaction: Send + 'static> BasicQueue<B, Transaction> {
|
|||||||
spawner: &impl sp_core::traits::SpawnEssentialNamed,
|
spawner: &impl sp_core::traits::SpawnEssentialNamed,
|
||||||
prometheus_registry: Option<&Registry>,
|
prometheus_registry: Option<&Registry>,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
let (result_sender, result_port) = buffered_link::buffered_link();
|
let (result_sender, result_port) = buffered_link::buffered_link(100_000);
|
||||||
|
|
||||||
let metrics = prometheus_registry.and_then(|r| {
|
let metrics = prometheus_registry.and_then(|r| {
|
||||||
Metrics::register(r)
|
Metrics::register(r)
|
||||||
@@ -276,10 +276,10 @@ impl<B: BlockT> BlockImportWorker<B> {
|
|||||||
use worker_messages::*;
|
use worker_messages::*;
|
||||||
|
|
||||||
let (justification_sender, mut justification_port) =
|
let (justification_sender, mut justification_port) =
|
||||||
tracing_unbounded("mpsc_import_queue_worker_justification");
|
tracing_unbounded("mpsc_import_queue_worker_justification", 100_000);
|
||||||
|
|
||||||
let (block_import_sender, block_import_port) =
|
let (block_import_sender, block_import_port) =
|
||||||
tracing_unbounded("mpsc_import_queue_worker_blocks");
|
tracing_unbounded("mpsc_import_queue_worker_blocks", 100_000);
|
||||||
|
|
||||||
let mut worker = BlockImportWorker { result_sender, justification_import, metrics };
|
let mut worker = BlockImportWorker { result_sender, justification_import, metrics };
|
||||||
|
|
||||||
@@ -595,7 +595,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn prioritizes_finality_work_over_block_import() {
|
fn prioritizes_finality_work_over_block_import() {
|
||||||
let (result_sender, mut result_port) = buffered_link::buffered_link();
|
let (result_sender, mut result_port) = buffered_link::buffered_link(100_000);
|
||||||
|
|
||||||
let (worker, mut finality_sender, mut block_import_sender) =
|
let (worker, mut finality_sender, mut block_import_sender) =
|
||||||
BlockImportWorker::new(result_sender, (), Box::new(()), Some(Box::new(())), None);
|
BlockImportWorker::new(result_sender, (), Box::new(()), Some(Box::new(())), None);
|
||||||
|
|||||||
@@ -28,7 +28,7 @@
|
|||||||
//! # use sp_test_primitives::Block;
|
//! # use sp_test_primitives::Block;
|
||||||
//! # struct DummyLink; impl Link<Block> for DummyLink {}
|
//! # struct DummyLink; impl Link<Block> for DummyLink {}
|
||||||
//! # let mut my_link = DummyLink;
|
//! # let mut my_link = DummyLink;
|
||||||
//! let (mut tx, mut rx) = buffered_link::<Block>();
|
//! let (mut tx, mut rx) = buffered_link::<Block>(100_000);
|
||||||
//! tx.blocks_processed(0, 0, vec![]);
|
//! tx.blocks_processed(0, 0, vec![]);
|
||||||
//!
|
//!
|
||||||
//! // Calls `my_link.blocks_processed(0, 0, vec![])` when polled.
|
//! // Calls `my_link.blocks_processed(0, 0, vec![])` when polled.
|
||||||
@@ -51,9 +51,11 @@ use super::BlockImportResult;
|
|||||||
|
|
||||||
/// Wraps around an unbounded channel from the `futures` crate. The sender implements `Link` and
|
/// Wraps around an unbounded channel from the `futures` crate. The sender implements `Link` and
|
||||||
/// can be used to buffer commands, and the receiver can be used to poll said commands and transfer
|
/// can be used to buffer commands, and the receiver can be used to poll said commands and transfer
|
||||||
/// them to another link.
|
/// them to another link. `queue_size_warning` sets the warning threshold of the channel queue size.
|
||||||
pub fn buffered_link<B: BlockT>() -> (BufferedLinkSender<B>, BufferedLinkReceiver<B>) {
|
pub fn buffered_link<B: BlockT>(
|
||||||
let (tx, rx) = tracing_unbounded("mpsc_buffered_link");
|
queue_size_warning: i64,
|
||||||
|
) -> (BufferedLinkSender<B>, BufferedLinkReceiver<B>) {
|
||||||
|
let (tx, rx) = tracing_unbounded("mpsc_buffered_link", queue_size_warning);
|
||||||
let tx = BufferedLinkSender { tx };
|
let tx = BufferedLinkSender { tx };
|
||||||
let rx = BufferedLinkReceiver { rx: rx.fuse() };
|
let rx = BufferedLinkReceiver { rx: rx.fuse() };
|
||||||
(tx, rx)
|
(tx, rx)
|
||||||
@@ -175,7 +177,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn is_closed() {
|
fn is_closed() {
|
||||||
let (tx, rx) = super::buffered_link::<Block>();
|
let (tx, rx) = super::buffered_link::<Block>(1);
|
||||||
assert!(!tx.is_closed());
|
assert!(!tx.is_closed());
|
||||||
drop(rx);
|
drop(rx);
|
||||||
assert!(tx.is_closed());
|
assert!(tx.is_closed());
|
||||||
|
|||||||
@@ -104,7 +104,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn subscribe_justifications(&self, mut sink: SubscriptionSink) -> SubscriptionResult {
|
fn subscribe_justifications(&self, mut sink: SubscriptionSink) -> SubscriptionResult {
|
||||||
let stream = self.justification_stream.subscribe().map(
|
let stream = self.justification_stream.subscribe(100_000).map(
|
||||||
|x: sc_finality_grandpa::GrandpaJustification<Block>| {
|
|x: sc_finality_grandpa::GrandpaJustification<Block>| {
|
||||||
JustificationNotification::from(x)
|
JustificationNotification::from(x)
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -1364,7 +1364,7 @@ impl<Block: BlockT> GossipValidator<Block> {
|
|||||||
None => None,
|
None => None,
|
||||||
};
|
};
|
||||||
|
|
||||||
let (tx, rx) = tracing_unbounded("mpsc_grandpa_gossip_validator");
|
let (tx, rx) = tracing_unbounded("mpsc_grandpa_gossip_validator", 100_000);
|
||||||
let val = GossipValidator {
|
let val = GossipValidator {
|
||||||
inner: parking_lot::RwLock::new(Inner::new(config)),
|
inner: parking_lot::RwLock::new(Inner::new(config)),
|
||||||
set_state,
|
set_state,
|
||||||
|
|||||||
@@ -70,6 +70,7 @@ impl<B: BlockT> NeighborPacketWorker<B> {
|
|||||||
pub(super) fn new(rebroadcast_period: Duration) -> (Self, NeighborPacketSender<B>) {
|
pub(super) fn new(rebroadcast_period: Duration) -> (Self, NeighborPacketSender<B>) {
|
||||||
let (tx, rx) = tracing_unbounded::<(Vec<PeerId>, NeighborPacket<NumberFor<B>>)>(
|
let (tx, rx) = tracing_unbounded::<(Vec<PeerId>, NeighborPacket<NumberFor<B>>)>(
|
||||||
"mpsc_grandpa_neighbor_packet_worker",
|
"mpsc_grandpa_neighbor_packet_worker",
|
||||||
|
100_000,
|
||||||
);
|
);
|
||||||
let delay = Delay::new(rebroadcast_period);
|
let delay = Delay::new(rebroadcast_period);
|
||||||
|
|
||||||
|
|||||||
@@ -135,7 +135,7 @@ impl NetworkEventStream for TestNetwork {
|
|||||||
&self,
|
&self,
|
||||||
_name: &'static str,
|
_name: &'static str,
|
||||||
) -> Pin<Box<dyn Stream<Item = NetworkEvent> + Send>> {
|
) -> Pin<Box<dyn Stream<Item = NetworkEvent> + Send>> {
|
||||||
let (tx, rx) = tracing_unbounded("test");
|
let (tx, rx) = tracing_unbounded("test", 100_000);
|
||||||
let _ = self.sender.unbounded_send(Event::EventStream(tx));
|
let _ = self.sender.unbounded_send(Event::EventStream(tx));
|
||||||
Box::pin(rx)
|
Box::pin(rx)
|
||||||
}
|
}
|
||||||
@@ -253,7 +253,7 @@ fn voter_set_state() -> SharedVoterSetState<Block> {
|
|||||||
|
|
||||||
// needs to run in a tokio runtime.
|
// needs to run in a tokio runtime.
|
||||||
pub(crate) fn make_test_network() -> (impl Future<Output = Tester>, TestNetwork) {
|
pub(crate) fn make_test_network() -> (impl Future<Output = Tester>, TestNetwork) {
|
||||||
let (tx, rx) = tracing_unbounded("test");
|
let (tx, rx) = tracing_unbounded("test", 100_000);
|
||||||
let net = TestNetwork { sender: tx };
|
let net = TestNetwork { sender: tx };
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
|
|||||||
@@ -566,7 +566,8 @@ where
|
|||||||
}
|
}
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
let (voter_commands_tx, voter_commands_rx) = tracing_unbounded("mpsc_grandpa_voter_command");
|
let (voter_commands_tx, voter_commands_rx) =
|
||||||
|
tracing_unbounded("mpsc_grandpa_voter_command", 100_000);
|
||||||
|
|
||||||
let (justification_sender, justification_stream) = GrandpaJustificationStream::channel();
|
let (justification_sender, justification_stream) = GrandpaJustificationStream::channel();
|
||||||
|
|
||||||
|
|||||||
@@ -437,7 +437,7 @@ mod tests {
|
|||||||
aux_schema::load_persistent(&*backend, client.info().genesis_hash, 0, || Ok(voters))
|
aux_schema::load_persistent(&*backend, client.info().genesis_hash, 0, || Ok(voters))
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
let (_tx, voter_command_rx) = tracing_unbounded("");
|
let (_tx, voter_command_rx) = tracing_unbounded("test_mpsc_voter_command", 100_000);
|
||||||
|
|
||||||
let observer = ObserverWork::new(
|
let observer = ObserverWork::new(
|
||||||
client,
|
client,
|
||||||
|
|||||||
@@ -579,7 +579,7 @@ mod tests {
|
|||||||
|
|
||||||
impl TestChainState {
|
impl TestChainState {
|
||||||
fn new() -> (Self, ImportNotifications<Block>) {
|
fn new() -> (Self, ImportNotifications<Block>) {
|
||||||
let (tx, rx) = tracing_unbounded("test");
|
let (tx, rx) = tracing_unbounded("test", 100_000);
|
||||||
let state =
|
let state =
|
||||||
TestChainState { sender: tx, known_blocks: Arc::new(Mutex::new(HashMap::new())) };
|
TestChainState { sender: tx, known_blocks: Arc::new(Mutex::new(HashMap::new())) };
|
||||||
|
|
||||||
@@ -680,7 +680,7 @@ mod tests {
|
|||||||
// enact all dependencies before importing the message
|
// enact all dependencies before importing the message
|
||||||
enact_dependencies(&chain_state);
|
enact_dependencies(&chain_state);
|
||||||
|
|
||||||
let (global_tx, global_rx) = tracing_unbounded("test");
|
let (global_tx, global_rx) = tracing_unbounded("test", 100_000);
|
||||||
|
|
||||||
let until_imported = UntilGlobalMessageBlocksImported::new(
|
let until_imported = UntilGlobalMessageBlocksImported::new(
|
||||||
import_notifications,
|
import_notifications,
|
||||||
@@ -708,7 +708,7 @@ mod tests {
|
|||||||
let (chain_state, import_notifications) = TestChainState::new();
|
let (chain_state, import_notifications) = TestChainState::new();
|
||||||
let block_status = chain_state.block_status();
|
let block_status = chain_state.block_status();
|
||||||
|
|
||||||
let (global_tx, global_rx) = tracing_unbounded("test");
|
let (global_tx, global_rx) = tracing_unbounded("test", 100_000);
|
||||||
|
|
||||||
let until_imported = UntilGlobalMessageBlocksImported::new(
|
let until_imported = UntilGlobalMessageBlocksImported::new(
|
||||||
import_notifications,
|
import_notifications,
|
||||||
@@ -896,7 +896,7 @@ mod tests {
|
|||||||
let (chain_state, import_notifications) = TestChainState::new();
|
let (chain_state, import_notifications) = TestChainState::new();
|
||||||
let block_status = chain_state.block_status();
|
let block_status = chain_state.block_status();
|
||||||
|
|
||||||
let (global_tx, global_rx) = tracing_unbounded("test");
|
let (global_tx, global_rx) = tracing_unbounded("test", 100_000);
|
||||||
|
|
||||||
let block_sync_requester = TestBlockSyncRequester::default();
|
let block_sync_requester = TestBlockSyncRequester::default();
|
||||||
|
|
||||||
|
|||||||
@@ -208,7 +208,7 @@ where
|
|||||||
¶ms.network_config.transport,
|
¶ms.network_config.transport,
|
||||||
)?;
|
)?;
|
||||||
|
|
||||||
let (to_worker, from_service) = tracing_unbounded("mpsc_network_worker");
|
let (to_worker, from_service) = tracing_unbounded("mpsc_network_worker", 100_000);
|
||||||
|
|
||||||
if let Some(path) = ¶ms.network_config.net_config_path {
|
if let Some(path) = ¶ms.network_config.net_config_path {
|
||||||
fs::create_dir_all(path)?;
|
fs::create_dir_all(path)?;
|
||||||
@@ -1003,7 +1003,7 @@ where
|
|||||||
H: ExHashT,
|
H: ExHashT,
|
||||||
{
|
{
|
||||||
fn event_stream(&self, name: &'static str) -> Pin<Box<dyn Stream<Item = Event> + Send>> {
|
fn event_stream(&self, name: &'static str) -> Pin<Box<dyn Stream<Item = Event> + Send>> {
|
||||||
let (tx, rx) = out_events::channel(name);
|
let (tx, rx) = out_events::channel(name, 100_000);
|
||||||
let _ = self.to_worker.unbounded_send(ServiceToWorkerMsg::EventStream(tx));
|
let _ = self.to_worker.unbounded_send(ServiceToWorkerMsg::EventStream(tx));
|
||||||
Box::pin(rx)
|
Box::pin(rx)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -32,25 +32,40 @@
|
|||||||
//! collection.
|
//! collection.
|
||||||
|
|
||||||
use futures::{channel::mpsc, prelude::*, ready, stream::FusedStream};
|
use futures::{channel::mpsc, prelude::*, ready, stream::FusedStream};
|
||||||
|
use log::error;
|
||||||
use parking_lot::Mutex;
|
use parking_lot::Mutex;
|
||||||
use prometheus_endpoint::{register, CounterVec, GaugeVec, Opts, PrometheusError, Registry, U64};
|
use prometheus_endpoint::{register, CounterVec, GaugeVec, Opts, PrometheusError, Registry, U64};
|
||||||
use sc_network_common::protocol::event::Event;
|
use sc_network_common::protocol::event::Event;
|
||||||
use std::{
|
use std::{
|
||||||
|
backtrace::{Backtrace, BacktraceStatus},
|
||||||
cell::RefCell,
|
cell::RefCell,
|
||||||
fmt,
|
fmt,
|
||||||
pin::Pin,
|
pin::Pin,
|
||||||
sync::Arc,
|
sync::{
|
||||||
|
atomic::{AtomicI64, Ordering},
|
||||||
|
Arc,
|
||||||
|
},
|
||||||
task::{Context, Poll},
|
task::{Context, Poll},
|
||||||
};
|
};
|
||||||
|
|
||||||
/// Creates a new channel that can be associated to a [`OutChannels`].
|
/// Creates a new channel that can be associated to a [`OutChannels`].
|
||||||
///
|
///
|
||||||
/// The name is used in Prometheus reports.
|
/// The name is used in Prometheus reports, the queue size threshold is used
|
||||||
pub fn channel(name: &'static str) -> (Sender, Receiver) {
|
/// to warn if there are too many unprocessed events in the channel.
|
||||||
|
pub fn channel(name: &'static str, queue_size_warning: i64) -> (Sender, Receiver) {
|
||||||
let (tx, rx) = mpsc::unbounded();
|
let (tx, rx) = mpsc::unbounded();
|
||||||
let metrics = Arc::new(Mutex::new(None));
|
let metrics = Arc::new(Mutex::new(None));
|
||||||
let tx = Sender { inner: tx, name, metrics: metrics.clone() };
|
let queue_size = Arc::new(AtomicI64::new(0));
|
||||||
let rx = Receiver { inner: rx, name, metrics };
|
let tx = Sender {
|
||||||
|
inner: tx,
|
||||||
|
name,
|
||||||
|
queue_size: queue_size.clone(),
|
||||||
|
queue_size_warning,
|
||||||
|
warning_fired: false,
|
||||||
|
creation_backtrace: Backtrace::capture(),
|
||||||
|
metrics: metrics.clone(),
|
||||||
|
};
|
||||||
|
let rx = Receiver { inner: rx, name, queue_size, metrics };
|
||||||
(tx, rx)
|
(tx, rx)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -63,7 +78,19 @@ pub fn channel(name: &'static str) -> (Sender, Receiver) {
|
|||||||
/// sync on drop. If someone adds a `#[derive(Clone)]` below, it is **wrong**.
|
/// sync on drop. If someone adds a `#[derive(Clone)]` below, it is **wrong**.
|
||||||
pub struct Sender {
|
pub struct Sender {
|
||||||
inner: mpsc::UnboundedSender<Event>,
|
inner: mpsc::UnboundedSender<Event>,
|
||||||
|
/// Name to identify the channel (e.g., in Prometheus and logs).
|
||||||
name: &'static str,
|
name: &'static str,
|
||||||
|
/// Number of events in the queue. Clone of [`Receiver::in_transit`].
|
||||||
|
// To not bother with ordering and possible underflow errors of the unsigned counter
|
||||||
|
// we just use `i64` and `Ordering::Relaxed`, and perceive `queue_size` as approximate.
|
||||||
|
// It can turn < 0 though.
|
||||||
|
queue_size: Arc<AtomicI64>,
|
||||||
|
/// Threshold queue size to generate an error message in the logs.
|
||||||
|
queue_size_warning: i64,
|
||||||
|
/// We generate the error message only once to not spam the logs.
|
||||||
|
warning_fired: bool,
|
||||||
|
/// Backtrace of a place where the channel was created.
|
||||||
|
creation_backtrace: Backtrace,
|
||||||
/// Clone of [`Receiver::metrics`].
|
/// Clone of [`Receiver::metrics`].
|
||||||
metrics: Arc<Mutex<Option<Arc<Option<Metrics>>>>>,
|
metrics: Arc<Mutex<Option<Arc<Option<Metrics>>>>>,
|
||||||
}
|
}
|
||||||
@@ -87,6 +114,7 @@ impl Drop for Sender {
|
|||||||
pub struct Receiver {
|
pub struct Receiver {
|
||||||
inner: mpsc::UnboundedReceiver<Event>,
|
inner: mpsc::UnboundedReceiver<Event>,
|
||||||
name: &'static str,
|
name: &'static str,
|
||||||
|
queue_size: Arc<AtomicI64>,
|
||||||
/// Initially contains `None`, and will be set to a value once the corresponding [`Sender`]
|
/// Initially contains `None`, and will be set to a value once the corresponding [`Sender`]
|
||||||
/// is assigned to an instance of [`OutChannels`].
|
/// is assigned to an instance of [`OutChannels`].
|
||||||
metrics: Arc<Mutex<Option<Arc<Option<Metrics>>>>>,
|
metrics: Arc<Mutex<Option<Arc<Option<Metrics>>>>>,
|
||||||
@@ -97,6 +125,7 @@ impl Stream for Receiver {
|
|||||||
|
|
||||||
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Option<Event>> {
|
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<Option<Event>> {
|
||||||
if let Some(ev) = ready!(Pin::new(&mut self.inner).poll_next(cx)) {
|
if let Some(ev) = ready!(Pin::new(&mut self.inner).poll_next(cx)) {
|
||||||
|
let _ = self.queue_size.fetch_sub(1, Ordering::Relaxed);
|
||||||
let metrics = self.metrics.lock().clone();
|
let metrics = self.metrics.lock().clone();
|
||||||
match metrics.as_ref().map(|m| m.as_ref()) {
|
match metrics.as_ref().map(|m| m.as_ref()) {
|
||||||
Some(Some(metrics)) => metrics.event_out(&ev, self.name),
|
Some(Some(metrics)) => metrics.event_out(&ev, self.name),
|
||||||
@@ -160,12 +189,28 @@ impl OutChannels {
|
|||||||
|
|
||||||
/// Sends an event.
|
/// Sends an event.
|
||||||
pub fn send(&mut self, event: Event) {
|
pub fn send(&mut self, event: Event) {
|
||||||
self.event_streams
|
self.event_streams.retain_mut(|sender| {
|
||||||
.retain(|sender| sender.inner.unbounded_send(event.clone()).is_ok());
|
let queue_size = sender.queue_size.fetch_add(1, Ordering::Relaxed);
|
||||||
|
if queue_size == sender.queue_size_warning && !sender.warning_fired {
|
||||||
|
sender.warning_fired = true;
|
||||||
|
match sender.creation_backtrace.status() {
|
||||||
|
BacktraceStatus::Captured => error!(
|
||||||
|
"The number of unprocessed events in channel `{}` reached {}.\n\
|
||||||
|
The channel was created at:\n{}",
|
||||||
|
sender.name, sender.queue_size_warning, sender.creation_backtrace,
|
||||||
|
),
|
||||||
|
_ => error!(
|
||||||
|
"The number of unprocessed events in channel `{}` reached {}.",
|
||||||
|
sender.name, sender.queue_size_warning,
|
||||||
|
),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sender.inner.unbounded_send(event.clone()).is_ok()
|
||||||
|
});
|
||||||
|
|
||||||
if let Some(metrics) = &*self.metrics {
|
if let Some(metrics) = &*self.metrics {
|
||||||
for ev in &self.event_streams {
|
for ev in &self.event_streams {
|
||||||
metrics.event_in(&event, 1, ev.name);
|
metrics.event_in(&event, ev.name);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -232,45 +277,35 @@ impl Metrics {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn event_in(&self, event: &Event, num: u64, name: &str) {
|
fn event_in(&self, event: &Event, name: &str) {
|
||||||
match event {
|
match event {
|
||||||
Event::Dht(_) => {
|
Event::Dht(_) => {
|
||||||
self.events_total.with_label_values(&["dht", "sent", name]).inc_by(num);
|
self.events_total.with_label_values(&["dht", "sent", name]).inc();
|
||||||
},
|
},
|
||||||
Event::SyncConnected { .. } => {
|
Event::SyncConnected { .. } => {
|
||||||
self.events_total
|
self.events_total.with_label_values(&["sync-connected", "sent", name]).inc();
|
||||||
.with_label_values(&["sync-connected", "sent", name])
|
|
||||||
.inc_by(num);
|
|
||||||
},
|
},
|
||||||
Event::SyncDisconnected { .. } => {
|
Event::SyncDisconnected { .. } => {
|
||||||
self.events_total
|
self.events_total.with_label_values(&["sync-disconnected", "sent", name]).inc();
|
||||||
.with_label_values(&["sync-disconnected", "sent", name])
|
|
||||||
.inc_by(num);
|
|
||||||
},
|
},
|
||||||
Event::NotificationStreamOpened { protocol, .. } => {
|
Event::NotificationStreamOpened { protocol, .. } => {
|
||||||
format_label("notif-open-", protocol, |protocol_label| {
|
format_label("notif-open-", protocol, |protocol_label| {
|
||||||
self.events_total
|
self.events_total.with_label_values(&[protocol_label, "sent", name]).inc();
|
||||||
.with_label_values(&[protocol_label, "sent", name])
|
|
||||||
.inc_by(num);
|
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
Event::NotificationStreamClosed { protocol, .. } => {
|
Event::NotificationStreamClosed { protocol, .. } => {
|
||||||
format_label("notif-closed-", protocol, |protocol_label| {
|
format_label("notif-closed-", protocol, |protocol_label| {
|
||||||
self.events_total
|
self.events_total.with_label_values(&[protocol_label, "sent", name]).inc();
|
||||||
.with_label_values(&[protocol_label, "sent", name])
|
|
||||||
.inc_by(num);
|
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
Event::NotificationsReceived { messages, .. } =>
|
Event::NotificationsReceived { messages, .. } =>
|
||||||
for (protocol, message) in messages {
|
for (protocol, message) in messages {
|
||||||
format_label("notif-", protocol, |protocol_label| {
|
format_label("notif-", protocol, |protocol_label| {
|
||||||
self.events_total
|
self.events_total.with_label_values(&[protocol_label, "sent", name]).inc();
|
||||||
.with_label_values(&[protocol_label, "sent", name])
|
|
||||||
.inc_by(num);
|
|
||||||
});
|
});
|
||||||
self.notifications_sizes.with_label_values(&[protocol, "sent", name]).inc_by(
|
self.notifications_sizes
|
||||||
num.saturating_mul(u64::try_from(message.len()).unwrap_or(u64::MAX)),
|
.with_label_values(&[protocol, "sent", name])
|
||||||
);
|
.inc_by(u64::try_from(message.len()).unwrap_or(u64::MAX));
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1436,7 +1436,7 @@ where
|
|||||||
state_request_protocol_name: ProtocolName,
|
state_request_protocol_name: ProtocolName,
|
||||||
warp_sync_protocol_name: Option<ProtocolName>,
|
warp_sync_protocol_name: Option<ProtocolName>,
|
||||||
) -> Result<(Self, ChainSyncInterfaceHandle<B>, NonDefaultSetConfig), ClientError> {
|
) -> Result<(Self, ChainSyncInterfaceHandle<B>, NonDefaultSetConfig), ClientError> {
|
||||||
let (tx, service_rx) = tracing_unbounded("mpsc_chain_sync");
|
let (tx, service_rx) = tracing_unbounded("mpsc_chain_sync", 100_000);
|
||||||
let block_announce_config = Self::get_block_announce_proto_config(
|
let block_announce_config = Self::get_block_announce_proto_config(
|
||||||
protocol_id,
|
protocol_id,
|
||||||
fork_id,
|
fork_id,
|
||||||
|
|||||||
@@ -99,7 +99,7 @@ impl NetworkServiceHandle {
|
|||||||
impl NetworkServiceProvider {
|
impl NetworkServiceProvider {
|
||||||
/// Create new `NetworkServiceProvider`
|
/// Create new `NetworkServiceProvider`
|
||||||
pub fn new() -> (Self, NetworkServiceHandle) {
|
pub fn new() -> (Self, NetworkServiceHandle) {
|
||||||
let (tx, rx) = tracing_unbounded("mpsc_network_service_provider");
|
let (tx, rx) = tracing_unbounded("mpsc_network_service_provider", 100_000);
|
||||||
|
|
||||||
(Self { rx }, NetworkServiceHandle::new(tx))
|
(Self { rx }, NetworkServiceHandle::new(tx))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,5 +22,6 @@ pin-project = "1.0.12"
|
|||||||
prometheus-endpoint = { package = "substrate-prometheus-endpoint", version = "0.10.0-dev", path = "../../../utils/prometheus" }
|
prometheus-endpoint = { package = "substrate-prometheus-endpoint", version = "0.10.0-dev", path = "../../../utils/prometheus" }
|
||||||
sc-network-common = { version = "0.10.0-dev", path = "../common" }
|
sc-network-common = { version = "0.10.0-dev", path = "../common" }
|
||||||
sc-peerset = { version = "4.0.0-dev", path = "../../peerset" }
|
sc-peerset = { version = "4.0.0-dev", path = "../../peerset" }
|
||||||
|
sc-utils = { version = "4.0.0-dev", path = "../../utils" }
|
||||||
sp-runtime = { version = "7.0.0", path = "../../../primitives/runtime" }
|
sp-runtime = { version = "7.0.0", path = "../../../primitives/runtime" }
|
||||||
sp-consensus = { version = "0.10.0-dev", path = "../../../primitives/consensus/common" }
|
sp-consensus = { version = "0.10.0-dev", path = "../../../primitives/consensus/common" }
|
||||||
|
|||||||
@@ -28,7 +28,7 @@
|
|||||||
|
|
||||||
use crate::config::*;
|
use crate::config::*;
|
||||||
use codec::{Decode, Encode};
|
use codec::{Decode, Encode};
|
||||||
use futures::{channel::mpsc, prelude::*, stream::FuturesUnordered};
|
use futures::{prelude::*, stream::FuturesUnordered};
|
||||||
use libp2p::{multiaddr, PeerId};
|
use libp2p::{multiaddr, PeerId};
|
||||||
use log::{debug, trace, warn};
|
use log::{debug, trace, warn};
|
||||||
use prometheus_endpoint::{register, Counter, PrometheusError, Registry, U64};
|
use prometheus_endpoint::{register, Counter, PrometheusError, Registry, U64};
|
||||||
@@ -40,6 +40,7 @@ use sc_network_common::{
|
|||||||
utils::{interval, LruHashSet},
|
utils::{interval, LruHashSet},
|
||||||
ExHashT,
|
ExHashT,
|
||||||
};
|
};
|
||||||
|
use sc_utils::mpsc::{tracing_unbounded, TracingUnboundedReceiver, TracingUnboundedSender};
|
||||||
use sp_runtime::traits::Block as BlockT;
|
use sp_runtime::traits::Block as BlockT;
|
||||||
use std::{
|
use std::{
|
||||||
collections::{hash_map::Entry, HashMap},
|
collections::{hash_map::Entry, HashMap},
|
||||||
@@ -168,7 +169,7 @@ impl TransactionsHandlerPrototype {
|
|||||||
metrics_registry: Option<&Registry>,
|
metrics_registry: Option<&Registry>,
|
||||||
) -> error::Result<(TransactionsHandler<B, H, S>, TransactionsHandlerController<H>)> {
|
) -> error::Result<(TransactionsHandler<B, H, S>, TransactionsHandlerController<H>)> {
|
||||||
let event_stream = service.event_stream("transactions-handler");
|
let event_stream = service.event_stream("transactions-handler");
|
||||||
let (to_handler, from_controller) = mpsc::unbounded();
|
let (to_handler, from_controller) = tracing_unbounded("mpsc_transactions_handler", 100_000);
|
||||||
|
|
||||||
let handler = TransactionsHandler {
|
let handler = TransactionsHandler {
|
||||||
protocol_name: self.protocol_name,
|
protocol_name: self.protocol_name,
|
||||||
@@ -197,7 +198,7 @@ impl TransactionsHandlerPrototype {
|
|||||||
|
|
||||||
/// Controls the behaviour of a [`TransactionsHandler`] it is connected to.
|
/// Controls the behaviour of a [`TransactionsHandler`] it is connected to.
|
||||||
pub struct TransactionsHandlerController<H: ExHashT> {
|
pub struct TransactionsHandlerController<H: ExHashT> {
|
||||||
to_handler: mpsc::UnboundedSender<ToHandler<H>>,
|
to_handler: TracingUnboundedSender<ToHandler<H>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<H: ExHashT> TransactionsHandlerController<H> {
|
impl<H: ExHashT> TransactionsHandlerController<H> {
|
||||||
@@ -246,7 +247,7 @@ pub struct TransactionsHandler<
|
|||||||
// All connected peers
|
// All connected peers
|
||||||
peers: HashMap<PeerId, Peer<H>>,
|
peers: HashMap<PeerId, Peer<H>>,
|
||||||
transaction_pool: Arc<dyn TransactionPool<H, B>>,
|
transaction_pool: Arc<dyn TransactionPool<H, B>>,
|
||||||
from_controller: mpsc::UnboundedReceiver<ToHandler<H>>,
|
from_controller: TracingUnboundedReceiver<ToHandler<H>>,
|
||||||
/// Prometheus metrics.
|
/// Prometheus metrics.
|
||||||
metrics: Option<Metrics>,
|
metrics: Option<Metrics>,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -66,8 +66,8 @@ impl SharedClient {
|
|||||||
|
|
||||||
/// Creates a pair of [`HttpApi`] and [`HttpWorker`].
|
/// Creates a pair of [`HttpApi`] and [`HttpWorker`].
|
||||||
pub fn http(shared_client: SharedClient) -> (HttpApi, HttpWorker) {
|
pub fn http(shared_client: SharedClient) -> (HttpApi, HttpWorker) {
|
||||||
let (to_worker, from_api) = tracing_unbounded("mpsc_ocw_to_worker");
|
let (to_worker, from_api) = tracing_unbounded("mpsc_ocw_to_worker", 100_000);
|
||||||
let (to_api, from_worker) = tracing_unbounded("mpsc_ocw_to_api");
|
let (to_api, from_worker) = tracing_unbounded("mpsc_ocw_to_api", 100_000);
|
||||||
|
|
||||||
let api = HttpApi {
|
let api = HttpApi {
|
||||||
to_worker,
|
to_worker,
|
||||||
|
|||||||
@@ -275,7 +275,7 @@ pub struct Peerset {
|
|||||||
impl Peerset {
|
impl Peerset {
|
||||||
/// Builds a new peerset from the given configuration.
|
/// Builds a new peerset from the given configuration.
|
||||||
pub fn from_config(config: PeersetConfig) -> (Self, PeersetHandle) {
|
pub fn from_config(config: PeersetConfig) -> (Self, PeersetHandle) {
|
||||||
let (tx, rx) = tracing_unbounded("mpsc_peerset_messages");
|
let (tx, rx) = tracing_unbounded("mpsc_peerset_messages", 10_000);
|
||||||
|
|
||||||
let handle = PeersetHandle { tx: tx.clone() };
|
let handle = PeersetHandle { tx: tx.clone() };
|
||||||
|
|
||||||
|
|||||||
@@ -52,7 +52,7 @@ impl Default for Status {
|
|||||||
fn api<T: Into<Option<Status>>>(sync: T) -> RpcModule<System<Block>> {
|
fn api<T: Into<Option<Status>>>(sync: T) -> RpcModule<System<Block>> {
|
||||||
let status = sync.into().unwrap_or_default();
|
let status = sync.into().unwrap_or_default();
|
||||||
let should_have_peers = !status.is_dev;
|
let should_have_peers = !status.is_dev;
|
||||||
let (tx, rx) = tracing_unbounded("rpc_system_tests");
|
let (tx, rx) = tracing_unbounded("rpc_system_tests", 10_000);
|
||||||
thread::spawn(move || {
|
thread::spawn(move || {
|
||||||
futures::executor::block_on(rx.for_each(move |request| {
|
futures::executor::block_on(rx.for_each(move |request| {
|
||||||
match request {
|
match request {
|
||||||
|
|||||||
@@ -962,7 +962,7 @@ where
|
|||||||
);
|
);
|
||||||
spawn_handle.spawn("import-queue", None, import_queue.run(Box::new(chain_sync_service)));
|
spawn_handle.spawn("import-queue", None, import_queue.run(Box::new(chain_sync_service)));
|
||||||
|
|
||||||
let (system_rpc_tx, system_rpc_rx) = tracing_unbounded("mpsc_system_rpc");
|
let (system_rpc_tx, system_rpc_rx) = tracing_unbounded("mpsc_system_rpc", 10_000);
|
||||||
|
|
||||||
let future = build_network_future(
|
let future = build_network_future(
|
||||||
config.role.clone(),
|
config.role.clone(),
|
||||||
|
|||||||
@@ -1913,13 +1913,13 @@ where
|
|||||||
{
|
{
|
||||||
/// Get block import event stream.
|
/// Get block import event stream.
|
||||||
fn import_notification_stream(&self) -> ImportNotifications<Block> {
|
fn import_notification_stream(&self) -> ImportNotifications<Block> {
|
||||||
let (sink, stream) = tracing_unbounded("mpsc_import_notification_stream");
|
let (sink, stream) = tracing_unbounded("mpsc_import_notification_stream", 100_000);
|
||||||
self.import_notification_sinks.lock().push(sink);
|
self.import_notification_sinks.lock().push(sink);
|
||||||
stream
|
stream
|
||||||
}
|
}
|
||||||
|
|
||||||
fn finality_notification_stream(&self) -> FinalityNotifications<Block> {
|
fn finality_notification_stream(&self) -> FinalityNotifications<Block> {
|
||||||
let (sink, stream) = tracing_unbounded("mpsc_finality_notification_stream");
|
let (sink, stream) = tracing_unbounded("mpsc_finality_notification_stream", 100_000);
|
||||||
self.finality_notification_sinks.lock().push(sink);
|
self.finality_notification_sinks.lock().push(sink);
|
||||||
stream
|
stream
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -310,7 +310,8 @@ impl TaskManager {
|
|||||||
let (signal, on_exit) = exit_future::signal();
|
let (signal, on_exit) = exit_future::signal();
|
||||||
|
|
||||||
// A side-channel for essential tasks to communicate shutdown.
|
// A side-channel for essential tasks to communicate shutdown.
|
||||||
let (essential_failed_tx, essential_failed_rx) = tracing_unbounded("mpsc_essential_tasks");
|
let (essential_failed_tx, essential_failed_rx) =
|
||||||
|
tracing_unbounded("mpsc_essential_tasks", 100);
|
||||||
|
|
||||||
let metrics = prometheus_registry.map(Metrics::register).transpose()?;
|
let metrics = prometheus_registry.map(Metrics::register).transpose()?;
|
||||||
|
|
||||||
|
|||||||
@@ -20,6 +20,7 @@ libp2p = { version = "0.49.0", default-features = false, features = ["dns", "tcp
|
|||||||
log = "0.4.17"
|
log = "0.4.17"
|
||||||
parking_lot = "0.12.1"
|
parking_lot = "0.12.1"
|
||||||
pin-project = "1.0.12"
|
pin-project = "1.0.12"
|
||||||
|
sc-utils = { version = "4.0.0-dev", path = "../utils" }
|
||||||
rand = "0.8.5"
|
rand = "0.8.5"
|
||||||
serde = { version = "1.0.136", features = ["derive"] }
|
serde = { version = "1.0.136", features = ["derive"] }
|
||||||
serde_json = "1.0.85"
|
serde_json = "1.0.85"
|
||||||
|
|||||||
@@ -40,6 +40,7 @@ use futures::{channel::mpsc, prelude::*};
|
|||||||
use libp2p::Multiaddr;
|
use libp2p::Multiaddr;
|
||||||
use log::{error, warn};
|
use log::{error, warn};
|
||||||
use parking_lot::Mutex;
|
use parking_lot::Mutex;
|
||||||
|
use sc_utils::mpsc::{tracing_unbounded, TracingUnboundedReceiver, TracingUnboundedSender};
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
use std::{
|
use std::{
|
||||||
collections::{
|
collections::{
|
||||||
@@ -147,8 +148,8 @@ pub struct SysInfo {
|
|||||||
pub struct TelemetryWorker {
|
pub struct TelemetryWorker {
|
||||||
message_receiver: mpsc::Receiver<TelemetryMessage>,
|
message_receiver: mpsc::Receiver<TelemetryMessage>,
|
||||||
message_sender: mpsc::Sender<TelemetryMessage>,
|
message_sender: mpsc::Sender<TelemetryMessage>,
|
||||||
register_receiver: mpsc::UnboundedReceiver<Register>,
|
register_receiver: TracingUnboundedReceiver<Register>,
|
||||||
register_sender: mpsc::UnboundedSender<Register>,
|
register_sender: TracingUnboundedSender<Register>,
|
||||||
id_counter: Arc<atomic::AtomicU64>,
|
id_counter: Arc<atomic::AtomicU64>,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -163,7 +164,8 @@ impl TelemetryWorker {
|
|||||||
// error as early as possible.
|
// error as early as possible.
|
||||||
let _transport = initialize_transport()?;
|
let _transport = initialize_transport()?;
|
||||||
let (message_sender, message_receiver) = mpsc::channel(buffer_size);
|
let (message_sender, message_receiver) = mpsc::channel(buffer_size);
|
||||||
let (register_sender, register_receiver) = mpsc::unbounded();
|
let (register_sender, register_receiver) =
|
||||||
|
tracing_unbounded("mpsc_telemetry_register", 10_000);
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
message_receiver,
|
message_receiver,
|
||||||
@@ -360,7 +362,7 @@ impl TelemetryWorker {
|
|||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct TelemetryWorkerHandle {
|
pub struct TelemetryWorkerHandle {
|
||||||
message_sender: mpsc::Sender<TelemetryMessage>,
|
message_sender: mpsc::Sender<TelemetryMessage>,
|
||||||
register_sender: mpsc::UnboundedSender<Register>,
|
register_sender: TracingUnboundedSender<Register>,
|
||||||
id_counter: Arc<atomic::AtomicU64>,
|
id_counter: Arc<atomic::AtomicU64>,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -386,7 +388,7 @@ impl TelemetryWorkerHandle {
|
|||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct Telemetry {
|
pub struct Telemetry {
|
||||||
message_sender: mpsc::Sender<TelemetryMessage>,
|
message_sender: mpsc::Sender<TelemetryMessage>,
|
||||||
register_sender: mpsc::UnboundedSender<Register>,
|
register_sender: TracingUnboundedSender<Register>,
|
||||||
id: Id,
|
id: Id,
|
||||||
connection_notifier: TelemetryConnectionNotifier,
|
connection_notifier: TelemetryConnectionNotifier,
|
||||||
endpoints: Option<TelemetryEndpoints>,
|
endpoints: Option<TelemetryEndpoints>,
|
||||||
@@ -460,7 +462,7 @@ impl TelemetryHandle {
|
|||||||
/// (re-)establishes.
|
/// (re-)establishes.
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct TelemetryConnectionNotifier {
|
pub struct TelemetryConnectionNotifier {
|
||||||
register_sender: mpsc::UnboundedSender<Register>,
|
register_sender: TracingUnboundedSender<Register>,
|
||||||
addresses: Vec<Multiaddr>,
|
addresses: Vec<Multiaddr>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -62,7 +62,7 @@ impl<H, BH> Default for Sender<H, BH> {
|
|||||||
impl<H: Clone, BH: Clone> Sender<H, BH> {
|
impl<H: Clone, BH: Clone> Sender<H, BH> {
|
||||||
/// Add a new watcher to this sender object.
|
/// Add a new watcher to this sender object.
|
||||||
pub fn new_watcher(&mut self, hash: H) -> Watcher<H, BH> {
|
pub fn new_watcher(&mut self, hash: H) -> Watcher<H, BH> {
|
||||||
let (tx, receiver) = tracing_unbounded("mpsc_txpool_watcher");
|
let (tx, receiver) = tracing_unbounded("mpsc_txpool_watcher", 100_000);
|
||||||
self.receivers.push(tx);
|
self.receivers.push(tx);
|
||||||
Watcher { receiver, hash }
|
Watcher { receiver, hash }
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -291,7 +291,7 @@ where
|
|||||||
pool: Arc<Pool<Api>>,
|
pool: Arc<Pool<Api>>,
|
||||||
interval: Duration,
|
interval: Duration,
|
||||||
) -> (Self, Pin<Box<dyn Future<Output = ()> + Send>>) {
|
) -> (Self, Pin<Box<dyn Future<Output = ()> + Send>>) {
|
||||||
let (to_worker, from_queue) = tracing_unbounded("mpsc_revalidation_queue");
|
let (to_worker, from_queue) = tracing_unbounded("mpsc_revalidation_queue", 100_000);
|
||||||
|
|
||||||
let worker = RevalidationWorker::new(api.clone(), pool.clone());
|
let worker = RevalidationWorker::new(api.clone(), pool.clone());
|
||||||
|
|
||||||
|
|||||||
@@ -45,71 +45,136 @@ mod inner {
|
|||||||
stream::{FusedStream, Stream},
|
stream::{FusedStream, Stream},
|
||||||
task::{Context, Poll},
|
task::{Context, Poll},
|
||||||
};
|
};
|
||||||
use std::pin::Pin;
|
use log::error;
|
||||||
|
use std::{
|
||||||
|
backtrace::{Backtrace, BacktraceStatus},
|
||||||
|
pin::Pin,
|
||||||
|
sync::{
|
||||||
|
atomic::{AtomicBool, AtomicI64, Ordering},
|
||||||
|
Arc,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
/// Wrapper Type around `UnboundedSender` that increases the global
|
/// Wrapper Type around `UnboundedSender` that increases the global
|
||||||
/// measure when a message is added
|
/// measure when a message is added
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct TracingUnboundedSender<T>(&'static str, UnboundedSender<T>);
|
pub struct TracingUnboundedSender<T> {
|
||||||
|
inner: UnboundedSender<T>,
|
||||||
|
name: &'static str,
|
||||||
|
// To not bother with ordering and possible underflow errors of the unsigned counter
|
||||||
|
// we just use `i64` and `Ordering::Relaxed`, and perceive `queue_size` as approximate.
|
||||||
|
// It can turn < 0 though.
|
||||||
|
queue_size: Arc<AtomicI64>,
|
||||||
|
queue_size_warning: i64,
|
||||||
|
warning_fired: Arc<AtomicBool>,
|
||||||
|
creation_backtrace: Arc<Backtrace>,
|
||||||
|
}
|
||||||
|
|
||||||
// Strangely, deriving `Clone` requires that `T` is also `Clone`.
|
// Strangely, deriving `Clone` requires that `T` is also `Clone`.
|
||||||
impl<T> Clone for TracingUnboundedSender<T> {
|
impl<T> Clone for TracingUnboundedSender<T> {
|
||||||
fn clone(&self) -> Self {
|
fn clone(&self) -> Self {
|
||||||
Self(self.0, self.1.clone())
|
Self {
|
||||||
|
inner: self.inner.clone(),
|
||||||
|
name: self.name,
|
||||||
|
queue_size: self.queue_size.clone(),
|
||||||
|
queue_size_warning: self.queue_size_warning,
|
||||||
|
warning_fired: self.warning_fired.clone(),
|
||||||
|
creation_backtrace: self.creation_backtrace.clone(),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Wrapper Type around `UnboundedReceiver` that decreases the global
|
/// Wrapper Type around `UnboundedReceiver` that decreases the global
|
||||||
/// measure when a message is polled
|
/// measure when a message is polled
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct TracingUnboundedReceiver<T>(&'static str, UnboundedReceiver<T>);
|
pub struct TracingUnboundedReceiver<T> {
|
||||||
|
inner: UnboundedReceiver<T>,
|
||||||
|
name: &'static str,
|
||||||
|
queue_size: Arc<AtomicI64>,
|
||||||
|
}
|
||||||
|
|
||||||
/// Wrapper around `mpsc::unbounded` that tracks the in- and outflow via
|
/// Wrapper around `mpsc::unbounded` that tracks the in- and outflow via
|
||||||
/// `UNBOUNDED_CHANNELS_COUNTER`
|
/// `UNBOUNDED_CHANNELS_COUNTER` and warns if the message queue grows
|
||||||
|
/// above the warning threshold.
|
||||||
pub fn tracing_unbounded<T>(
|
pub fn tracing_unbounded<T>(
|
||||||
key: &'static str,
|
name: &'static str,
|
||||||
|
queue_size_warning: i64,
|
||||||
) -> (TracingUnboundedSender<T>, TracingUnboundedReceiver<T>) {
|
) -> (TracingUnboundedSender<T>, TracingUnboundedReceiver<T>) {
|
||||||
let (s, r) = mpsc::unbounded();
|
let (s, r) = mpsc::unbounded();
|
||||||
(TracingUnboundedSender(key, s), TracingUnboundedReceiver(key, r))
|
let queue_size = Arc::new(AtomicI64::new(0));
|
||||||
|
let sender = TracingUnboundedSender {
|
||||||
|
inner: s,
|
||||||
|
name,
|
||||||
|
queue_size: queue_size.clone(),
|
||||||
|
queue_size_warning,
|
||||||
|
warning_fired: Arc::new(AtomicBool::new(false)),
|
||||||
|
creation_backtrace: Arc::new(Backtrace::capture()),
|
||||||
|
};
|
||||||
|
let receiver = TracingUnboundedReceiver { inner: r, name, queue_size };
|
||||||
|
(sender, receiver)
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T> TracingUnboundedSender<T> {
|
impl<T> TracingUnboundedSender<T> {
|
||||||
/// Proxy function to mpsc::UnboundedSender
|
/// Proxy function to mpsc::UnboundedSender
|
||||||
pub fn poll_ready(&self, ctx: &mut Context) -> Poll<Result<(), SendError>> {
|
pub fn poll_ready(&self, ctx: &mut Context) -> Poll<Result<(), SendError>> {
|
||||||
self.1.poll_ready(ctx)
|
self.inner.poll_ready(ctx)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Proxy function to mpsc::UnboundedSender
|
/// Proxy function to mpsc::UnboundedSender
|
||||||
pub fn is_closed(&self) -> bool {
|
pub fn is_closed(&self) -> bool {
|
||||||
self.1.is_closed()
|
self.inner.is_closed()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Proxy function to mpsc::UnboundedSender
|
/// Proxy function to mpsc::UnboundedSender
|
||||||
pub fn close_channel(&self) {
|
pub fn close_channel(&self) {
|
||||||
self.1.close_channel()
|
self.inner.close_channel()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Proxy function to mpsc::UnboundedSender
|
/// Proxy function to mpsc::UnboundedSender
|
||||||
pub fn disconnect(&mut self) {
|
pub fn disconnect(&mut self) {
|
||||||
self.1.disconnect()
|
self.inner.disconnect()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Proxy function to mpsc::UnboundedSender
|
|
||||||
pub fn start_send(&mut self, msg: T) -> Result<(), SendError> {
|
pub fn start_send(&mut self, msg: T) -> Result<(), SendError> {
|
||||||
self.1.start_send(msg)
|
// The underlying implementation of [`UnboundedSender::start_send`] is the same as
|
||||||
|
// [`UnboundedSender::unbounded_send`], so we just reuse the message counting and
|
||||||
|
// error reporting code from `unbounded_send`.
|
||||||
|
self.unbounded_send(msg).map_err(TrySendError::into_send_error)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Proxy function to mpsc::UnboundedSender
|
/// Proxy function to mpsc::UnboundedSender
|
||||||
pub fn unbounded_send(&self, msg: T) -> Result<(), TrySendError<T>> {
|
pub fn unbounded_send(&self, msg: T) -> Result<(), TrySendError<T>> {
|
||||||
self.1.unbounded_send(msg).map(|s| {
|
self.inner.unbounded_send(msg).map(|s| {
|
||||||
UNBOUNDED_CHANNELS_COUNTER.with_label_values(&[self.0, "send"]).inc();
|
UNBOUNDED_CHANNELS_COUNTER.with_label_values(&[self.name, "send"]).inc();
|
||||||
|
|
||||||
|
let queue_size = self.queue_size.fetch_add(1, Ordering::Relaxed);
|
||||||
|
if queue_size == self.queue_size_warning &&
|
||||||
|
!self.warning_fired.load(Ordering::Relaxed)
|
||||||
|
{
|
||||||
|
// `warning_fired` and `queue_size` are not synchronized, so it's possible
|
||||||
|
// that the warning is fired few times before the `warning_fired` is seen
|
||||||
|
// by all threads. This seems better than introducing a mutex guarding them.
|
||||||
|
self.warning_fired.store(true, Ordering::Relaxed);
|
||||||
|
match self.creation_backtrace.status() {
|
||||||
|
BacktraceStatus::Captured => error!(
|
||||||
|
"The number of unprocessed messages in channel `{}` reached {}.\n\
|
||||||
|
The channel was created at:\n{}",
|
||||||
|
self.name, self.queue_size_warning, self.creation_backtrace,
|
||||||
|
),
|
||||||
|
_ => error!(
|
||||||
|
"The number of unprocessed messages in channel `{}` reached {}.",
|
||||||
|
self.name, self.queue_size_warning,
|
||||||
|
),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
s
|
s
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Proxy function to mpsc::UnboundedSender
|
/// Proxy function to mpsc::UnboundedSender
|
||||||
pub fn same_receiver(&self, other: &UnboundedSender<T>) -> bool {
|
pub fn same_receiver(&self, other: &UnboundedSender<T>) -> bool {
|
||||||
self.1.same_receiver(other)
|
self.inner.same_receiver(other)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -118,7 +183,7 @@ mod inner {
|
|||||||
// consume all items, make sure to reflect the updated count
|
// consume all items, make sure to reflect the updated count
|
||||||
let mut count = 0;
|
let mut count = 0;
|
||||||
loop {
|
loop {
|
||||||
if self.1.is_terminated() {
|
if self.inner.is_terminated() {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -129,7 +194,9 @@ mod inner {
|
|||||||
}
|
}
|
||||||
// and discount the messages
|
// and discount the messages
|
||||||
if count > 0 {
|
if count > 0 {
|
||||||
UNBOUNDED_CHANNELS_COUNTER.with_label_values(&[self.0, "dropped"]).inc_by(count);
|
UNBOUNDED_CHANNELS_COUNTER
|
||||||
|
.with_label_values(&[self.name, "dropped"])
|
||||||
|
.inc_by(count);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -137,15 +204,16 @@ mod inner {
|
|||||||
/// that consumes all messages first and updates the counter
|
/// that consumes all messages first and updates the counter
|
||||||
pub fn close(&mut self) {
|
pub fn close(&mut self) {
|
||||||
self.consume();
|
self.consume();
|
||||||
self.1.close()
|
self.inner.close()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Proxy function to mpsc::UnboundedReceiver
|
/// Proxy function to mpsc::UnboundedReceiver
|
||||||
/// that discounts the messages taken out
|
/// that discounts the messages taken out
|
||||||
pub fn try_next(&mut self) -> Result<Option<T>, TryRecvError> {
|
pub fn try_next(&mut self) -> Result<Option<T>, TryRecvError> {
|
||||||
self.1.try_next().map(|s| {
|
self.inner.try_next().map(|s| {
|
||||||
if s.is_some() {
|
if s.is_some() {
|
||||||
UNBOUNDED_CHANNELS_COUNTER.with_label_values(&[self.0, "received"]).inc();
|
let _ = self.queue_size.fetch_sub(1, Ordering::Relaxed);
|
||||||
|
UNBOUNDED_CHANNELS_COUNTER.with_label_values(&[self.name, "received"]).inc();
|
||||||
}
|
}
|
||||||
s
|
s
|
||||||
})
|
})
|
||||||
@@ -165,10 +233,11 @@ mod inner {
|
|||||||
|
|
||||||
fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<T>> {
|
fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<T>> {
|
||||||
let s = self.get_mut();
|
let s = self.get_mut();
|
||||||
match Pin::new(&mut s.1).poll_next(cx) {
|
match Pin::new(&mut s.inner).poll_next(cx) {
|
||||||
Poll::Ready(msg) => {
|
Poll::Ready(msg) => {
|
||||||
if msg.is_some() {
|
if msg.is_some() {
|
||||||
UNBOUNDED_CHANNELS_COUNTER.with_label_values(&[s.0, "received"]).inc();
|
let _ = s.queue_size.fetch_sub(1, Ordering::Relaxed);
|
||||||
|
UNBOUNDED_CHANNELS_COUNTER.with_label_values(&[s.name, "received"]).inc();
|
||||||
}
|
}
|
||||||
Poll::Ready(msg)
|
Poll::Ready(msg)
|
||||||
},
|
},
|
||||||
@@ -179,7 +248,7 @@ mod inner {
|
|||||||
|
|
||||||
impl<T> FusedStream for TracingUnboundedReceiver<T> {
|
impl<T> FusedStream for TracingUnboundedReceiver<T> {
|
||||||
fn is_terminated(&self) -> bool {
|
fn is_terminated(&self) -> bool {
|
||||||
self.1.is_terminated()
|
self.inner.is_terminated()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -223,6 +292,10 @@ mod inner {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn poll_close(self: Pin<&mut Self>, _: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
fn poll_close(self: Pin<&mut Self>, _: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
||||||
|
// The difference with `TracingUnboundedSender` is intentional. The underlying
|
||||||
|
// implementation differs for `UnboundedSender<T>` and `&UnboundedSender<T>`:
|
||||||
|
// the latter closes the channel completely with `close_channel()`, while the former
|
||||||
|
// only closes this specific sender with `disconnect()`.
|
||||||
self.close_channel();
|
self.close_channel();
|
||||||
Poll::Ready(Ok(()))
|
Poll::Ready(Ok(()))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -79,8 +79,8 @@ impl<Payload, TK: TracingKeyStr> NotificationStream<Payload, TK> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Subscribe to a channel through which the generic payload can be received.
|
/// Subscribe to a channel through which the generic payload can be received.
|
||||||
pub fn subscribe(&self) -> NotificationReceiver<Payload> {
|
pub fn subscribe(&self, queue_size_warning: i64) -> NotificationReceiver<Payload> {
|
||||||
let receiver = self.hub.subscribe(());
|
let receiver = self.hub.subscribe((), queue_size_warning);
|
||||||
NotificationReceiver { receiver }
|
NotificationReceiver { receiver }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -36,7 +36,7 @@ fn notification_channel_simple() {
|
|||||||
|
|
||||||
// Create a future to receive a single notification
|
// Create a future to receive a single notification
|
||||||
// from the stream and verify its payload.
|
// from the stream and verify its payload.
|
||||||
let future = stream.subscribe().take(1).for_each(move |payload| {
|
let future = stream.subscribe(100_000).take(1).for_each(move |payload| {
|
||||||
let test_payload = closure_payload.clone();
|
let test_payload = closure_payload.clone();
|
||||||
async move {
|
async move {
|
||||||
assert_eq!(payload, test_payload);
|
assert_eq!(payload, test_payload);
|
||||||
|
|||||||
@@ -164,7 +164,7 @@ impl<M, R> Hub<M, R> {
|
|||||||
/// Subscribe to this Hub using the `subs_key: K`.
|
/// Subscribe to this Hub using the `subs_key: K`.
|
||||||
///
|
///
|
||||||
/// A subscription with a key `K` is possible if the Registry implements `Subscribe<K>`.
|
/// A subscription with a key `K` is possible if the Registry implements `Subscribe<K>`.
|
||||||
pub fn subscribe<K>(&self, subs_key: K) -> Receiver<M, R>
|
pub fn subscribe<K>(&self, subs_key: K, queue_size_warning: i64) -> Receiver<M, R>
|
||||||
where
|
where
|
||||||
R: Subscribe<K> + Unsubscribe,
|
R: Subscribe<K> + Unsubscribe,
|
||||||
{
|
{
|
||||||
@@ -178,7 +178,7 @@ impl<M, R> Hub<M, R> {
|
|||||||
// have the sink disposed.
|
// have the sink disposed.
|
||||||
shared_borrowed.registry.subscribe(subs_key, subs_id);
|
shared_borrowed.registry.subscribe(subs_key, subs_id);
|
||||||
|
|
||||||
let (tx, rx) = crate::mpsc::tracing_unbounded(self.tracing_key);
|
let (tx, rx) = crate::mpsc::tracing_unbounded(self.tracing_key, queue_size_warning);
|
||||||
assert!(shared_borrowed.sinks.insert(subs_id, tx).is_none(), "Used IDSequence to create another ID. Should be unique until u64 is overflowed. Should be unique.");
|
assert!(shared_borrowed.sinks.insert(subs_id, tx).is_none(), "Used IDSequence to create another ID. Should be unique until u64 is overflowed. Should be unique.");
|
||||||
|
|
||||||
Receiver { shared: Arc::downgrade(&self.shared), subs_id, rx }
|
Receiver { shared: Arc::downgrade(&self.shared), subs_id, rx }
|
||||||
|
|||||||
@@ -27,7 +27,7 @@ fn positive_rx_receives_relevant_messages_and_terminates_upon_hub_drop() {
|
|||||||
// No subscribers yet. That message is not supposed to get to anyone.
|
// No subscribers yet. That message is not supposed to get to anyone.
|
||||||
hub.send(0);
|
hub.send(0);
|
||||||
|
|
||||||
let mut rx_01 = hub.subscribe(SubsKey::new());
|
let mut rx_01 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_eq!(hub.subs_count(), 1);
|
assert_eq!(hub.subs_count(), 1);
|
||||||
|
|
||||||
// That message is sent after subscription. Should be delivered into rx_01.
|
// That message is sent after subscription. Should be delivered into rx_01.
|
||||||
@@ -49,9 +49,9 @@ fn positive_subs_count_is_correct_upon_drop_of_rxs() {
|
|||||||
let hub = TestHub::new(TK);
|
let hub = TestHub::new(TK);
|
||||||
assert_eq!(hub.subs_count(), 0);
|
assert_eq!(hub.subs_count(), 0);
|
||||||
|
|
||||||
let rx_01 = hub.subscribe(SubsKey::new());
|
let rx_01 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_eq!(hub.subs_count(), 1);
|
assert_eq!(hub.subs_count(), 1);
|
||||||
let rx_02 = hub.subscribe(SubsKey::new());
|
let rx_02 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_eq!(hub.subs_count(), 2);
|
assert_eq!(hub.subs_count(), 2);
|
||||||
|
|
||||||
std::mem::drop(rx_01);
|
std::mem::drop(rx_01);
|
||||||
@@ -69,11 +69,11 @@ fn positive_subs_count_is_correct_upon_drop_of_rxs_on_cloned_hubs() {
|
|||||||
assert_eq!(hub_01.subs_count(), 0);
|
assert_eq!(hub_01.subs_count(), 0);
|
||||||
assert_eq!(hub_02.subs_count(), 0);
|
assert_eq!(hub_02.subs_count(), 0);
|
||||||
|
|
||||||
let rx_01 = hub_02.subscribe(SubsKey::new());
|
let rx_01 = hub_02.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_eq!(hub_01.subs_count(), 1);
|
assert_eq!(hub_01.subs_count(), 1);
|
||||||
assert_eq!(hub_02.subs_count(), 1);
|
assert_eq!(hub_02.subs_count(), 1);
|
||||||
|
|
||||||
let rx_02 = hub_02.subscribe(SubsKey::new());
|
let rx_02 = hub_02.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_eq!(hub_01.subs_count(), 2);
|
assert_eq!(hub_01.subs_count(), 2);
|
||||||
assert_eq!(hub_02.subs_count(), 2);
|
assert_eq!(hub_02.subs_count(), 2);
|
||||||
|
|
||||||
|
|||||||
@@ -30,7 +30,7 @@ fn t01() {
|
|||||||
let hub = TestHub::new(TK);
|
let hub = TestHub::new(TK);
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
let rx_01 = hub.subscribe(SubsKey::new());
|
let rx_01 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
|
|
||||||
std::mem::drop(rx_01);
|
std::mem::drop(rx_01);
|
||||||
@@ -45,17 +45,17 @@ fn t02() {
|
|||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
// Subscribe rx-01
|
// Subscribe rx-01
|
||||||
let rx_01 = hub.subscribe(SubsKey::new());
|
let rx_01 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
|
|
||||||
// Subscribe rx-02 so that its unsubscription will lead to an attempt to drop rx-01 in the
|
// Subscribe rx-02 so that its unsubscription will lead to an attempt to drop rx-01 in the
|
||||||
// middle of unsubscription of rx-02
|
// middle of unsubscription of rx-02
|
||||||
let rx_02 = hub.subscribe(SubsKey::new().with_receiver(rx_01));
|
let rx_02 = hub.subscribe(SubsKey::new().with_receiver(rx_01), 100_000);
|
||||||
assert_hub_props(&hub, 2, 2);
|
assert_hub_props(&hub, 2, 2);
|
||||||
|
|
||||||
// Subscribe rx-03 in order to see that it will receive messages after the unclean
|
// Subscribe rx-03 in order to see that it will receive messages after the unclean
|
||||||
// unsubscription
|
// unsubscription
|
||||||
let mut rx_03 = hub.subscribe(SubsKey::new());
|
let mut rx_03 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_hub_props(&hub, 3, 3);
|
assert_hub_props(&hub, 3, 3);
|
||||||
|
|
||||||
// drop rx-02 leads to an attempt to unsubscribe rx-01
|
// drop rx-02 leads to an attempt to unsubscribe rx-01
|
||||||
@@ -69,7 +69,7 @@ fn t02() {
|
|||||||
|
|
||||||
// Subscribe rx-04 in order to see that it will receive messages after the unclean
|
// Subscribe rx-04 in order to see that it will receive messages after the unclean
|
||||||
// unsubscription
|
// unsubscription
|
||||||
let mut rx_04 = hub.subscribe(SubsKey::new());
|
let mut rx_04 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
assert_hub_props(&hub, 3, 3);
|
assert_hub_props(&hub, 3, 3);
|
||||||
|
|
||||||
hub.send(2);
|
hub.send(2);
|
||||||
@@ -96,8 +96,8 @@ fn t02() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async fn add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(hub: &TestHub) {
|
async fn add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(hub: &TestHub) {
|
||||||
let rx_01 = hub.subscribe(SubsKey::new());
|
let rx_01 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
let rx_02 = hub.subscribe(SubsKey::new());
|
let rx_02 = hub.subscribe(SubsKey::new(), 100_000);
|
||||||
|
|
||||||
hub.send(1);
|
hub.send(1);
|
||||||
hub.send(2);
|
hub.send(2);
|
||||||
@@ -121,9 +121,8 @@ fn t03() {
|
|||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
assert!(catch_unwind(AssertUnwindSafe(
|
assert!(catch_unwind(AssertUnwindSafe(|| hub
|
||||||
|| hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnSubscribePanicBefore))
|
.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnSubscribePanicBefore), 100_000)))
|
||||||
))
|
|
||||||
.is_err());
|
.is_err());
|
||||||
|
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
@@ -141,9 +140,8 @@ fn t04() {
|
|||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
assert!(catch_unwind(AssertUnwindSafe(
|
assert!(catch_unwind(AssertUnwindSafe(|| hub
|
||||||
|| hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnSubscribePanicAfter))
|
.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnSubscribePanicAfter), 100_000)))
|
||||||
))
|
|
||||||
.is_err());
|
.is_err());
|
||||||
|
|
||||||
// the registry has panicked after it has added a subs-id into its internal storage — the
|
// the registry has panicked after it has added a subs-id into its internal storage — the
|
||||||
@@ -163,8 +161,8 @@ fn t05() {
|
|||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
let rx_01 =
|
let rx_01 = hub
|
||||||
hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnUnsubscribePanicBefore));
|
.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnUnsubscribePanicBefore), 100_000);
|
||||||
|
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
@@ -189,7 +187,8 @@ fn t06() {
|
|||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
let rx_01 = hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnUnsubscribePanicAfter));
|
let rx_01 = hub
|
||||||
|
.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnUnsubscribePanicAfter), 100_000);
|
||||||
|
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
@@ -214,7 +213,8 @@ fn t07() {
|
|||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
let rx_01 = hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnDispatchPanicBefore));
|
let rx_01 =
|
||||||
|
hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnDispatchPanicBefore), 100_000);
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
assert!(catch_unwind(AssertUnwindSafe(|| hub.send(1))).is_err());
|
assert!(catch_unwind(AssertUnwindSafe(|| hub.send(1))).is_err());
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
@@ -235,7 +235,8 @@ fn t08() {
|
|||||||
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
add_some_subscribers_see_that_messages_are_delivered_and_unsubscribe(&hub).await;
|
||||||
assert_hub_props(&hub, 0, 0);
|
assert_hub_props(&hub, 0, 0);
|
||||||
|
|
||||||
let rx_01 = hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnDispatchPanicAfter));
|
let rx_01 =
|
||||||
|
hub.subscribe(SubsKey::new().with_panic(SubsKeyPanic::OnDispatchPanicAfter), 100_000);
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
assert!(catch_unwind(AssertUnwindSafe(|| hub.send(1))).is_err());
|
assert!(catch_unwind(AssertUnwindSafe(|| hub.send(1))).is_err());
|
||||||
assert_hub_props(&hub, 1, 1);
|
assert_hub_props(&hub, 1, 1);
|
||||||
|
|||||||
@@ -58,7 +58,7 @@ impl<T> Default for StatusSinks<T> {
|
|||||||
impl<T> StatusSinks<T> {
|
impl<T> StatusSinks<T> {
|
||||||
/// Builds a new empty collection.
|
/// Builds a new empty collection.
|
||||||
pub fn new() -> StatusSinks<T> {
|
pub fn new() -> StatusSinks<T> {
|
||||||
let (entries_tx, entries_rx) = tracing_unbounded("status-sinks-entries");
|
let (entries_tx, entries_rx) = tracing_unbounded("status-sinks-entries", 100_000);
|
||||||
|
|
||||||
StatusSinks {
|
StatusSinks {
|
||||||
inner: Mutex::new(Inner { entries: stream::FuturesUnordered::new(), entries_rx }),
|
inner: Mutex::new(Inner { entries: stream::FuturesUnordered::new(), entries_rx }),
|
||||||
@@ -196,7 +196,7 @@ mod tests {
|
|||||||
|
|
||||||
let status_sinks = StatusSinks::new();
|
let status_sinks = StatusSinks::new();
|
||||||
|
|
||||||
let (tx, rx) = tracing_unbounded("test");
|
let (tx, rx) = tracing_unbounded("test", 100_000);
|
||||||
status_sinks.push(Duration::from_millis(100), tx);
|
status_sinks.push(Duration::from_millis(100), tx);
|
||||||
|
|
||||||
let mut val_order = 5;
|
let mut val_order = 5;
|
||||||
|
|||||||
Reference in New Issue
Block a user