From: Eric Leblond <eric@regit.org>
Date: Fri, 15 Mar 2013 09:24:32 +0000 (+0100)
Subject: Add one shot run option to suri-graphite.
X-Git-Tag: suricata-2.0beta1~110
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=f7c5026c314296f2fd0410274bc4377864b398be;p=thirdparty%2Fsuricata.git

Add one shot run option to suri-graphite.
---

diff --git a/contrib/suri-graphite b/contrib/suri-graphite
index c7d675b4f8..7a71ecf2dd 100755
--- a/contrib/suri-graphite
+++ b/contrib/suri-graphite
@@ -23,6 +23,7 @@ import argparse
 parser = argparse.ArgumentParser(prog='suri-graphite', description='Export suricata stats to Graphite')
 parser.add_argument('-H', '--host', default='localhost', help='Host running Graphite')
 parser.add_argument('-P', '--port', default=2003, help='Port of Graphite data socket')
+parser.add_argument('-O', '--oneshot', action='store_const', const=True, help='Send one update and exit', default=False)
 parser.add_argument('-D', '--delay', default=10, help='Delay between data dump')
 parser.add_argument('-r', '--root', default='suricata.perf', help='Prefix of data name in Graphite')
 parser.add_argument('socket', help='suricata socket file to connect to',
@@ -45,4 +46,6 @@ while 1:
             sck.send("%s.%s.%s %s %d\n" % (args.root, thread , counter, res[thread][counter], tnow))
             if args.verbose:
                 print "%s.%s.%s %s %d\n" % (args.root, thread , counter, res[thread][counter], tnow)
+    if args.oneshot:
+        break
     time.sleep(float(args.delay))