agent.py 文件源码

def _set_handler(self, socket, handler, update=False):
        """
        Set the socket handler(s).

        Parameters
        ----------
        socket : zmq.Socket
            Socket to set its handler(s).
        handler : function(s)
            Handler(s) for the socket. This can be a list or a dictionary too.
        """
        if update:
            try:
                self.handler[socket].update(self._curated_handlers(handler))
            except KeyError:
                self.handler[socket] = self._curated_handlers(handler)
        else:
            self.handler[socket] = self._curated_handlers(handler)