heartbeat.py
45 lines
| 1.4 KiB
| text/x-python
|
PythonLexer
Brian Granger
|
r2910 | """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 | ||||
MinRK
|
r3144 | from IPython.utils.localinterfaces import LOCALHOST | ||
|
Brian Granger
|
r2910 | #----------------------------------------------------------------------------- | ||
| # Code | ||||
| #----------------------------------------------------------------------------- | ||||
| class Heartbeat(Thread): | ||||
| "A simple ping-pong style heartbeat that runs in a thread." | ||||
|
MinRK
|
r3144 | def __init__(self, context, addr=(LOCALHOST, 0)): | ||
|
Brian Granger
|
r2910 | 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) | ||||