ail-framework/bin/lib/queues_modules.py

#!/usr/bin/env python3
# -*-coding:UTF-8 -*

import os
import sys
import datetime

sys.path.append(os.environ['AIL_BIN'])
##################################
# Import Project packages
##################################
from lib import ConfigLoader

config_loader = ConfigLoader.ConfigLoader()
r_serv_queues = config_loader.get_redis_conn("Redis_Queues")
config_loader = None

def get_all_queues_name():
    return r_serv_queues.hkeys('queues')

def get_all_queues_dict_with_nb_elem():
    return r_serv_queues.hgetall('queues')

def get_all_queues_with_sorted_nb_elem():
    res = r_serv_queues.hgetall('queues')
    res = sorted(res.items())
    return res

def get_module_pid_by_queue_name(queue_name):
    return r_serv_queues.smembers('MODULE_TYPE_{}'.format(queue_name))

# # TODO: remove last msg part
def get_module_last_process_start_time(queue_name, module_pid):
    res = r_serv_queues.get('MODULE_{}_{}'.format(queue_name, module_pid))
    if res:
        return res.split(',')[0]
    return None

def get_module_last_msg(queue_name, module_pid):
    return r_serv_queues.get('MODULE_{}_{}_PATH'.format(queue_name, module_pid))

def get_all_modules_queues_stats():
    all_modules_queues_stats = []
    for queue_name, nb_elem_queue in get_all_queues_with_sorted_nb_elem():
        l_module_pid = get_module_pid_by_queue_name(queue_name)
        for module_pid in l_module_pid:
            last_process_start_time = get_module_last_process_start_time(queue_name, module_pid)
            if last_process_start_time:
                last_process_start_time = datetime.datetime.fromtimestamp(int(last_process_start_time))
                seconds = int((datetime.datetime.now() - last_process_start_time).total_seconds())
            else:
                seconds = 0
            all_modules_queues_stats.append((queue_name, nb_elem_queue, seconds, module_pid))
    return all_modules_queues_stats


def _get_all_messages_from_queue(queue_name):
    queue_in = f'{queue_name}in'
    #self.r_temp.hset('queues', self.subscriber_name, int(self.r_temp.scard(in_set)))
    return r_serv_queues.smembers(queue_in)

# def is_message_in queue(queue_name):
#     pass

def remove_message_from_queue(queue_name, message, out=False):
    if out:
        queue_key = f'{queue_name}out'
    else:
        queue_key = f'{queue_name}in'
    r_serv_queues.srem(queue_in, message)
    if not out:
        r_serv_queues.hset('queues', queue_name, int(r_serv_queues.scard(queue_key)) )

if __name__ == '__main__':
    print(get_all_queues_with_sorted_nb_elem())
    queue_name = 'Tags'
    res = _get_all_messages_from_queue(queue_name)
    print(res)