使用RabbitMQ实现带权限的Routing

一、RabbitMQ的安装（官网安装教程）

#ubuntu安装方式，如果出错，按照终端提示方式进行安装
sudo dpkg -i rabbitmq-server_3.3.5-1_all.deb

检查rabbitmq默认服务端口5672是否已监听

netstat -ntlp | grep LISTEN

二、创建用户、虚拟主机、配置权限

sudo rabbitmqctl add_vhost test
sudo rabbitmqctl add_user test 123456
sudo rabbitmqctl set_permissions -p test test ".*" ".*" ".*"

三、简单Routing程序

rabbitmq.py

#-*-coding:utf-8-*-
#!/usr/bin/env python
#rabbitmq.py

import pika
import logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

mq = {
	'host': '127.0.0.1',
	'port': 5672,
	'user': 'test',
	'pwd': '123456',
	'virt': 'test',
	'exchange': 'securityte',
	'exchange_type': 'direct',
}

class MQRouting():
	"""rabbitmq操作类,广播消息到所有队列，各队列进行消息筛选"""
	def __init__(self):
		credentials = pika.PlainCredentials(mq['user'], mq['pwd'])
		self.conn = pika.BlockingConnection(pika.ConnectionParameters(mq['host'], mq['port'], mq['virt'], credentials))
		self.channel = self.conn.channel()
		self.channel.exchange_declare(exchange=mq['exchange'], type=mq['exchange_type'])
		logger.info('Rabbitmq连接已建立，host={}, port={}'.format(mq['host'], mq['port']))

	def public(self, routing_key, msg):
		"""发布消息"""
		self.channel.basic_publish(
			exchange = mq['exchange'],
			routing_key = routing_key,
			body=msg,
			properties=pika.BasicProperties(
				delivery_mode = 2, # make message persistent
			)
		)
		logger.info('消息已发布：{}:{}'.format(routing_key,msg))

	def consume(self, routing_key):
		"""接收消息"""
		def callback(ch, method, properties, body):
			print " [x] Received %r" % (body,)
		self.channel.queue_declare(queue=routing_key)
		self.channel.queue_bind(exchange=mq['exchange'], queue=routing_key, routing_key=routing_key)
		self.channel.basic_consume(callback, queue=routing_key, no_ack=True)
		self.channel.start_consuming()

	def close(self):
		if self.channel: self.channel.close()
		if self.conn: self.conn.close()


if __name__ == '__main__':
	routing = MQRouting()
	routing.public('first','hello first!')
	routing.public('second','hello second!')

consume1.py

from rabbitmq import MQRouting

routing = MQRouting()
routing.consume('first')

consume2.py

from rabbitmq import MQRouting

routing = MQRouting()
routing.consume('second')

运行代码

python rabbitmq.py
python consume1.py
python consume2.py