rabbitmq 高并发mysql

import pika
import mysql.connector

# 连接到 RabbitMQ
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
channel = connection.channel()

# 声明一个队列
channel.queue_declare(queue='task_queue', durable=True)

# 连接到 MySQL
db = mysql.connector.connect(
    host="localhost",
    user="user",
    password="password",
    database="mydatabase"
)
cursor = db.cursor()

# 模拟发送消息
for i in range(10):
    message = f"Task {i}"
    channel.basic_publish(exchange='',
                          routing_key='task_queue',
                          body=message,
                          properties=pika.BasicProperties(
                             delivery_mode=2,  # 消息持久化
                          ))
    print(f" [x] Sent {message}")

# 关闭连接
cursor.close()
db.close()
connection.close()

import pika
import mysql.connector

def callback(ch, method, properties, body):
    print(f" [x] Received {body}")

    # 连接到 MySQL
    db = mysql.connector.connect(
        host="localhost",
        user="user",
        password="password",
        database="mydatabase"
    )
    cursor = db.cursor()

    # 处理消息并更新数据库
    cursor.execute("INSERT INTO tasks (task) VALUES (%s)", (body,))
    db.commit()

    # 关闭连接
    cursor.close()
    db.close()

    print(" [x] Done")
    ch.basic_ack(delivery_tag=method.delivery_tag)

# 连接到 RabbitMQ
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
channel = connection.channel()

# 声明一个队列
channel.queue_declare(queue='task_queue', durable=True)

# 设置 QoS
channel.basic_qos(prefetch_count=1)
channel.basic_consume(queue='task_queue', on_message_callback=callback)

print(' [*] Waiting for messages. To exit press CTRL+C')
channel.start_consuming()