IPython/zmq/heartbeat.py

"""The client and server for a basic ping-pong style heartbeat.
"""

#-----------------------------------------------------------------------------
#  Copyright (C) 2008-2010  The IPython Development Team
#
#  Distributed under the terms of the BSD License.  The full license is in
#  the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------

#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------

import sys
from threading import Thread

import zmq

#-----------------------------------------------------------------------------
# Code
#-----------------------------------------------------------------------------


class Heartbeat(Thread):
    "A simple ping-pong style heartbeat that runs in a thread."

    def __init__(self, context, addr=('127.0.0.1', 0)):
        Thread.__init__(self)
        self.context = context
        self.addr = addr
        self.ip = addr[0]
        self.port = addr[1]
        self.daemon = True

    def run(self):
        self.socket = self.context.socket(zmq.REP)
        if self.port == 0:
            self.port = self.socket.bind_to_random_port('tcp://%s' % self.ip)
        else:
            self.socket.bind('tcp://%s:%i' % self.addr)
        zmq.device(zmq.FORWARDER, self.socket, self.socket)