def test_message(self): data = {'1': 2} msg = Message(**data) self.assertEquals(msg.serialize(), json.dumps(data)) msg = Message.load_from_string(json.dumps(data)) self.assertEquals(msg.serialize(), json.dumps(data))
def _execute(self, job, timeout=None, extract=False): if not isinstance(job, Message): job = Message(**job) if timeout is None: timeout = self.timeout_max_overflow with self.lock: send(self.master, job.serialize()) while True: try: socks = dict(self.poller.poll(timeout)) break except zmq.ZMQError as e: if e.errno != errno.EINTR: raise if socks.get(self.master) == zmq.POLLIN: if extract: return extract_result(recv(self.master)) else: return recv(self.master) raise TimeoutError(timeout / 1000)
def _execute(self, job, timeout=None): if not isinstance(job, Message): job = Message(**job) if timeout is None: timeout = self.timeout_max_overflow with self.lock: send(self.master, job.serialize()) while True: try: socks = dict(self.poller.poll(timeout)) break except zmq.ZMQError as e: if e.errno != errno.EINTR: raise if socks.get(self.master) == zmq.POLLIN: return json.loads(recv(self.master)) raise TimeoutError(timeout)
def _execute(self, job, timeout=None): if not isinstance(job, Message): job = Message(**job) if timeout is None: timeout = self.timeout_max_overflow with self.lock: send(self.master, job.serialize()) while True: try: socks = dict(self.poller.poll(timeout)) break except zmq.ZMQError as e: if e.errno != errno.EINTR: raise if socks.get(self.master) == zmq.POLLIN: data = recv(self.master) return json.loads(data) raise TimeoutError(timeout)