Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
96 lines (77 sloc) 3.72 KB
import argparse
import json
import sys
def escape_id(id_str):
return id_str.replace('-', '').replace('.', '_')
def build_definitions(definitions, vhost, render_producers, render_consumers):
def is_same_vhost(entity_dict):
return entity_dict['vhost'] == vhost
return ''.join([
'digraph {\n',
' bgcolor=transparent;\n',
' truecolor=true;\n',
' rankdir=LR;\n',
' node [style="filled"];\n\n',
''.join([build_queue(q, render_consumers) for q in filter(is_same_vhost, definitions['queues'])]),
''.join([build_exchange(x, render_producers) for x in filter(is_same_vhost, definitions['exchanges'])]),
''.join([build_binding(b) for b in filter(is_same_vhost, definitions['bindings'])]),
''.join([build_dlx_link(q) for q in filter(is_same_vhost, definitions['queues'])]),
'}'])
def build_queue(queue, render_consumers):
q_name = queue['name']
q_id = escape_id(q_name)
lines = [
' subgraph cluster_Q_%s {\n' % q_id,
' label="%s";\n' % q_name,
' color=transparent;\n',
' "Q_%s" [label="{||||}", fillcolor="red", shape="record"];\n' % q_id,
' }\n\n']
if render_consumers:
lines.append(' "C_%s" [label="C", fillcolor="#33ccff"];' % q_id)
lines.append(' "Q_%(name)s" -> "C_%(name)s"' % { 'name': q_id})
return ''.join(lines)
def build_exchange(exchange, render_producers):
x_name = exchange['name']
x_id = escape_id(x_name)
lines = [
' subgraph cluster_X_%s {\n' % x_id,
' label="%s\\ntype=%s";\n' % (x_name, exchange['type']),
' color=transparent;\n',
' "X_%s" [label="X", fillcolor="#3333CC", shape="ellipse"];\n' % x_id,
' }\n\n']
if render_producers:
lines.append(' "P_%s" [label="P", style="filled", fillcolor="#00ffff"];' % x_id)
lines.append(' "P_%(name)s" -> "X_%(name)s";' % { 'name': x_id })
return ''.join(lines)
def build_binding(binding):
if binding['destination_type'] == 'exchange':
return ' X_%s -> X_%s [label="%s"];\n' % (escape_id(binding['source']), escape_id(binding['destination']), binding['routing_key'])
else:
return ' X_%s -> Q_%s [label="%s"];\n' % (escape_id(binding['source']), escape_id(binding['destination']), binding['routing_key'])
def build_dlx_link(queue):
q_args = queue.get('arguments', None)
if(q_args):
dlx_name = q_args.get('x-dead-letter-exchange', None)
msg_ttl = q_args.get('x-message-ttl', None)
if(dlx_name):
ttl_txt = ''
if(msg_ttl):
ttl_txt = '\nMessage TTL: %sms' % msg_ttl
return ' Q_%s -> X_%s [label="DLX%s", style="dotted"];\n' % (escape_id(queue['name']), escape_id(dlx_name), ttl_txt)
return ''
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('-d', '--definitions', nargs='?', type=argparse.FileType('r'), default=sys.stdin,
help='Definitions file')
parser.add_argument('-o', '--outfile', nargs='?', type=argparse.FileType('w'), default=sys.stdout,
help='Output file')
parser.add_argument('-p', '--producers', action='store_true', help='Render producers')
parser.add_argument('-c', '--consumers', action='store_true', help='Render consumers')
parser.add_argument('-x', '--vhost', nargs='?', default='/', help='Restrict to this vhost')
return parser.parse_args()
if __name__ == '__main__':
args = parse_args()
definitions = json.load(args.definitions)
args.definitions.close()
args.outfile.write(build_definitions(definitions, args.vhost, args.producers, args.consumers))
args.outfile.close()