def _subscribe_error_request(self): cmd = pubsub_pb2.Command( channel_type=pubsub_pb2.RAY_ERROR_INFO_CHANNEL, subscribe_message={}) req = gcs_service_pb2.GcsSubscriberCommandBatchRequest( subscriber_id=self._subscriber_id, commands=[cmd]) return req
def _unsubscribe_request(self, channels): req = gcs_service_pb2.GcsSubscriberCommandBatchRequest( subscriber_id=self._subscriber_id, commands=[]) for channel in channels: req.commands.append( pubsub_pb2.Command(channel_type=channel, unsubscribe_message={})) return req
def _unsubscribe_request(self): req = gcs_service_pb2.GcsSubscriberCommandBatchRequest( subscriber_id=self._subscriber_id, commands=[]) if self._subscribed_error: cmd = pubsub_pb2.Command( channel_type=pubsub_pb2.RAY_ERROR_INFO_CHANNEL, unsubscribe_message={}) req.commands.append(cmd) return req
def _subscribe_request(self, channel): cmd = pubsub_pb2.Command(channel_type=channel, subscribe_message={}) req = gcs_service_pb2.GcsSubscriberCommandBatchRequest( subscriber_id=self._subscriber_id, commands=[cmd]) return req