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

sjrpc/core/rpcconnection.py

+8 −0
Original line number Diff line number Diff line
@@ -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: