Pyhon Pika如何使用GeventConnection

Question

我們有幾個從消息隊列中使用的任務。 這些任務的運行時間取決於從數據庫中獲取一些數據。 因此，如果某些數據庫請求需要很長時間，我們希望與Gevent合作，不要阻止程序。 我們正試圖將它與Pika客戶端結合起來，它有一些異步適配器，其中之一是用於 gevent 的： pika.adapters.gevent_connection.GeventConnection 。

我設置了一些玩具代碼，它從一個由整數組成的 MQ 任務中消耗並將它們發布到另一個隊列，同時為每個奇數休眠 4 秒：

# from gevent import monkey
# # Monkeypatch core python libraries to support asynchronous operations.
# monkey.patch_time()

import pika
from pika.adapters.gevent_connection import GeventConnection
from datetime import datetime
import time


def handle_delivery(unused_channel, method, header, body):
    """Called when we receive a message from RabbitMQ"""
    print(f"Received: {body} at {datetime.now()}")
    channel.basic_ack(method.delivery_tag)

    num = int(body)
    print(num)

    if num % 2 != 0:
        time.sleep(4)

    channel.basic_publish(
        exchange='my_test_exchange2',
        routing_key='my_test_queue2',
        body=body
    )

    print("Finished processing")


def on_connected(connection):
    """Called when we are fully connected to RabbitMQ"""
    # Open a channel
    connection.channel(on_open_callback=on_channel_open)


def on_channel_open(new_channel):
    """Called when our channel has opened"""
    global channel
    channel = new_channel
    channel.basic_qos(prefetch_count=1)
    channel.queue_declare(queue="my_queue_gevent5")
    channel.exchange_declare("my_test_exchange2")
    channel.queue_declare(queue="my_test_queue2")
    channel.queue_bind(exchange="my_test_exchange2", queue="my_test_queue2")
    channel.basic_consume("my_queue_gevent5", handle_delivery)


def start_loop(i):
    conn = GeventConnection(pika.ConnectionParameters('localhost'), on_open_callback=on_connected)
    conn.ioloop.start()


start_loop(1)

如果我在沒有調用monkey.patch_time()的情況下運行它，它可以正常工作，並且它會在my_test_queue2上發布結果，但它會按順序工作。 添加monkey.patch_time()補丁后的預期行為是它仍然有效但同時發生。 但是，代碼在調用time.sleep(4) 。 它處理並發布第一個 integer，即 0，然后當 if 子句被觸發時卡在 1。 我究竟做錯了什么？

Answer 1

在ChatGPT的幫助下，我設法讓它工作了。 缺少一個gevent.spawn()調用：

def handle_delivery(unused_channel, method, header, body):
    print("Handling delivery")
    gevent.spawn(process_message, method, body)


def process_message(method, body):
    print(f"Received: {body} at {datetime.now()}")
    channel.basic_ack(method.delivery_tag)

    num = int(body)
    print(num)

    if num % 2 != 0:
        time.sleep(4)

    channel.basic_publish(
        exchange='my_test_exchange2',
        routing_key='my_test_queue2',
        body=body
    )

    print("Finished processing")