Skip to content

Commit 5c06aac

Browse files
committed
rename MsgCollector to MsgBuffer
1 parent fbddd5f commit 5c06aac

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

p2p/peer.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -201,16 +201,16 @@ async def process_sub_proto_handshake(
201201
raise NotImplementedError("Must be implemented by subclasses")
202202

203203
@contextlib.contextmanager
204-
def collect_sub_proto_messages(self) -> Iterator['MsgCollector']:
204+
def collect_sub_proto_messages(self) -> Iterator['MsgBuffer']:
205205
"""
206206
Can be used to gather up all messages that are sent to the peer.
207207
"""
208208
if not self.is_running:
209209
raise RuntimeError("Cannot collect messages if peer is not running")
210-
msg_collector = MsgCollector()
210+
msg_buffer = MsgBuffer()
211211

212-
with msg_collector.subscribe_peer(self):
213-
yield msg_collector
212+
with msg_buffer.subscribe_peer(self):
213+
yield msg_buffer
214214

215215
@property
216216
def received_msgs_count(self) -> int:
@@ -635,9 +635,9 @@ def subscribe_peer(self, peer: BasePeer) -> Iterator[None]:
635635
peer.remove_subscriber(self)
636636

637637

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
641641
subscription_msg_types = {protocol.Command}
642642

643643
@to_tuple
@@ -706,14 +706,14 @@ async def start_peer(self, peer: BasePeer) -> None:
706706
# Although connect() may seem like a more appropriate place to perform the DAO fork
707707
# check, we do it here because we want to perform it for incoming peer connections as
708708
# well.
709-
with peer.collect_sub_proto_messages() as collector:
709+
with peer.collect_sub_proto_messages() as buffer:
710710
await self.ensure_same_side_on_dao_fork(peer)
711711
except DAOForkCheckFailure as err:
712712
self.logger.debug("DAO fork check with %s failed: %s", peer, err)
713713
await peer.disconnect(DisconnectReason.useless_peer)
714714
return
715715
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())
717717
self._add_peer(peer, msgs)
718718

719719
def _add_peer(self,

0 commit comments

Comments
 (0)