Skip to content

Commit a9d3bab

Browse files
committed
Minor adjustment to fee rate handling in RBF fee bumping
Use user-provided fee rate for bumping transaction and return an error if it is too low, otherwise use system estimated fee rate and retry mechanism. Also switches the RBF test to use `random_chain_source`, and removes unnecessary sleep-based waits.
1 parent 2f5a966 commit a9d3bab

2 files changed

Lines changed: 27 additions & 12 deletions

File tree

src/wallet/mod.rs

Lines changed: 26 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -338,6 +338,7 @@ impl Wallet {
338338
WalletEvent::TxUnconfirmed { txid, tx, old_block_time: None } => {
339339
let payment_id = self
340340
.find_payment_by_txid(txid)
341+
.or_else(|| self.find_payment_by_tx_conflicts(locked_wallet, &tx))
341342
.unwrap_or_else(|| PaymentId(txid.to_byte_array()));
342343

343344
let payment = self.create_payment_from_tx(
@@ -378,8 +379,8 @@ impl Wallet {
378379
);
379380
let payment =
380381
self.payment_store.get(&payment_id).ok_or(Error::InvalidPaymentId)?;
381-
let pending_payment_details = self
382-
.create_pending_payment_from_tx(payment.clone(), conflict_txids.clone());
382+
let pending_payment_details =
383+
self.create_pending_payment_from_tx(payment, conflict_txids.clone());
383384

384385
self.pending_payment_store.insert_or_update(pending_payment_details)?;
385386
},
@@ -1053,6 +1054,15 @@ impl Wallet {
10531054
None
10541055
}
10551056

1057+
fn find_payment_by_tx_conflicts(
1058+
&self, locked_wallet: &PersistedWallet<KVStoreWalletPersister>, tx: &Transaction,
1059+
) -> Option<PaymentId> {
1060+
locked_wallet
1061+
.tx_graph()
1062+
.direct_conflicts(tx)
1063+
.find_map(|(_, conflict_txid)| self.find_payment_by_txid(conflict_txid))
1064+
}
1065+
10561066
#[allow(deprecated)]
10571067
pub(crate) fn bump_fee_rbf(
10581068
&self, payment_id: PaymentId, fee_rate: Option<FeeRate>,
@@ -1125,13 +1135,13 @@ impl Wallet {
11251135
old_fee_rate.to_sat_per_kwu() + INCREMENTAL_RELAY_FEE_SAT_PER_1000_WEIGHT as u64;
11261136

11271137
let confirmation_target = ConfirmationTarget::OnchainPayment;
1128-
let estimated_fee_rate =
1129-
fee_rate.unwrap_or_else(|| self.fee_estimator.estimate_fee_rate(confirmation_target));
1138+
let estimated_fee_rate = self.fee_estimator.estimate_fee_rate(confirmation_target);
11301139

11311140
// Use the higher of minimum RBF requirement or current network estimate
11321141
let final_fee_rate_sat_per_kwu =
11331142
min_required_fee_rate_sat_per_kwu.max(estimated_fee_rate.to_sat_per_kwu());
1134-
let final_fee_rate = FeeRate::from_sat_per_kwu(final_fee_rate_sat_per_kwu);
1143+
let final_fee_rate =
1144+
fee_rate.unwrap_or_else(|| FeeRate::from_sat_per_kwu(final_fee_rate_sat_per_kwu));
11351145

11361146
let mut psbt = {
11371147
let mut builder = locked_wallet.build_fee_bump(txid).map_err(|e| {
@@ -1156,6 +1166,17 @@ impl Wallet {
11561166
match builder.finish() {
11571167
Ok(psbt) => Ok(psbt),
11581168
Err(CreateTxError::FeeRateTooLow { required: required_fee_rate }) => {
1169+
if fee_rate.is_some() {
1170+
log_error!(
1171+
self.logger,
1172+
"Provided fee rate {} is too low for RBF fee bump of txid {}, required minimum fee rate: {}",
1173+
fee_rate.unwrap(),
1174+
txid,
1175+
required_fee_rate
1176+
);
1177+
return Err(Error::InvalidFeeRate);
1178+
}
1179+
11591180
log_info!(self.logger, "BDK requires higher fee rate: {}", required_fee_rate);
11601181

11611182
// BDK may require a higher fee rate than our estimate due to

tests/integration_tests_rust.rs

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2466,7 +2466,7 @@ async fn persistence_backwards_compatibility() {
24662466
#[tokio::test(flavor = "multi_thread", worker_threads = 1)]
24672467
async fn onchain_fee_bump_rbf() {
24682468
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
2469-
let chain_source = TestChainSource::Esplora(&electrsd);
2469+
let chain_source = random_chain_source(&bitcoind, &electrsd);
24702470
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
24712471

24722472
// Fund both nodes
@@ -2516,9 +2516,6 @@ async fn onchain_fee_bump_rbf() {
25162516
let new_txid = node_b.onchain_payment().bump_fee_rbf(payment_id, None).unwrap();
25172517
wait_for_tx(&electrsd.client, new_txid).await;
25182518

2519-
// Sleep to allow for transaction propagation
2520-
tokio::time::sleep(std::time::Duration::from_secs(5)).await;
2521-
25222519
node_a.sync_wallets().unwrap();
25232520
node_b.sync_wallets().unwrap();
25242521

@@ -2557,9 +2554,6 @@ async fn onchain_fee_bump_rbf() {
25572554
let second_bump_txid = node_b.onchain_payment().bump_fee_rbf(payment_id, None).unwrap();
25582555
wait_for_tx(&electrsd.client, second_bump_txid).await;
25592556

2560-
// Sleep to allow for transaction propagation
2561-
tokio::time::sleep(std::time::Duration::from_secs(5)).await;
2562-
25632557
node_a.sync_wallets().unwrap();
25642558
node_b.sync_wallets().unwrap();
25652559

0 commit comments

Comments
 (0)