Skip to content

Commit 5cb0880

Browse files
author
Antoine Riard
committed
Remove TestBroadcaster temporary dedup buffer
1 parent 0de8d5d commit 5cb0880

File tree

3 files changed

+3
-14
lines changed

3 files changed

+3
-14
lines changed

lightning/src/ln/functional_test_utils.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@ use std::cell::RefCell;
3535
use std::rc::Rc;
3636
use std::sync::{Arc, Mutex};
3737
use std::mem;
38-
use std::collections::HashSet;
3938

4039
pub const CHAN_CONFIRM_DEPTH: u32 = 100;
4140
pub fn confirm_transaction<'a, 'b: 'a>(notifier: &'a chaininterface::BlockNotifierRef<'b>, chain: &chaininterface::ChainWatchInterfaceUtil, tx: &Transaction, chan_id: u32) {
@@ -858,7 +857,7 @@ pub fn create_node_cfgs(node_count: usize) -> Vec<NodeCfg> {
858857
let logger = Arc::new(test_utils::TestLogger::with_id(format!("node {}", i)));
859858
let fee_estimator = Arc::new(test_utils::TestFeeEstimator { sat_per_kw: 253 });
860859
let chain_monitor = Arc::new(chaininterface::ChainWatchInterfaceUtil::new(Network::Testnet, logger.clone() as Arc<Logger>));
861-
let tx_broadcaster = Arc::new(test_utils::TestBroadcaster{txn_broadcasted: Mutex::new(Vec::new()), broadcasted_txn: Mutex::new(HashSet::new())});
860+
let tx_broadcaster = Arc::new(test_utils::TestBroadcaster{txn_broadcasted: Mutex::new(Vec::new()) });
862861
let mut seed = [0; 32];
863862
rng.fill_bytes(&mut seed);
864863
let keys_manager = Arc::new(test_utils::TestKeysInterface::new(&seed, Network::Testnet, logger.clone() as Arc<Logger>));

lightning/src/ln/functional_tests.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6371,7 +6371,7 @@ fn test_data_loss_protect() {
63716371
let logger: Arc<Logger> = Arc::new(test_utils::TestLogger::with_id(format!("node {}", 0)));
63726372
let mut chan_monitor = <(Sha256dHash, ChannelMonitor<EnforcingChannelKeys>)>::read(&mut ::std::io::Cursor::new(previous_chan_monitor_state.0), Arc::clone(&logger)).unwrap().1;
63736373
let chain_monitor = Arc::new(ChainWatchInterfaceUtil::new(Network::Testnet, Arc::clone(&logger)));
6374-
let tx_broadcaster = Arc::new(test_utils::TestBroadcaster{txn_broadcasted: Mutex::new(Vec::new()), broadcasted_txn: Mutex::new(HashSet::new())});
6374+
let tx_broadcaster = Arc::new(test_utils::TestBroadcaster{txn_broadcasted: Mutex::new(Vec::new())});
63756375
let feeest = Arc::new(test_utils::TestFeeEstimator { sat_per_kw: 253 });
63766376
monitor = test_utils::TestChannelMonitor::new(chain_monitor.clone(), tx_broadcaster.clone(), logger.clone(), feeest.clone());
63776377
node_state_0 = {

lightning/src/util/test_utils.rs

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ use secp256k1::{SecretKey, PublicKey};
2323
use std::time::{SystemTime, UNIX_EPOCH};
2424
use std::sync::{Arc,Mutex};
2525
use std::{mem};
26-
use std::collections::{HashMap, HashSet};
26+
use std::collections::HashMap;
2727

2828
pub struct TestVecWriter(pub Vec<u8>);
2929
impl Writer for TestVecWriter {
@@ -81,19 +81,9 @@ impl channelmonitor::ManyChannelMonitor<EnforcingChannelKeys> for TestChannelMon
8181

8282
pub struct TestBroadcaster {
8383
pub txn_broadcasted: Mutex<Vec<Transaction>>,
84-
pub broadcasted_txn: Mutex<HashSet<Sha256dHash>> // Temporary field while refactoring out tx duplication
8584
}
8685
impl chaininterface::BroadcasterInterface for TestBroadcaster {
8786
fn broadcast_transaction(&self, tx: &Transaction) {
88-
{
89-
if let Some(_) = self.broadcasted_txn.lock().unwrap().get(&tx.txid()) {
90-
// If commitment tx, HTLC-timeout or HTLC-Success, duplicate broadcast are still ok
91-
if tx.input[0].sequence == 0xfffffffd {
92-
return;
93-
}
94-
}
95-
}
96-
self.broadcasted_txn.lock().unwrap().insert(tx.txid());
9787
self.txn_broadcasted.lock().unwrap().push(tx.clone());
9888
}
9989
}

0 commit comments

Comments
 (0)