blob: 8b942286e44cfd330f1c208f62274327b2b35a32 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
//! Utilities used for the tor protocol.
pub(crate) mod ct;
pub(crate) mod err;
pub(crate) mod keyed_futures_unordered;
pub(crate) mod notify;
pub(crate) mod oneshot_broadcast;
pub(crate) mod poll_all;
pub(crate) mod sink_blocker;
pub(crate) mod skew;
pub(crate) mod sometimes_unbounded_sink;
pub(crate) mod stream_poll_set;
pub(crate) mod timeout;
pub(crate) mod token_bucket;
pub(crate) mod ts;
pub(crate) mod tunnel_activity;
use futures::Sink;
use std::pin::Pin;
use std::task::{Context, Poll};
/// Extension trait for `Sink`
pub(crate) trait SinkExt<T>: Sink<T> {
/// Calls `futures::Sink::poll_ready` but requires `Unpin` and returns `bool`
///
/// Various gnarly places in the circuit reactor find this convenient.
///
/// TODO #1397 (circuit reactor) probably remove this when the circuit reactor is rewritten.
fn poll_ready_unpin_bool(&mut self, cx: &mut Context<'_>) -> Result<bool, Self::Error>
where
Self: Unpin,
{
Ok(match Sink::poll_ready(Pin::new(self), cx) {
Poll::Ready(Ok(())) => true,
Poll::Ready(Err(e)) => return Err(e),
Poll::Pending => false,
})
}
}
impl<T, S: Sink<T>> SinkExt<T> for S {}
/// Convenience alias for
/// [`memquota::SpecificAccount::new_noop()`](crate::memquota::SpecificAccount::new_noop())
///
/// Available only in tests, which makes diff hunks which call this more obviously correct.
#[cfg(any(test, feature = "testing"))]
pub(crate) fn fake_mq<A: crate::memquota::SpecificAccount>() -> A {
A::new_noop()
}
/// A timeout estimator that returns dummy values.
///
/// Used in the tests where the timeout estimates aren't relevant.
#[cfg(test)]
pub(crate) struct DummyTimeoutEstimator;
#[cfg(test)]
impl crate::client::circuit::TimeoutEstimator for DummyTimeoutEstimator {
fn circuit_build_timeout(&self, _length: usize) -> std::time::Duration {
// Dummy value
std::time::Duration::from_millis(1000)
}
}
|
