| #!/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. |
| # |
| |
| import optparse, time |
| from qpid.messaging import * |
| from qpid.util import URL |
| |
| def nameval(st): |
| idx = st.find("=") |
| if idx >= 0: |
| name = st[0:idx] |
| value = st[idx+1:] |
| else: |
| name = st |
| value = None |
| return name, value |
| |
| parser = optparse.OptionParser(usage="usage: %prog [options] ADDRESS [ CONTENT ... ]", |
| description="Send messages to the supplied address.") |
| parser.add_option("-b", "--broker", default="localhost", |
| help="connect to specified BROKER (default %default)") |
| parser.add_option("-c", "--count", type=int, default=1, |
| help="stop after count messages have been sent, zero disables (default %default)") |
| parser.add_option("-t", "--timeout", type=float, default=None, |
| help="exit after the specified time") |
| parser.add_option("-i", "--id", help="use the supplied id instead of generating one") |
| parser.add_option("-r", "--reply-to", help="specify reply-to address") |
| parser.add_option("-P", "--property", dest="properties", action="append", default=[], |
| metavar="NAME=VALUE", help="specify message property") |
| parser.add_option("-M", "--map", dest="entries", action="append", default=[], |
| metavar="KEY=VALUE", |
| help="specify map entry for message body") |
| |
| opts, args = parser.parse_args() |
| |
| url = URL(opts.broker) |
| if opts.id is None: |
| spout_id = str(uuid4()) |
| else: |
| spout_id = opts.id |
| if args: |
| addr = args.pop(0) |
| else: |
| parser.error("address is required") |
| |
| content = None |
| |
| if args: |
| text = " ".join(args) |
| else: |
| text = None |
| |
| if opts.entries: |
| content = {} |
| if text: |
| content["text"] = text |
| for e in opts.entries: |
| name, val = nameval(e) |
| content[name] = val |
| else: |
| content = text |
| |
| # XXX: should make URL default the port for us |
| conn = Connection.open(url.host, url.port or AMQP_PORT, |
| username=url.user, password=url.password) |
| ssn = conn.session() |
| snd = ssn.sender(addr) |
| |
| count = 0 |
| start = time.time() |
| while (opts.count == 0 or count < opts.count) and \ |
| (opts.timeout is None or time.time() - start < opts.timeout): |
| msg = Message(content, reply_to=opts.reply_to) |
| msg.properties["spout-id"] = "%s:%s" % (spout_id, count) |
| for p in opts.properties: |
| name, val = nameval(p) |
| msg.properties[name] = val |
| |
| try: |
| snd.send(msg) |
| count += 1 |
| print msg |
| except SendError, e: |
| print e |
| break |
| |
| conn.close() |