Skip to content

Commit e2505ad

Browse files
committed
f use AtomicBool
1 parent 7ce999a commit e2505ad

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

lightning/src/ln/peer_handler.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1305,8 +1305,8 @@ mod tests {
13051305
// Create 2 peers with custom TestRoutingMessageHandlers and connect them.
13061306
let (chan_handlers, routing_handlers) = create_chan_handlers(2);
13071307
let peers = create_network(2, &chan_handlers, &routing_handlers);
1308-
routing_handlers[0].request_full_sync.store(1, Ordering::Release);
1309-
routing_handlers[1].request_full_sync.store(1, Ordering::Release);
1308+
routing_handlers[0].request_full_sync.store(true, Ordering::Release);
1309+
routing_handlers[1].request_full_sync.store(true, Ordering::Release);
13101310

13111311
// By calling establish_connect, we trigger do_attempt_write_data between
13121312
// the peers. Previously this function would mistakenly enter an infinite loop
@@ -1333,7 +1333,7 @@ mod tests {
13331333
// Inbound peer 0 requests initial_routing_sync, but outbound peer 1 does not.
13341334
{
13351335
let (chan_handlers, routing_handlers) = create_chan_handlers(2);
1336-
routing_handlers[0].request_full_sync.store(1, Ordering::Release);
1336+
routing_handlers[0].request_full_sync.store(true, Ordering::Release);
13371337
let peers = create_network(2, &chan_handlers, &routing_handlers);
13381338
let (fd_0_to_1, fd_1_to_0) = establish_connection_and_read_events(&peers[0], &peers[1]);
13391339

@@ -1350,7 +1350,7 @@ mod tests {
13501350
// Outbound peer 1 requests initial_routing_sync, but inbound peer 0 does not.
13511351
{
13521352
let (chan_handlers, routing_handlers) = create_chan_handlers(2);
1353-
routing_handlers[1].request_full_sync.store(1, Ordering::Release);
1353+
routing_handlers[1].request_full_sync.store(true, Ordering::Release);
13541354
let peers = create_network(2, &chan_handlers, &routing_handlers);
13551355
let (fd_0_to_1, fd_1_to_0) = establish_connection_and_read_events(&peers[0], &peers[1]);
13561356

lightning/src/util/test_utils.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ use bitcoin::secp256k1::{SecretKey, PublicKey, Secp256k1, Signature};
2525

2626
use std::time::{SystemTime, UNIX_EPOCH};
2727
use std::sync::{Arc,Mutex};
28-
use std::sync::atomic::{AtomicUsize, Ordering};
28+
use std::sync::atomic::{AtomicBool, AtomicUsize, Ordering};
2929
use std::{cmp, mem};
3030
use std::collections::HashMap;
3131

@@ -226,7 +226,7 @@ pub struct TestRoutingMessageHandler {
226226
pub chan_upds_recvd: AtomicUsize,
227227
pub chan_anns_recvd: AtomicUsize,
228228
pub chan_anns_sent: AtomicUsize,
229-
pub request_full_sync: AtomicUsize,
229+
pub request_full_sync: AtomicBool,
230230
}
231231

232232
impl TestRoutingMessageHandler {
@@ -235,7 +235,7 @@ impl TestRoutingMessageHandler {
235235
chan_upds_recvd: AtomicUsize::new(0),
236236
chan_anns_recvd: AtomicUsize::new(0),
237237
chan_anns_sent: AtomicUsize::new(0),
238-
request_full_sync: AtomicUsize::new(0),
238+
request_full_sync: AtomicBool::new(false),
239239
}
240240
}
241241
}
@@ -273,7 +273,7 @@ impl msgs::RoutingMessageHandler for TestRoutingMessageHandler {
273273
}
274274

275275
fn should_request_full_sync(&self, _node_id: &PublicKey) -> bool {
276-
self.request_full_sync.load(Ordering::Acquire) != 0
276+
self.request_full_sync.load(Ordering::Acquire)
277277
}
278278
}
279279

0 commit comments

Comments
 (0)