import redis
import sys
import time
import threading
import signal

START_FLAG = True

def random_enqueue(client: redis.Redis, quene_name: str):
    while START_FLAG:
        now_sec = int(time.time())
        pipeline = client.pipeline(transaction=False)
        for i in range(100):
            score = now_sec << 16 | i
            pipeline.zadd(quene_name, {str(i): score})
        pipeline.execute()
    print('random_enqueue exit')


def random_dequeue(client: redis.Redis, quene_name: str):
    loop = 0
    while START_FLAG:
        start_time = time.time()
        n = client.zcard(quene_name)
        if n < 100:
            continue
        res = client.zremrangebyrank(quene_name, 0, 49)
        duration = time.time() - start_time
        loop += 1
        if loop % 10 == 0:
            print('duration: {}, res: {}'.format(duration, res))
            loop = 0
    print('random_dequeue exit')


def main():
    if len(sys.argv) != 4:
        print('Usage: python redis_queue.py <redis_host> <port> <passwd>')
        sys.exit(1)
    host = sys.argv[1]
    port = int(sys.argv[2])
    passwd = sys.argv[3]
    client_enque = redis.Redis(host=host, port=port, password=passwd)
    client_deque = redis.Redis(host=host, port=port, password=passwd)
    quene_name = 'test_queue'
    t1 = threading.Thread(target=random_enqueue, args=(client_enque, quene_name))
    t1.start()
    t2 = threading.Thread(target=random_dequeue, args=(client_deque, quene_name))
    t2.start()
    
    def signal_handler(signal, frame):  
        print("You pressed Ctrl+C!")
        global START_FLAG
        START_FLAG = False
        t1.join()
        t2.join()
        print('exit')
        sys.exit(0)

    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)
    signal.signal(signal.SIGQUIT, signal_handler)
    while True:
        time.sleep(60)


if __name__ == '__main__':
    main()