def agent_activemq_main(args: Args) -> None:
api_url = parse_api_url(
server_address=args.servername,
api_path="/admin/xml/",
port=args.port,
protocol=args.protocol,
)
auth = None
if args.username:
auth = HTTPBasicAuth(args.username, args.password)
session = create_api_connect_session(api_url, auth=auth)
try:
response = session.get("queues.jsp")
if response.status_code == 401:
raise Exception("Unauthorized")
xml = response.text
data = ET.fromstring(xml)
except Exception as e:
sys.stderr.write(
"Unable to connect. Credentials might be incorrect: %s\n" % e)
return
attributes = ['size', 'consumerCount', 'enqueueCount', 'dequeueCount']
count = 0
output_lines = []
try:
if not args.piggyback:
output_lines.append("<<<mq_queues>>>")
for line in data:
count += 1
if args.piggyback:
output_lines.append("<<<<%s>>>>" % line.get('name'))
output_lines.append("<<<mq_queues>>>")
output_lines.append("[[%s]]" % line.get('name'))
stats = line.findall('stats')
values = ""
for job in attributes:
values += "%s " % stats[0].get(job)
output_lines.append(values)
if args.piggyback:
output_lines.append("<<<<>>>>")
output_lines.append("<<<local:sep(0)>>>")
output_lines.append("0 Active_MQ - Found %s Queues in total" %
count)
except Exception as e: # Probably an IndexError
sys.stderr.write(
"Unable to process data. Returned data might be incorrect: %r" % e)
return
print("\n".join(output_lines))
def generate_api_session(connection_options):
if "url_custom" in connection_options:
api_url = parse_api_custom_url(
url_custom=connection_options["url_custom"],
api_path="api/v1/",
protocol=connection_options["protocol"],
)
else:
api_url = parse_api_url(
server_address=connection_options["address"],
api_path="api/v1/",
protocol=connection_options["protocol"],
port=connection_options["port"],
)
return create_api_connect_session(
api_url,
auth=connection_options.get("auth"),
token=connection_options.get("token"),
)
def main(sys_argv=None):
if sys_argv is None:
sys_argv = sys.argv[1:]
short_options = ""
long_options = ["piggyback", "servername=", "port=", "username="******"password="******"%s\n" % err)
return 1
opt_servername = None
opt_port = None
opt_username = None
opt_password = None
opt_piggyback_mode = False
opt_protocol = "http"
for o, a in opts:
if o in ['--piggyback']:
opt_piggyback_mode = True
elif o in ['--servername']:
opt_servername = a
elif o in ['--port']:
opt_port = a
elif o in ['--username']:
opt_username = a
elif o in ['--password']:
opt_password = a
elif o in ['--protocol']:
opt_protocol = a
if not opt_servername or not opt_port:
usage()
return 1
api_url = parse_api_url(
server_address=opt_servername,
api_path="/admin/xml/",
port=opt_port,
protocol=opt_protocol,
)
auth = None
if opt_username:
auth = HTTPBasicAuth(opt_username, opt_password)
session = create_api_connect_session(api_url, auth=auth)
try:
response = session.get("queues.jsp")
if response.status_code == 401:
raise Exception("Unauthorized")
xml = response.text
data = ET.fromstring(xml)
except Exception as e:
sys.stderr.write("Unable to connect. Credentials might be incorrect: %s\n" % e)
return 1
attributes = ['size', 'consumerCount', 'enqueueCount', 'dequeueCount']
count = 0
output_lines = []
try:
if not opt_piggyback_mode:
output_lines.append("<<<mq_queues>>>")
for line in data:
count += 1
if opt_piggyback_mode:
output_lines.append("<<<<%s>>>>" % line.get('name'))
output_lines.append("<<<mq_queues>>>")
output_lines.append("[[%s]]" % line.get('name'))
stats = line.findall('stats')
values = ""
for job in attributes:
values += "%s " % stats[0].get(job)
output_lines.append(values)
if opt_piggyback_mode:
output_lines.append("<<<<>>>>")
output_lines.append("<<<local:sep(0)>>>")
output_lines.append("0 Active_MQ - Found %s Queues in total" % count)
except Exception as e: # Probably an IndexError
sys.stderr.write("Unable to process data. Returned data might be incorrect: %r" % e)
return 1
print("\n".join(output_lines))