Skip to content

Commit 56a7d0a

Browse files
committed
tweak(p2p): Abstract temporary state clearing into methods
1 parent d20387f commit 56a7d0a

File tree

2 files changed

+18
-9
lines changed

2 files changed

+18
-9
lines changed

p2p/src/network/pubsub/p2p_network_pubsub_reducer.rs

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -214,22 +214,16 @@ impl P2pNetworkPubsubState {
214214
Ok(())
215215
}
216216
P2pNetworkPubsubAction::IncomingMessageCleanup { peer_id } => {
217-
pubsub_state.incoming_transactions.clear();
218-
pubsub_state.incoming_snarks.clear();
219-
220-
pubsub_state.incoming_transactions.shrink_to(0x20);
221-
pubsub_state.incoming_snarks.shrink_to(0x20);
222-
223-
pubsub_state.incoming_block = None;
217+
pubsub_state.clear_incoming();
224218

225219
let Some(client_state) = pubsub_state.clients.get_mut(&peer_id) else {
226220
bug_condition!(
227221
"State not found for action P2pNetworkPubsubAction::IncomingMessageCleanup"
228222
);
229223
return Ok(());
230224
};
231-
client_state.incoming_messages.clear();
232-
client_state.incoming_messages.shrink_to(0x20);
225+
226+
client_state.clear_incoming();
233227

234228
Ok(())
235229
}

p2p/src/network/pubsub/p2p_network_pubsub_state.rs

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,16 @@ impl P2pNetworkPubsubState {
116116
}
117117
}
118118
}
119+
120+
pub fn clear_incoming(&mut self) {
121+
self.incoming_transactions.clear();
122+
self.incoming_snarks.clear();
123+
124+
self.incoming_transactions.shrink_to(0x20);
125+
self.incoming_snarks.shrink_to(0x20);
126+
127+
self.incoming_block = None;
128+
}
119129
}
120130

121131
/// State of a pubsub client connected to a peer.
@@ -181,6 +191,11 @@ impl P2pNetworkPubsubClientState {
181191
self.buffer.clear();
182192
self.buffer.shrink_to(0x2000);
183193
}
194+
195+
pub fn clear_incoming(&mut self) {
196+
self.incoming_messages.clear();
197+
self.incoming_messages.shrink_to(0x20)
198+
}
184199
}
185200

186201
#[derive(Default, Serialize, Deserialize, Debug, Clone)]

0 commit comments

Comments
 (0)