统一消息系统

import redis

import time

r = redis.Redis(host='localhost', port=6379, db=0)

def send_message():

for i in range(10):

r.rpush('training_tasks', f'task_{i}')

print(f"Sent task {i}")

time.sleep(1)

def receive_message():

while True:

task = r.lpop('training_tasks')

if task:

print(f"Received task: {task.decode()}")

else:

print("No more tasks.")

break

if __name__ == "__main__":

import threading

t1 = threading.Thread(target=send_message)

t2 = threading.Thread(target=receive_message)

t1.start()

t2.start()