mypy: Add read_buffer_nonblocking in Mplex

libp2p/crypto/secp256k1.py

@@ -1,5 +1,4 @@
 import coincurve
-
 from libp2p.crypto.keys import KeyPair, KeyType, PrivateKey, PublicKey
 
 

libp2p/stream_muxer/abc.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Optional
 
 from multiaddr import Multiaddr
 
@@ -56,6 +56,12 @@ class IMuxedConn(ABC):
         :return: message read
         """
 
+    @abstractmethod
+    async def read_buffer_nonblocking(self, stream_id: int) -> Optional[bytes]:
+        """
+        Read a message from `stream_id`'s buffer, non-blockingly.
+        """
+
     @abstractmethod
     async def open_stream(
         self, protocol_id: str, multi_addr: Multiaddr