热门文章
 当前位置：技术文章 > 编程面试题 >

使用pika发布消息时如何添加标头键：值对

问题内容：

我正在编写一个自动测试来测试消费者。到目前为止，发布消息时我不需要包括标题，但现在可以了。而且似乎缺少文档。

这是我的出版商：

class RMQProducer(object):

    def __init__(self, host, exchange, routing_key):
        self.host = host
        self.exchange = exchange
        self.routing_key = routing_key

    def publish_message(self, message):
        connection = pika.BlockingConnection(pika.ConnectionParameters(self.host))
        channel = connection.channel()
        message = json.dumps(message)
        channel.basic_publish(exchange=self.exchange,
                              routing_key=self.routing_key,
                              body=message)

我想像这样做：

channel.basic_publish(exchange=self.exchange,
                      routing_key=self.routing_key,
                      body=message,
                      headers={"key": "value"})

将标头添加到此消息的正确方法是什么？


问题答案：

您将使用pika.BasicProperties添加标头。

channel.basic_publish(exchange=self.exchange,
                      routing_key=self.routing_key,
                      properties=pika.BasicProperties(
                          headers={'key': 'value'} # Add a key/value header
                      ),
                      body=message)

pika的官方文档确实并未完全涵盖此情况，但是文档中确实列出了规格。如果您要继续使用pika，我强烈建议您将此页面添加为书签。