Skip to content

Commit 3e4474f

Browse files
committed
Drop unnecessary AtomicBools in connect methods
1 parent 10c049d commit 3e4474f

File tree

1 file changed

+4
-21
lines changed

1 file changed

+4
-21
lines changed

src/lib.rs

Lines changed: 4 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,6 @@ use std::convert::TryInto;
161161
use std::default::Default;
162162
use std::fs;
163163
use std::net::ToSocketAddrs;
164-
use std::sync::atomic::{AtomicBool, Ordering};
165164
use std::sync::{Arc, Mutex, RwLock};
166165
use std::time::{Duration, Instant, SystemTime};
167166

@@ -1208,22 +1207,14 @@ impl<K: KVStore + Sync + Send + 'static> Node<K> {
12081207

12091208
let con_node_id = peer_info.node_id;
12101209
let con_addr = peer_info.address.clone();
1211-
let con_success = Arc::new(AtomicBool::new(false));
1212-
let con_success_cloned = Arc::clone(&con_success);
12131210
let con_logger = Arc::clone(&self.logger);
12141211
let con_pm = Arc::clone(&self.peer_manager);
12151212

12161213
tokio::task::block_in_place(move || {
12171214
runtime.block_on(async move {
1218-
let res =
1219-
connect_peer_if_necessary(con_node_id, con_addr, con_pm, con_logger).await;
1220-
con_success_cloned.store(res.is_ok(), Ordering::Release);
1215+
connect_peer_if_necessary(con_node_id, con_addr, con_pm, con_logger).await
12211216
})
1222-
});
1223-
1224-
if !con_success.load(Ordering::Acquire) {
1225-
return Err(Error::ConnectionFailed);
1226-
}
1217+
})?;
12271218

12281219
log_info!(self.logger, "Connected to peer {}@{}. ", peer_info.node_id, peer_info.address);
12291220

@@ -1286,22 +1277,14 @@ impl<K: KVStore + Sync + Send + 'static> Node<K> {
12861277

12871278
let con_node_id = peer_info.node_id;
12881279
let con_addr = peer_info.address.clone();
1289-
let con_success = Arc::new(AtomicBool::new(false));
1290-
let con_success_cloned = Arc::clone(&con_success);
12911280
let con_logger = Arc::clone(&self.logger);
12921281
let con_pm = Arc::clone(&self.peer_manager);
12931282

12941283
tokio::task::block_in_place(move || {
12951284
runtime.block_on(async move {
1296-
let res =
1297-
connect_peer_if_necessary(con_node_id, con_addr, con_pm, con_logger).await;
1298-
con_success_cloned.store(res.is_ok(), Ordering::Release);
1285+
connect_peer_if_necessary(con_node_id, con_addr, con_pm, con_logger).await
12991286
})
1300-
});
1301-
1302-
if !con_success.load(Ordering::Acquire) {
1303-
return Err(Error::ConnectionFailed);
1304-
}
1287+
})?;
13051288

13061289
let user_config = UserConfig {
13071290
channel_handshake_limits: ChannelHandshakeLimits {

0 commit comments

Comments
 (0)