@@ -201,16 +201,16 @@ async def process_sub_proto_handshake(
201
201
raise NotImplementedError ("Must be implemented by subclasses" )
202
202
203
203
@contextlib .contextmanager
204
- def collect_sub_proto_messages (self ) -> Iterator ['MsgCollector ' ]:
204
+ def collect_sub_proto_messages (self ) -> Iterator ['MsgBuffer ' ]:
205
205
"""
206
206
Can be used to gather up all messages that are sent to the peer.
207
207
"""
208
208
if not self .is_running :
209
209
raise RuntimeError ("Cannot collect messages if peer is not running" )
210
- msg_collector = MsgCollector ()
210
+ msg_buffer = MsgBuffer ()
211
211
212
- with msg_collector .subscribe_peer (self ):
213
- yield msg_collector
212
+ with msg_buffer .subscribe_peer (self ):
213
+ yield msg_buffer
214
214
215
215
@property
216
216
def received_msgs_count (self ) -> int :
@@ -635,9 +635,9 @@ def subscribe_peer(self, peer: BasePeer) -> Iterator[None]:
635
635
peer .remove_subscriber (self )
636
636
637
637
638
- class MsgCollector (PeerSubscriber ):
639
- logger = logging .getLogger ('p2p.peer.MsgCollector ' )
640
- msg_queue_maxsize = 200
638
+ class MsgBuffer (PeerSubscriber ):
639
+ logger = logging .getLogger ('p2p.peer.MsgBuffer ' )
640
+ msg_queue_maxsize = 500
641
641
subscription_msg_types = {protocol .Command }
642
642
643
643
@to_tuple
@@ -706,14 +706,14 @@ async def start_peer(self, peer: BasePeer) -> None:
706
706
# Although connect() may seem like a more appropriate place to perform the DAO fork
707
707
# check, we do it here because we want to perform it for incoming peer connections as
708
708
# well.
709
- with peer .collect_sub_proto_messages () as collector :
709
+ with peer .collect_sub_proto_messages () as buffer :
710
710
await self .ensure_same_side_on_dao_fork (peer )
711
711
except DAOForkCheckFailure as err :
712
712
self .logger .debug ("DAO fork check with %s failed: %s" , peer , err )
713
713
await peer .disconnect (DisconnectReason .useless_peer )
714
714
return
715
715
else :
716
- msgs = tuple ((cmd , msg ) for _ , cmd , msg in collector .get_messages ())
716
+ msgs = tuple ((cmd , msg ) for _ , cmd , msg in buffer .get_messages ())
717
717
self ._add_peer (peer , msgs )
718
718
719
719
def _add_peer (self ,
0 commit comments