def requeueNonDeletedKeys(self): rabbitHelper = RabbitHelper() task_type = 'app.sdk_client_tasks.mdelete' # requeue pending delete keys so that they may be deleted in another workload while rabbitHelper.qsize(self.task_queue) > 0: task_set = rabbitHelper.getJsonMsg(self.task_queue) if len(task_set) > 0: keys = [task['args'] for task in task_set \ if task['task'] == task_type] if len(keys) > 0: # put back on to consume_queue msg = json.dumps(keys[0][0]) rabbitHelper.putMsg(self.consume_queue, msg) try: # delete task queue rabbitHelper.delete(self.task_queue) # delete consume queue if it was a miss_queue if self.miss_queue is not None and self.consume_queue is not None: rabbitHelper.delete(self.consume_queue) except: pass
def get_remote_phase_status(remoteIP, taskID, retry=5): # assemble a request to remoteIP phase_status method rabbitHelper = RabbitHelper(mq_server=remoteIP) rcq = getResponseQueue(rabbitHelper) task_method = "app.systest_manager.get_phase_status" task_queue = "phase_status_" + cfg.CB_CLUSTER_TAG task_routing_queue = cfg.CB_CLUSTER_TAG + ".phase.status" args = (taskID, rcq) # call phase_status task rawTaskPublisher(task_method, args, task_queue, broker=remoteIP, exchange="default", routing_key=task_routing_queue) # retrieve status of phase rc = None while rc is None and retry > 0: rc = rabbitHelper.getMsg(rcq) time.sleep(2) retry = retry - 1 rabbitHelper.delete(rcq) return rc == 'True'
def get_remote_phase_status(remoteIP, taskID, retry = 5): # assemble a request to remoteIP phase_status method rabbitHelper = RabbitHelper(mq_server = remoteIP) rcq = getResponseQueue(rabbitHelper) task_method = "app.systest_manager.get_phase_status" task_queue = "phase_status_"+cfg.CB_CLUSTER_TAG task_routing_queue = cfg.CB_CLUSTER_TAG+".phase.status" args = (taskID, rcq) # call phase_status task rawTaskPublisher(task_method, args, task_queue, broker = remoteIP, exchange="default", routing_key = task_routing_queue) # retrieve status of phase rc = None while rc is None and retry > 0: rc = rabbitHelper.getMsg(rcq) time.sleep(2) retry = retry - 1 rabbitHelper.delete(rcq) return rc == 'True'