libp2p/builder/phase/
bandwidth_metrics.rs

1use std::{marker::PhantomData, sync::Arc};
2
3use super::*;
4#[allow(deprecated)]
5use crate::bandwidth::BandwidthSinks;
6use crate::{transport_ext::TransportExt, SwarmBuilder};
7
8pub struct BandwidthMetricsPhase<T, R> {
9    pub(crate) relay_behaviour: R,
10    pub(crate) transport: T,
11}
12
13#[cfg(feature = "metrics")]
14impl<T: AuthenticatedMultiplexedTransport, Provider, R>
15    SwarmBuilder<Provider, BandwidthMetricsPhase<T, R>>
16{
17    pub fn with_bandwidth_metrics(
18        self,
19        registry: &mut libp2p_metrics::Registry,
20    ) -> SwarmBuilder<Provider, BehaviourPhase<impl AuthenticatedMultiplexedTransport, R>> {
21        SwarmBuilder {
22            phase: BehaviourPhase {
23                relay_behaviour: self.phase.relay_behaviour,
24                transport: libp2p_metrics::BandwidthTransport::new(self.phase.transport, registry)
25                    .map(|(peer_id, conn), _| (peer_id, StreamMuxerBox::new(conn))),
26            },
27            keypair: self.keypair,
28            phantom: PhantomData,
29        }
30    }
31}
32
33impl<T, Provider, R> SwarmBuilder<Provider, BandwidthMetricsPhase<T, R>> {
34    pub fn without_bandwidth_metrics(self) -> SwarmBuilder<Provider, BehaviourPhase<T, R>> {
35        SwarmBuilder {
36            phase: BehaviourPhase {
37                relay_behaviour: self.phase.relay_behaviour,
38                transport: self.phase.transport,
39            },
40            keypair: self.keypair,
41            phantom: PhantomData,
42        }
43    }
44}
45
46// Shortcuts
47#[cfg(feature = "relay")]
48impl<Provider, T: AuthenticatedMultiplexedTransport>
49    SwarmBuilder<Provider, BandwidthMetricsPhase<T, libp2p_relay::client::Behaviour>>
50{
51    pub fn with_behaviour<B, R: TryIntoBehaviour<B>>(
52        self,
53        constructor: impl FnOnce(&libp2p_identity::Keypair, libp2p_relay::client::Behaviour) -> R,
54    ) -> Result<SwarmBuilder<Provider, SwarmPhase<T, B>>, R::Error> {
55        self.without_bandwidth_metrics().with_behaviour(constructor)
56    }
57}
58
59impl<Provider, T: AuthenticatedMultiplexedTransport>
60    SwarmBuilder<Provider, BandwidthMetricsPhase<T, NoRelayBehaviour>>
61{
62    pub fn with_behaviour<B, R: TryIntoBehaviour<B>>(
63        self,
64        constructor: impl FnOnce(&libp2p_identity::Keypair) -> R,
65    ) -> Result<SwarmBuilder<Provider, SwarmPhase<T, B>>, R::Error> {
66        self.without_bandwidth_metrics().with_behaviour(constructor)
67    }
68}
libp2p/builder/phase/bandwidth_metrics.rs

libp2p/builder/phase/
bandwidth_metrics.rs
