Skip to content

Commit c85edf2

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

File tree

2 files changed

+17
-5
lines changed

2 files changed

+17
-5
lines changed

lightning/src/ln/channelmanager.rs

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6338,7 +6338,7 @@ where
63386338
/// Will regularly be called by the background processor.
63396339
pub fn process_pending_htlc_forwards(&self) {
63406340
let _persistence_guard = PersistenceNotifierGuard::optionally_notify(self, || {
6341-
let should_persist = NotifyOption::DoPersist;
6341+
let mut should_persist = NotifyOption::SkipPersistNoEvents;
63426342
self.process_pending_update_add_htlcs();
63436343

63446344
let mut new_events = VecDeque::new();
@@ -6349,6 +6349,7 @@ where
63496349
mem::swap(&mut forward_htlcs, &mut self.forward_htlcs.lock().unwrap());
63506350

63516351
for (short_chan_id, mut pending_forwards) in forward_htlcs {
6352+
should_persist = NotifyOption::DoPersist;
63526353
if short_chan_id != 0 {
63536354
let mut forwarding_counterparty = None;
63546355
macro_rules! forwarding_channel_not_found {
@@ -7135,7 +7136,7 @@ where
71357136
}
71367137

71377138
let best_block_height = self.best_block.read().unwrap().height;
7138-
self.pending_outbound_payments.check_retry_payments(
7139+
let needs_persist = self.pending_outbound_payments.check_retry_payments(
71397140
&self.router,
71407141
|| self.list_usable_channels(),
71417142
|| self.compute_inflight_htlcs(),
@@ -7146,6 +7147,9 @@ where
71467147
&self.logger,
71477148
|args| self.send_payment_along_path(args),
71487149
);
7150+
if needs_persist {
7151+
should_persist = NotifyOption::DoPersist;
7152+
}
71497153

71507154
for (htlc_source, payment_hash, failure_reason, destination) in
71517155
failed_forwards.drain(..)
@@ -7163,11 +7167,14 @@ where
71637167
// next get a `get_and_clear_pending_msg_events` call, but some tests rely on it, and it's
71647168
// nice to do the work now if we can rather than while we're trying to get messages in the
71657169
// network stack.
7166-
self.check_free_holding_cells();
7170+
if self.check_free_holding_cells() {
7171+
should_persist = NotifyOption::DoPersist;
7172+
}
71677173

71687174
if !new_events.is_empty() {
71697175
let mut events = self.pending_events.lock().unwrap();
71707176
events.append(&mut new_events);
7177+
should_persist = NotifyOption::DoPersist;
71717178
}
71727179
should_persist
71737180
});

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)