Cache socket port number in bolt class

Author: robsdedude

neo4j/_async/io/_bolt.py

@@ -100,6 +100,8 @@ def __init__(self, unresolved_address, sock, max_connection_lifetime, *,
                  auth=None, user_agent=None, routing_context=None):
         self.unresolved_address = unresolved_address
         self.socket = sock
+        self.local_port = self.socket.getsockname()[1]
+        assert self.local_port  # TODO: remove, just for testing
         self.server_info = ServerInfo(Address(sock.getpeername()),
                                       self.PROTOCOL_VERSION)
         # so far `connection.recv_timeout_seconds` is the only available
@@ -379,11 +381,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         pass
 
-    @property
-    @abc.abstractmethod
-    def local_port(self):
-        pass
-
     @abc.abstractmethod
     async def hello(self):
         """ Appends a HELLO message to the outgoing queue, sends it and consumes

neo4j/_async/io/_bolt3.py

@@ -137,13 +137,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         return self.socket.getpeercert(binary_form=True)
 
-    @property
-    def local_port(self):
-        try:
-            return self.socket.getsockname()[1]
-        except OSError:
-            return 0
-
     def get_base_headers(self):
         return {
             "user_agent": self.user_agent,

neo4j/_async/io/_bolt4.py

@@ -90,13 +90,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         return self.socket.getpeercert(binary_form=True)
 
-    @property
-    def local_port(self):
-        try:
-            return self.socket.getsockname()[1]
-        except OSError:
-            return 0
-
     def get_base_headers(self):
         return {
             "user_agent": self.user_agent,

neo4j/_async/io/_bolt5.py

@@ -89,13 +89,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         return self.socket.getpeercert(binary_form=True)
 
-    @property
-    def local_port(self):
-        try:
-            return self.socket.getsockname()[1]
-        except OSError:
-            return 0
-
     def get_base_headers(self):
         headers = {"user_agent": self.user_agent}
         if self.routing_context is not None:

neo4j/_async/io/_common.py

@@ -42,6 +42,8 @@ class AsyncMessageInbox:
 
     def __init__(self, s, on_error):
         self.on_error = on_error
+        self._local_port = s.getsockname()[1]
+        assert self._local_port  # TODO: remove, just for testing
         self._messages = self._yield_messages(s)
 
     async def _yield_messages(self, sock):
@@ -56,7 +58,7 @@ async def _yield_messages(self, sock):
                     await receive_into_buffer(sock, buffer, 2)
                     chunk_size = buffer.pop_u16()
                     if chunk_size == 0:
-                        log.debug("[#%04X]  S: <NOOP>", sock.getsockname()[1])
+                        log.debug("[#%04X]  S: <NOOP>", self._local_port)
 
                 await receive_into_buffer(sock, buffer, chunk_size + 2)
                 chunk_size = buffer.pop_u16()

neo4j/_sync/io/_bolt.py

@@ -100,6 +100,8 @@ def __init__(self, unresolved_address, sock, max_connection_lifetime, *,
                  auth=None, user_agent=None, routing_context=None):
         self.unresolved_address = unresolved_address
         self.socket = sock
+        self.local_port = self.socket.getsockname()[1]
+        assert self.local_port  # TODO: remove, just for testing
         self.server_info = ServerInfo(Address(sock.getpeername()),
                                       self.PROTOCOL_VERSION)
         # so far `connection.recv_timeout_seconds` is the only available
@@ -379,11 +381,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         pass
 
-    @property
-    @abc.abstractmethod
-    def local_port(self):
-        pass
-
     @abc.abstractmethod
     def hello(self):
         """ Appends a HELLO message to the outgoing queue, sends it and consumes

neo4j/_sync/io/_bolt3.py

@@ -137,13 +137,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         return self.socket.getpeercert(binary_form=True)
 
-    @property
-    def local_port(self):
-        try:
-            return self.socket.getsockname()[1]
-        except OSError:
-            return 0
-
     def get_base_headers(self):
         return {
             "user_agent": self.user_agent,

neo4j/_sync/io/_bolt4.py

@@ -90,13 +90,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         return self.socket.getpeercert(binary_form=True)
 
-    @property
-    def local_port(self):
-        try:
-            return self.socket.getsockname()[1]
-        except OSError:
-            return 0
-
     def get_base_headers(self):
         return {
             "user_agent": self.user_agent,

neo4j/_sync/io/_bolt5.py

@@ -89,13 +89,6 @@ def encrypted(self):
     def der_encoded_server_certificate(self):
         return self.socket.getpeercert(binary_form=True)
 
-    @property
-    def local_port(self):
-        try:
-            return self.socket.getsockname()[1]
-        except OSError:
-            return 0
-
     def get_base_headers(self):
         headers = {"user_agent": self.user_agent}
         if self.routing_context is not None:

neo4j/_sync/io/_common.py

@@ -42,6 +42,8 @@ class MessageInbox:
 
     def __init__(self, s, on_error):
         self.on_error = on_error
+        self._local_port = s.getsockname()[1]
+        assert self._local_port  # TODO: remove, just for testing
         self._messages = self._yield_messages(s)
 
     def _yield_messages(self, sock):
@@ -56,7 +58,7 @@ def _yield_messages(self, sock):
                     receive_into_buffer(sock, buffer, 2)
                     chunk_size = buffer.pop_u16()
                     if chunk_size == 0:
-                        log.debug("[#%04X]  S: <NOOP>", sock.getsockname()[1])
+                        log.debug("[#%04X]  S: <NOOP>", self._local_port)
 
                 receive_into_buffer(sock, buffer, chunk_size + 2)
                 chunk_size = buffer.pop_u16()