diff options
Diffstat (limited to 'qpid/cpp/management/python/bin/qpid-printevents')
| -rwxr-xr-x | qpid/cpp/management/python/bin/qpid-printevents | 191 |
1 files changed, 191 insertions, 0 deletions
diff --git a/qpid/cpp/management/python/bin/qpid-printevents b/qpid/cpp/management/python/bin/qpid-printevents new file mode 100755 index 0000000000..f702ca91e8 --- /dev/null +++ b/qpid/cpp/management/python/bin/qpid-printevents @@ -0,0 +1,191 @@ +#!/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 os +import optparse +import sys +from optparse import IndentedHelpFormatter +from time import time, strftime, gmtime, sleep +from threading import Lock, Condition, Thread +from qpid.messaging import Connection +import qpid.messaging.exceptions + +home = os.environ.get("QPID_TOOLS_HOME", os.path.normpath("/usr/share/qpid-tools")) +sys.path.append(os.path.join(home, "python")) + +from qpidtoollibs.broker import EventHelper + + +class Printer(object): + """ + This class serializes printed lines so that events coming from different + threads don't overlap each other. + """ + def __init__(self): + self.lock = Lock() + + def pr(self, text): + self.lock.acquire() + try: + print text + finally: + self.lock.release() + sys.stdout.flush() + + +class EventReceiver(Thread): + """ + One instance of this class is created for each broker that is being monitored. + This class does not use the "reconnect" option because it needs to report as + events when the connection is established and when it's lost. + """ + def __init__(self, printer, url, options): + Thread.__init__(self) + self.printer = printer + self.url = url + self.options = options + self.running = True + self.helper = EventHelper() + + def cancel(self): + self.running = False + + def run(self): + isOpen = False + while self.running: + try: + conn = Connection.establish(self.url, **self.options) + isOpen = True + self.printer.pr(strftime("%c", gmtime(time())) + " NOTIC qpid-printevents:brokerConnected broker=%s" % self.url) + + sess = conn.session() + rx = sess.receiver(self.helper.eventAddress()) + + while self.running: + try: + msg = rx.fetch(1) + event = self.helper.event(msg) + self.printer.pr(event.__repr__()) + sess.acknowledge() + except qpid.messaging.exceptions.Empty: + pass + + except Exception, e: + if isOpen: + self.printer.pr(strftime("%c", gmtime(time())) + " NOTIC qpid-printevents:brokerDisconnected broker=%s" % self.url) + isOpen = False + sleep(1) + + +class JHelpFormatter(IndentedHelpFormatter): + """ + Format usage and description without stripping newlines from usage strings + """ + def format_usage(self, usage): + return usage + + def format_description(self, description): + if description: + return description + "\n" + else: + return "" + +_usage = "%prog [options] [broker-addr]..." + +_description = \ +""" +Collect and print events from one or more Qpid message brokers. + +If no broker-addr is supplied, %prog connects to 'localhost:5672'. + +[broker-addr] syntax: + + [username/password@] hostname + ip-address [:<port>] + +Examples: + +$ %prog localhost:5672 +$ %prog 10.1.1.7:10000 +$ %prog guest/guest@broker-host:10000 +""" + +def main(argv=None): + p = optparse.OptionParser(usage=_usage, description=_description, formatter=JHelpFormatter()) + p.add_option("--heartbeats", action="store_true", default=False, help="Use heartbeats.") + p.add_option("--sasl-mechanism", action="store", type="string", metavar="<mech>", help="SASL mechanism for authentication (e.g. EXTERNAL, ANONYMOUS, PLAIN, CRAM-MD5, DIGEST-MD5, GSSAPI). SASL automatically picks the most secure available mechanism - use this option to override.") + p.add_option("--sasl-service-name", action="store", type="string", help="SASL service name to use") + p.add_option("--ssl-certificate", action="store", type="string", metavar="<cert>", help="Client SSL certificate (PEM Format)") + p.add_option("--ssl-key", action="store", type="string", metavar="<key>", help="Client SSL private key (PEM Format)") + p.add_option("--ha-admin", action="store_true", help="Allow connection to a HA backup broker.") + + options, arguments = p.parse_args(args=argv) + if len(arguments) == 0: + arguments.append("localhost") + + brokers = [] + conn_options = {} + props = {} + printer = Printer() + + if options.sasl_mechanism: + conn_options['sasl_mechanisms'] = options.sasl_mechanism + if options.sasl_service_name: + conn_options['sasl_service'] = options.sasl_service_name + if options.ssl_certificate: + conn_options['ssl_certfile'] = options.ssl_certificate + if options.ssl_key: + if not options.ssl_certificate: + p.error("missing '--ssl-certificate' (required by '--ssl-key')") + conn_options['ssl_keyfile'] = options.ssl_key + if options.ha_admin: + props['qpid.ha-admin'] = 1 + if options.heartbeats: + props['heartbeat'] = 5 + + if len(props) > 0: + conn_options['client_properties'] = props + + try: + try: + for host in arguments: + er = EventReceiver(printer, host, conn_options) + brokers.append(er) + er.start() + + while (True): + sleep(10) + + except KeyboardInterrupt: + print + return 0 + + except Exception, e: + print "Failed: %s - %s" % (e.__class__.__name__, e) + return 1 + finally: + for b in brokers: + b.cancel() + for b in brokers: + b.join() + +if __name__ == '__main__': + sys.exit(main()) |