Skip to content

Commit c6045a9

Browse files
committed
f Move more message_handler variables out
1 parent edf8427 commit c6045a9

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

lightning/src/ln/peer_handler.rs

+13-8
Original file line numberDiff line numberDiff line change
@@ -2279,6 +2279,11 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
22792279
}
22802280
}
22812281

2282+
let route_handler = &self.message_handler.route_handler;
2283+
let chan_handler = &self.message_handler.chan_handler;
2284+
let onion_message_handler = &self.message_handler.onion_message_handler;
2285+
let custom_message_handler = &self.message_handler.custom_message_handler;
2286+
22822287
// Handles a `MessageSendEvent`, using `from_chan_handler` to decide if we should
22832288
// robustly gossip broadcast events even if a peer's message buffer is full.
22842289
let mut handle_event = |event, from_chan_handler| {
@@ -2485,15 +2490,15 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
24852490
},
24862491
MessageSendEvent::BroadcastChannelAnnouncement { msg, update_msg } => {
24872492
log_debug!(self.logger, "Handling BroadcastChannelAnnouncement event in peer_handler for short channel id {}", msg.contents.short_channel_id);
2488-
match self.message_handler.route_handler.handle_channel_announcement(None, &msg) {
2493+
match route_handler.handle_channel_announcement(None, &msg) {
24892494
Ok(_) | Err(LightningError { action: msgs::ErrorAction::IgnoreDuplicateGossip, .. }) => {
24902495
let forward = wire::Message::ChannelAnnouncement(msg);
24912496
self.forward_broadcast_msg(peers, &forward, None, from_chan_handler);
24922497
},
24932498
_ => {},
24942499
}
24952500
if let Some(msg) = update_msg {
2496-
match self.message_handler.route_handler.handle_channel_update(None, &msg) {
2501+
match route_handler.handle_channel_update(None, &msg) {
24972502
Ok(_) | Err(LightningError { action: msgs::ErrorAction::IgnoreDuplicateGossip, .. }) => {
24982503
let forward = wire::Message::ChannelUpdate(msg);
24992504
self.forward_broadcast_msg(peers, &forward, None, from_chan_handler);
@@ -2504,7 +2509,7 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
25042509
},
25052510
MessageSendEvent::BroadcastChannelUpdate { msg } => {
25062511
log_debug!(self.logger, "Handling BroadcastChannelUpdate event in peer_handler for contents {:?}", msg.contents);
2507-
match self.message_handler.route_handler.handle_channel_update(None, &msg) {
2512+
match route_handler.handle_channel_update(None, &msg) {
25082513
Ok(_) | Err(LightningError { action: msgs::ErrorAction::IgnoreDuplicateGossip, .. }) => {
25092514
let forward = wire::Message::ChannelUpdate(msg);
25102515
self.forward_broadcast_msg(peers, &forward, None, from_chan_handler);
@@ -2514,7 +2519,7 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
25142519
},
25152520
MessageSendEvent::BroadcastNodeAnnouncement { msg } => {
25162521
log_debug!(self.logger, "Handling BroadcastNodeAnnouncement event in peer_handler for node {}", msg.contents.node_id);
2517-
match self.message_handler.route_handler.handle_node_announcement(None, &msg) {
2522+
match route_handler.handle_node_announcement(None, &msg) {
25182523
Ok(_) | Err(LightningError { action: msgs::ErrorAction::IgnoreDuplicateGossip, .. }) => {
25192524
let forward = wire::Message::NodeAnnouncement(msg);
25202525
self.forward_broadcast_msg(peers, &forward, None, from_chan_handler);
@@ -2606,22 +2611,22 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
26062611
Some(())
26072612
};
26082613

2609-
let chan_events = self.message_handler.chan_handler.get_and_clear_pending_msg_events();
2614+
let chan_events = chan_handler.get_and_clear_pending_msg_events();
26102615
for event in chan_events {
26112616
handle_event(event, true);
26122617
}
26132618

2614-
let route_events = self.message_handler.route_handler.get_and_clear_pending_msg_events();
2619+
let route_events = route_handler.get_and_clear_pending_msg_events();
26152620
for event in route_events {
26162621
handle_event(event, false);
26172622
}
26182623

2619-
let onion_msg_events = self.message_handler.onion_message_handler.get_and_clear_pending_msg_events();
2624+
let onion_msg_events = onion_message_handler.get_and_clear_pending_msg_events();
26202625
for event in onion_msg_events {
26212626
handle_event(event, false);
26222627
}
26232628

2624-
for (node_id, msg) in self.message_handler.custom_message_handler.get_and_clear_pending_msg() {
2629+
for (node_id, msg) in custom_message_handler.get_and_clear_pending_msg() {
26252630
if peers_to_disconnect.get(&node_id).is_some() { continue; }
26262631
let mut peer = if let Some(peer) = get_peer_for_forwarding!(&node_id) { peer } else { continue; };
26272632
self.enqueue_message(&mut peer, &msg);

0 commit comments

Comments
 (0)