Skip to content

Commit 9524689

Browse files
committed
Skip unnecessary persists in process_pending_htlc_forwards
We skip repersisting `ChannelManager` when nothing is actually processed.
1 parent 9949a33 commit 9524689

File tree

2 files changed

+18
-5
lines changed

2 files changed

+18
-5
lines changed

lightning/src/ln/channelmanager.rs

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6363,7 +6363,7 @@ where
63636363

63646364
// Returns whether or not we need to re-persist.
63656365
fn internal_process_pending_htlc_forwards(&self) -> NotifyOption {
6366-
let should_persist = NotifyOption::DoPersist;
6366+
let mut should_persist = NotifyOption::SkipPersistNoEvents;
63676367
self.process_pending_update_add_htlcs();
63686368

63696369
let mut new_events = VecDeque::new();
@@ -6374,6 +6374,7 @@ where
63746374
mem::swap(&mut forward_htlcs, &mut self.forward_htlcs.lock().unwrap());
63756375

63766376
for (short_chan_id, mut pending_forwards) in forward_htlcs {
6377+
should_persist = NotifyOption::DoPersist;
63776378
if short_chan_id != 0 {
63786379
self.process_forward_htlcs(
63796380
short_chan_id,
@@ -6391,7 +6392,7 @@ where
63916392
}
63926393

63936394
let best_block_height = self.best_block.read().unwrap().height;
6394-
self.pending_outbound_payments.check_retry_payments(
6395+
let needs_persist = self.pending_outbound_payments.check_retry_payments(
63956396
&self.router,
63966397
|| self.list_usable_channels(),
63976398
|| self.compute_inflight_htlcs(),
@@ -6402,6 +6403,9 @@ where
64026403
&self.logger,
64036404
|args| self.send_payment_along_path(args),
64046405
);
6406+
if needs_persist {
6407+
should_persist = NotifyOption::DoPersist;
6408+
}
64056409

64066410
for (htlc_source, payment_hash, failure_reason, destination) in failed_forwards.drain(..) {
64076411
self.fail_htlc_backwards_internal(
@@ -6417,13 +6421,17 @@ where
64176421
// next get a `get_and_clear_pending_msg_events` call, but some tests rely on it, and it's
64186422
// nice to do the work now if we can rather than while we're trying to get messages in the
64196423
// network stack.
6420-
self.check_free_holding_cells();
6424+
if self.check_free_holding_cells() {
6425+
should_persist = NotifyOption::DoPersist;
6426+
}
64216427

64226428
if new_events.is_empty() {
64236429
return should_persist;
64246430
}
64256431
let mut events = self.pending_events.lock().unwrap();
64266432
events.append(&mut new_events);
6433+
should_persist = NotifyOption::DoPersist;
6434+
64276435
should_persist
64286436
}
64296437

lightning/src/ln/outbound_payment.rs

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1226,13 +1226,14 @@ impl OutboundPayments {
12261226
)
12271227
}
12281228

1229+
// Returns whether the data changed and needs to be repersisted.
12291230
#[rustfmt::skip]
12301231
pub(super) fn check_retry_payments<R: Deref, ES: Deref, NS: Deref, SP, IH, FH, L: Deref>(
12311232
&self, router: &R, first_hops: FH, inflight_htlcs: IH, entropy_source: &ES, node_signer: &NS,
12321233
best_block_height: u32,
12331234
pending_events: &Mutex<VecDeque<(events::Event, Option<EventCompletionAction>)>>, logger: &L,
12341235
send_payment_along_path: SP,
1235-
)
1236+
) -> bool
12361237
where
12371238
R::Target: Router,
12381239
ES::Target: EntropySource,
@@ -1243,6 +1244,7 @@ impl OutboundPayments {
12431244
L::Target: Logger,
12441245
{
12451246
let _single_thread = self.retry_lock.lock().unwrap();
1247+
let mut should_persist = false;
12461248
loop {
12471249
let mut outbounds = self.pending_outbound_payments.lock().unwrap();
12481250
let mut retry_id_route_params = None;
@@ -1262,7 +1264,8 @@ impl OutboundPayments {
12621264
}
12631265
core::mem::drop(outbounds);
12641266
if let Some((payment_hash, payment_id, route_params)) = retry_id_route_params {
1265-
self.find_route_and_send_payment(payment_hash, payment_id, route_params, router, first_hops(), &inflight_htlcs, entropy_source, node_signer, best_block_height, logger, pending_events, &send_payment_along_path)
1267+
self.find_route_and_send_payment(payment_hash, payment_id, route_params, router, first_hops(), &inflight_htlcs, entropy_source, node_signer, best_block_height, logger, pending_events, &send_payment_along_path);
1268+
should_persist = true;
12661269
} else { break }
12671270
}
12681271

@@ -1278,10 +1281,12 @@ impl OutboundPayments {
12781281
reason: *reason,
12791282
}, None));
12801283
retain = false;
1284+
should_persist = true;
12811285
}
12821286
}
12831287
retain
12841288
});
1289+
should_persist
12851290
}
12861291

12871292
#[rustfmt::skip]

0 commit comments

Comments
 (0)