Python连接ActiveMQ的操作

安装操作库

pip install stomp.py

 

基本写法：

# -*-coding:utf-8-*-
import stomp
import time
 
 
queue_name = '/queue/SampleQueue'
topic_name = '/topic/SampleTopic'
listener_name = 'SampleListener'
 
class SampleListener(object):
    def on_message(self, headers, message):
        print 'headers: %s' % headers
        print 'message: %s' % message
 
# 推送到队列queue
def send_to_queue(msg):
    conn = stomp.Connection10([('127.0.0.1',61613)])
    conn.start()
    conn.connect()
    conn.send(queue_name, msg)
    conn.disconnect()
 
#推送到主题
def send_to_topic(msg):
    conn = stomp.Connection10([('127.0.0.1',61613)])
    conn.start()
    conn.connect()
    conn.send(topic_name, msg)
    conn.disconnect()
 
##从队列接收消息
def receive_from_queue():
    conn = stomp.Connection10([('127.0.0.1',61613)])
    conn.set_listener(listener_name, SampleListener())
    conn.start()
    conn.connect()
    conn.subscribe(queue_name)
    time.sleep(1) # secs
    conn.disconnect()
 
##从主题接收消息
def receive_from_topic():
    conn = stomp.Connection10([('127.0.0.1',61613)])
    conn.set_listener(listener_name, SampleListener())
    conn.start()
    conn.connect()
    conn.subscribe(topic_name)
    while 1:
        send_to_topic('topic')
        time.sleep(3) # secs
 
    conn.disconnect()
 
if __name__=='__main__':
    # send_to_queue('len 123')
    # receive_from_queue()
 
    receive_from_topic()

 

发送数据到队列

发送端

import stomp
import time

# 通过调用stomp下的Connection10方法，创建连接，指定ip和端口
conn = stomp.Connection10([("47.dsadsa74.89", 61613)])
# 启动，连接创建完成
conn.start()
# 但是activemq是需要账号密码的，因此这里连接到指定用户上面
conn.connect(username="admin", passcode="admin", wait=True)

# 发送消息
for i in range(10):
    # 第一个参数：队列的名称
    # 第二个参数：消息的内容
    conn.send("queue1", f"message1:{i}")
    time.sleep(3)

# 断开连接
conn.disconnect()

接收端

import stomp


class SampleListener(stomp.ConnectionListener):
    def on_message(self, headers, message):
        # headers是一个字典，里面很多内容，是一个字典，headers['destination']是对应的队列名称
        print(f"headers:{headers['destination']}, message:{message}")

    def on_error(self, headers, message):
        print(f"headers:{headers['destination']}, message:{message}")


# 同样创建一个连接
conn = stomp.Connection10([("47.9dsad.89", 61613)])
# 绑定监听器，我们这里只有一个，所以名字什么的无所谓
conn.set_listener("", SampleListener())
# 启动
conn.start()
# 连接到指定用户
conn.connect(username="admin", passcode="admin", wait=True)
# 订阅到指定的队列，这个要和发送端发送的队列保持一致
# 当发送端有消息过来时，消息的内容会自动传递到SampleListener类下的on_message的message参数里。headers则包含了一些额外信息，比如时间戳、队列名等等
# 当然，如果有错误就会传到on_error中，不过这里没有定义，可以的话定义一下也是好的，参数和on_message是一样的
conn.subscribe("queue1")

# 让程序不停下，不然程序立马就结束了
while True:
    pass
"""
headers:/queue/queue1, message:message1:0
headers:/queue/queue1, message:message1:1
headers:/queue/queue1, message:message1:2
headers:/queue/queue1, message:message1:3
headers:/queue/queue1, message:message1:4
headers:/queue/queue1, message:message1:5
headers:/queue/queue1, message:message1:6
headers:/queue/queue1, message:message1:7
headers:/queue/queue1, message:message1:8
headers:/queue/queue1, message:message1:9
"""

 

同时指定多个消费者

发送端

import stomp

conn = stomp.Connection10([("47.94dasd.89", 61613)])
conn.start()
conn.connect(username="admin", passcode="admin", wait=True)

for i in range(10):
    conn.send("queue1", f"message1:{i}")
    conn.send("queue2", f"message2:{i}")

conn.disconnect()

接收端1

import stomp


class SampleListener(stomp.ConnectionListener):
    def on_message(self, headers, message):
        print(f"headers:{headers['destination']}, message:{message}")

    def on_error(self, headers, message):
        print(f"headers:{headers['destination']}, message:{message}")


conn = stomp.Connection10([("47.dasdas4.89", 61613)])
conn.set_listener("", SampleListener())
conn.start()
conn.connect(username="admin", passcode="admin", wait=True)
conn.subscribe("queue1")

while True:
    pass

接收端2

import stomp


class SampleListener(stomp.ConnectionListener):
    def on_message(self, headers, message):
        print(f"headers:{headers['destination']}, message:{message}")

    def on_error(self, headers, message):
        print(f"headers:{headers['destination']}, message:{message}")


conn = stomp.Connection10([("47.9dasdsa89", 61613)])
conn.set_listener("", SampleListener())
conn.start()
conn.connect(username="admin", passcode="admin", wait=True)
conn.subscribe("queue1")

while True:
    pass

 

参考：

https://www.cnblogs.com/traditional/p/11144123.html