Initial commit

rspamd

@@ -0,0 +1,145 @@
+#!/usr/bin/env python3
+import json
+import os
+import re
+import shlex
+import sys
+
+CONFIG="""\
+multigraph rspamd_actions
+graph_title Rspamd actions
+graph_vlabel mails
+graph_args --base 1000 -l 0
+graph_category antispam
+graph_scale yes
+reject.label Rejection
+soft_reject.label Soft rejection
+rewrite_subject.label Rewrite subject
+add_header.label Set headers
+greylist.label Graylist
+no_action.label No action
+scanned.label Scanned/total
+learned.label Learned
+
+multigraph rspamd_result
+graph_title Rspamd results
+graph_vlabel mails
+graph_args --base 1000 -l 0
+graph_category antispam
+graph_scale yes
+spam.label Spams
+ham.label Ham
+
+multigraph rspamd_scantime
+graph_title Rspamd scantime
+graph_args --base 1000 -l 0
+graph_category antispam
+graph_scale yes
+graph_vlabel seconds
+scantime.label Average scantime
+
+multigraph rspamd_uptime
+graph_title Rspamd uptime
+graph_args --base 1000 -l 0
+graph_category antispam
+graph_scale no
+graph_vlabel uptime in days
+uptime.label uptime
+uptime.label uptime
+
+multigraph rspamd_memory
+graph_title Rspamd memory usage
+graph_args --base 1024 -l 0
+graph_category antispam
+graph_scale yes
+graph_vlabel bytes
+memory.label Memory usage
+
+multigraph rspamd_connections
+graph_title Rspamd connections
+graph_args --base 1000 -l 0
+graph_category antispam
+graph_scale yes
+graph_vlabel connections
+connections.label opened connections
+
+multigraph rspamd_ctrl_connections
+graph_title Rspamd control connections count
+graph_args --base 1000 -l 0
+graph_category antispam
+graph_scale yes
+graph_vlabel connections
+connections.label control connections count
+"""
+
+def storage2name(name):
+    return re.sub('[\.\- ]', '_', name)
+
+rspamc='/usr/bin/rspamc'
+# TODO check env for rspamc path
+cmd = shlex.join([rspamc, '--json', 'stat'])
+
+with os.popen(cmd, mode='r') as pipe:
+    data = json.loads(pipe.read())
+
+
+if len(sys.argv) > 1 and sys.argv[1] == 'config':
+    print(CONFIG)
+    print('''multigraph rspamd_statfiles
+graph_title rspamd bayes statfiles
+graph_args --base 1000 -l 0
+graph_scale yes
+graph_vlabel learned mails''')
+    for stat in data['statfiles']:
+        sym = stat['symbol']
+        print(f'learned_{sym}.label {sym} learned')
+        print(f'users_{sym}.label {sym} users')
+    print()
+
+    print('''multigraph rspamd_fuzzy_hash
+graph_title rspamd fuzzy hases storage
+graph_args --base 1000 -l 0
+graph_scale yes
+graph_vlabel hashes''')
+    for storage in data['fuzzy_hashes']:
+        storage_name = storage2name(storage)
+        print(f'{storage_name}.label {storage}')
+    exit(0)
+
+print('multigraph rspamd_actions')
+for action, value in data['actions'].items():
+    print('%s.value %d' % (action.replace(' ', '_'), value))
+print('scanned.value %d' % data['scanned'])
+print('learned.value %d' % data['learned'])
+
+print('multigraph rspamd_result')
+print('spam.value %d' % data['spam_count'])
+print('ham.value %d' % data['ham_count'])
+
+print('multigraph rspamd_scantime')
+scantimes = [float(t) for t in data['scan_times'] if t]
+scantime = sum(scantimes) / len(scantimes)
+print('scantime.value %f' % scantime)
+
+print('multigraph rspamd_uptime')
+print('uptime.value %f' % (data['uptime']/(3600*24)))
+
+print('multigraph rspamd_memory')
+print('memory.value %d' % data['bytes_allocated'])
+
+print('multigraph rspamd_connections')
+print('connections.value %d' % data['connections'])
+
+print('multigraph rspamd_ctrl_connections')
+print('connections.value %d' % data['control_connections'])
+
+print('multigraph rspamd_statfiles')
+for stat in data['statfiles']:
+    sym = stat['symbol']
+    print(f'learned_{sym}.value %d' % stat['revision'])
+    print(f'users_{sym}.value %d' % stat['users'])
+
+print('multigraph rspamd_fuzzy_hash')
+for storage, count in data['fuzzy_hashes'].items():
+    storage_name = storage2name(storage)
+    print(f'{storage_name}.value %d' % count)