Skip to content

Commit d06b490

Browse files
committed
Add PeerManager::init_features to DRY up code
1 parent 390a284 commit d06b490

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

lightning/src/ln/peer_handler.rs

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -815,6 +815,13 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
815815
SecretKey::from_slice(&Sha256::from_engine(ephemeral_hash).into_inner()).expect("You broke SHA-256!")
816816
}
817817

818+
fn init_features(&self, their_node_id: &PublicKey) -> InitFeatures {
819+
self.message_handler.chan_handler.provided_init_features(their_node_id)
820+
| self.message_handler.route_handler.provided_init_features(their_node_id)
821+
| self.message_handler.onion_message_handler.provided_init_features(their_node_id)
822+
| self.custom_message_handler.provided_init_features(their_node_id)
823+
}
824+
818825
/// Indicates a new outbound connection has been established to a node with the given `node_id`
819826
/// and an optional remote network address.
820827
///
@@ -1210,10 +1217,7 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
12101217

12111218
peer.set_their_node_id(their_node_id);
12121219
insert_node_id!();
1213-
let features = self.message_handler.chan_handler.provided_init_features(&their_node_id)
1214-
| self.message_handler.route_handler.provided_init_features(&their_node_id)
1215-
| self.message_handler.onion_message_handler.provided_init_features(&their_node_id)
1216-
| self.custom_message_handler.provided_init_features(&their_node_id);
1220+
let features = self.init_features(&their_node_id);
12171221
let resp = msgs::Init { features, remote_network_address: filter_addresses(peer.their_net_address.clone()) };
12181222
self.enqueue_message(peer, &resp);
12191223
peer.awaiting_pong_timer_tick_intervals = 0;
@@ -1225,10 +1229,7 @@ impl<Descriptor: SocketDescriptor, CM: Deref, RM: Deref, OM: Deref, L: Deref, CM
12251229
peer.pending_read_is_header = true;
12261230
peer.set_their_node_id(their_node_id);
12271231
insert_node_id!();
1228-
let features = self.message_handler.chan_handler.provided_init_features(&their_node_id)
1229-
| self.message_handler.route_handler.provided_init_features(&their_node_id)
1230-
| self.message_handler.onion_message_handler.provided_init_features(&their_node_id)
1231-
| self.custom_message_handler.provided_init_features(&their_node_id);
1232+
let features = self.init_features(&their_node_id);
12321233
let resp = msgs::Init { features, remote_network_address: filter_addresses(peer.their_net_address.clone()) };
12331234
self.enqueue_message(peer, &resp);
12341235
peer.awaiting_pong_timer_tick_intervals = 0;

0 commit comments

Comments
 (0)