e2ecdcfe33
Former-commit-id:a02aeb236c[formerly9f19e3f712] [formerlya02aeb236c[formerly9f19e3f712] [formerly06a8b51d6d[formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]]] Former-commit-id:06a8b51d6dFormer-commit-id:8e80217e59[formerly3360eb6c5f] Former-commit-id:377dcd10b9
154 lines
4.7 KiB
Python
Executable file
154 lines
4.7 KiB
Python
Executable file
#!/usr/bin/env python
|
|
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
#
|
|
"""
|
|
topic_subscriber.py
|
|
|
|
This subscriber creates private queues and binds them
|
|
to the topics 'usa.#', 'europe.#', '#.news', and '#.weather'.
|
|
"""
|
|
|
|
import qpid
|
|
import sys
|
|
import os
|
|
from qpid.util import connect
|
|
from qpid.connection import Connection
|
|
from qpid.datatypes import Message, RangedSet, uuid4
|
|
from qpid.queue import Empty
|
|
|
|
#----- Functions -------------------------------------------
|
|
|
|
def dump_queue(queue):
|
|
|
|
content = "" # Content of the last message read
|
|
final = "That's all, folks!" # In a message body, signals the last message
|
|
message = 0
|
|
|
|
while content != final:
|
|
try:
|
|
message = queue.get(timeout=10)
|
|
content = message.body
|
|
session.message_accept(RangedSet(message.id))
|
|
print content
|
|
except Empty:
|
|
print "No more messages!"
|
|
return
|
|
|
|
|
|
|
|
def subscribe_queue(server_queue_name, local_queue_name):
|
|
|
|
print "Subscribing local queue '" + local_queue_name + "' to " + server_queue_name + "'"
|
|
|
|
queue = session.incoming(local_queue_name)
|
|
|
|
session.message_subscribe(queue=server_queue_name, destination=local_queue_name)
|
|
queue.start()
|
|
|
|
return queue
|
|
|
|
#----- Initialization --------------------------------------
|
|
|
|
# Set parameters for login
|
|
|
|
host="127.0.0.1"
|
|
port=5672
|
|
user="guest"
|
|
password="guest"
|
|
|
|
# If an alternate host or port has been specified, use that instead
|
|
# (this is used in our unit tests)
|
|
if len(sys.argv) > 1 :
|
|
host=sys.argv[1]
|
|
if len(sys.argv) > 2 :
|
|
port=int(sys.argv[2])
|
|
|
|
# Create a connection.
|
|
socket = connect(host, port)
|
|
connection = Connection (sock=socket, username=user, password=password)
|
|
connection.start()
|
|
session = connection.session(str(uuid4()))
|
|
|
|
#----- Main Body -- ----------------------------------------
|
|
|
|
# declare queues on the server
|
|
|
|
news = "news-" + session.name
|
|
weather = "weather-" + session.name
|
|
usa = "usa-" + session.name
|
|
europe = "europe-" + session.name
|
|
|
|
session.queue_declare(queue=news, exclusive=True)
|
|
session.queue_declare(queue=weather, exclusive=True)
|
|
session.queue_declare(queue=usa, exclusive=True)
|
|
session.queue_declare(queue=europe, exclusive=True)
|
|
|
|
# Routing keys may be "usa.news", "usa.weather", "europe.news", or "europe.weather".
|
|
|
|
# The '#' symbol matches one component of a multipart name, e.g. "#.news" matches
|
|
# "europe.news" or "usa.news".
|
|
|
|
session.exchange_bind(exchange="amq.topic", queue=news, binding_key="#.news")
|
|
session.exchange_bind(exchange="amq.topic", queue=weather, binding_key="#.weather")
|
|
session.exchange_bind(exchange="amq.topic", queue=usa, binding_key="usa.#")
|
|
session.exchange_bind(exchange="amq.topic", queue=europe, binding_key="europe.#")
|
|
|
|
# Bind each queue to the control queue so we know when to stop
|
|
|
|
session.exchange_bind(exchange="amq.topic", queue=news, binding_key="control")
|
|
session.exchange_bind(exchange="amq.topic", queue=weather, binding_key="control")
|
|
session.exchange_bind(exchange="amq.topic", queue=usa, binding_key="control")
|
|
session.exchange_bind(exchange="amq.topic", queue=europe, binding_key="control")
|
|
|
|
# Remind the user to start the topic producer
|
|
|
|
print "Queues created - please start the topic producer"
|
|
sys.stdout.flush()
|
|
|
|
# Subscribe local queues to server queues
|
|
|
|
local_news = "local_news"
|
|
local_weather = "local_weather"
|
|
local_usa = "local_usa"
|
|
local_europe = "local_europe"
|
|
|
|
local_news_queue = subscribe_queue(news, local_news)
|
|
local_weather_queue = subscribe_queue(weather, local_weather)
|
|
local_usa_queue = subscribe_queue(usa, local_usa)
|
|
local_europe_queue = subscribe_queue(europe, local_europe)
|
|
|
|
# Call dump_queue to print messages from each queue
|
|
|
|
print "Messages on 'news' queue:"
|
|
dump_queue(local_news_queue)
|
|
|
|
print "Messages on 'weather' queue:"
|
|
dump_queue(local_weather_queue)
|
|
|
|
print "Messages on 'usa' queue:"
|
|
dump_queue(local_usa_queue)
|
|
|
|
print "Messages on 'europe' queue:"
|
|
dump_queue(local_europe_queue)
|
|
|
|
#----- Cleanup ------------------------------------------------
|
|
|
|
# Clean up before exiting so there are no open threads.
|
|
|
|
session.close(timeout=10)
|