multiprocessing_tcp.py 文件源码

def get(self):
        try:
            packet = self._socket2.recv(copy=False)
            sample = msgpack.loads(packet)
            self._retry_count = 0
        except zmq.error.Again:
            self._retry_count += 1
            sample = None
        if sample is not None:
            identifier = sample.pop('__process_id__')
            if identifier not in self._conn1_recv_count:
                self._conn1_recv_count[identifier] = 0
                self._conn2_send_count[identifier] = 0
            self._conn1_recv_count[identifier] = max(self._conn1_recv_count[identifier],
                                                     sample.pop('__recv_count__'))
            self._conn2_send_count[identifier] = max(self._conn2_send_count[identifier],
                                                     sample.pop('__send_count__'))
            self._conn2_recv_count += 1
        self.status = WAITING
        return sample