actix/actix-extras
1
0
Fork 0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-24 16:02:59 +01:00
Code Issues Releases Wiki Activity

refactor client connector waiters maintenance

Browse Source
This commit is contained in:
Nikolay Kim 2018-06-23 12:40:21 +06:00
parent e3dc6f0ca8
commit cf38183dcb
1 changed files with 120 additions and 117 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/client/connector.rs
View File

@ -207,7 +207,7 @@ pub struct ClientConnector {
acquired_per_host: HashMap<Key, usize>,
available: HashMap<Key, VecDeque<Conn>>,
to_close: Vec<Connection>,
waiters: HashMap<Key, VecDeque<Waiter>>,
waiters: Option<HashMap<Key, VecDeque<Waiter>>>,
wait_timeout: Option<(Instant, Delay)>,
paused: Paused,
}
@ -255,7 +255,7 @@ impl Default for ClientConnector {
acquired_per_host: HashMap::new(),
available: HashMap::new(),
to_close: Vec::new(),
waiters: HashMap::new(),
waiters: Some(HashMap::new()),
wait_timeout: None,
paused: Paused::No,
}
@ -278,7 +278,7 @@ impl Default for ClientConnector {
acquired_per_host: HashMap::new(),
available: HashMap::new(),
to_close: Vec::new(),
waiters: HashMap::new(),
waiters: Some(HashMap::new()),
wait_timeout: None,
paused: Paused::No,
}
@ -344,7 +344,7 @@ impl ClientConnector {
acquired_per_host: HashMap::new(),
available: HashMap::new(),
to_close: Vec::new(),
waiters: HashMap::new(),
waiters: Some(HashMap::new()),
wait_timeout: None,
paused: Paused::No,
}
@ -504,7 +504,7 @@ impl ClientConnector {
let now = Instant::now();
let mut next = None;
for waiters in self.waiters.values_mut() {
for waiters in self.waiters.as_mut().unwrap().values_mut() {
let mut idx = 0;
while idx < waiters.len() {
if waiters[idx].wait <= now {
@ -556,6 +556,8 @@ impl ClientConnector {
conn_timeout,
};
self.waiters
.as_mut()
.unwrap()
.entry(key)
.or_insert_with(VecDeque::new)
.push_back(waiter);
@ -835,9 +837,9 @@ impl fut::ActorFuture for Maintenance {
act.collect_waiters();
// check waiters
let tmp: &mut ClientConnector = unsafe { &mut *(act as *mut _) };
let mut waiters = act.waiters.take().unwrap();
for (key, waiters) in &mut tmp.waiters {
for (key, waiters) in &mut waiters {
while let Some(waiter) = waiters.pop_front() {
if waiter.tx.is_canceled() {
continue;
@ -865,7 +867,6 @@ impl fut::ActorFuture for Maintenance {
),
).map_err(|_, _, _| ())
.and_then(move |res, act, _| {
#[cfg_attr(rustfmt, rustfmt_skip)]
#[cfg(feature = "alpn")]
match res {
Err(err) => {
@ -910,7 +911,6 @@ impl fut::ActorFuture for Maintenance {
}
}
#[cfg_attr(rustfmt, rustfmt_skip)]
#[cfg(all(feature = "tls", not(feature = "alpn")))]
match res {
Err(err) => {
@ -934,7 +934,8 @@ impl fut::ActorFuture for Maintenance {
Ok(stream) => {
let _ = waiter.tx.send(
Ok(Connection::new(
conn.0.clone(), Some(conn),
conn.0.clone(),
Some(conn),
Box::new(stream),
)),
);
@ -955,7 +956,6 @@ impl fut::ActorFuture for Maintenance {
}
}
#[cfg_attr(rustfmt, rustfmt_skip)]
#[cfg(not(any(feature = "alpn", feature = "tls")))]
match res {
Err(err) => {
@ -966,7 +966,9 @@ impl fut::ActorFuture for Maintenance {
Ok(stream) => {
act.stats.opened += 1;
if conn.0.ssl {
let _ = waiter.tx.send(Err(ClientConnectorError::SslIsNotSupported));
let _ = waiter.tx.send(Err(
ClientConnectorError::SslIsNotSupported,
));
} else {
let _ = waiter.tx.send(Ok(Connection::new(
conn.0.clone(),
@ -984,6 +986,7 @@ impl fut::ActorFuture for Maintenance {
}
}
act.waiters = Some(waiters);
Ok(Async::NotReady)
}
}