multiprocessing_tcp.py 文件源码

def get(self):
        try:
            sample = self._socket.recv(copy=False)
            sample = msgpack.loads(sample)
        except zmq.error.Again:
            sample = None
        if sample == b'END':
            sample = None
            self._drained = True
        elif sample is not None:
            self._num_recv += 1
            sample['__process_id__'] = mp.current_process().name
            sample['__recv_count__'] = self._num_recv
        self.status = WAITING
        return sample