Skip to content
Snippets Groups Projects
Commit 322e1128 authored by Antoine Millet's avatar Antoine Millet
Browse files

Calls now raise an exception if RpcConnection is disconnected from its peer.

parent 554b242b
No related branches found
No related tags found
No related merge requests found
......@@ -55,6 +55,9 @@ class RpcConnection(object):
# The manager of this connection
self._manager = manager
# Is the RpcConnection connected to its peer:
self._connected = True
@classmethod
def from_addr(cls, addr, port, manager, enable_ssl=False, timeout=30.0,
cert=None, handler=None):
......@@ -163,6 +166,9 @@ class RpcConnection(object):
Message must be a jsonisable structure.
'''
if not self._connected:
raise RpcError('SendError', 'disconnected from the peer')
logging.debug('Sending: %s', message)
json_msg = json.dumps(message)
size = struct.pack('!L', len(json_msg))
......@@ -373,6 +379,8 @@ class RpcConnection(object):
:type callback: :class:`str` or callable or None
'''
self._connected = False
try:
self._sock.close()
except socket.error as err:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment