varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-04-06 22:11:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement get_remote_address via delegation pattern

Browse Source
This commit is contained in:
acul71
2025-03-13 14:08:13 +01:00
committed by Paul Robinson
parent 798229cd3a
commit fabf2cefc4
8 changed files with 72 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
libp2p/io/abc.py
View File

@ -2,6 +2,9 @@ from abc import (
ABC,
abstractmethod,
)
from typing import (
Optional,
)
class Closer(ABC):
@ -35,7 +38,14 @@ class ReadWriter(Reader, Writer):
class ReadWriteCloser(Reader, Writer, Closer):
pass
@abstractmethod
def get_remote_address(self) -> Optional[tuple[str, int]]:
"""
Return the remote address of the connected peer.
:return: A tuple of (host, port) or None if not available
"""
...
class MsgReader(ABC):
@ -66,3 +76,9 @@ class Encrypter(ABC):
class EncryptedMsgReadWriter(MsgReadWriteCloser, Encrypter):
"""Read/write message with encryption/decryption."""
def get_remote_address(self) -> Optional[tuple[str, int]]:
"""Get remote address if supported by the underlying connection."""
if hasattr(self, "conn") and hasattr(self.conn, "get_remote_address"):
return self.conn.get_remote_address()
return None