[counterpoll]: cli to change counter polling configuration (sonic-net…

…#257)

Signed-off-by: Sihui Han <sihan@microsoft.com>

counterpoll/main.py

@@ -0,0 +1,59 @@
+#! /usr/bin/python -u
+
+import click
+import swsssdk
+from tabulate import tabulate
+
+@click.group()
+def cli():
+    """ SONiC Static Counter Poll configurations """
+
+# Queue counter commands
+@cli.group()
+def queue():
+    """ Queue counter commands """
+
+@queue.command()
+@click.argument('poll_interval', type=click.IntRange(100, 30000))
+def interval(poll_interval):
+    """ Set queue counter query interval """
+    configdb = swsssdk.ConfigDBConnector()
+    configdb.connect()
+    queue_info = {}
+    if poll_interval is not None:
+        queue_info['POLL_INTERVAL'] = poll_interval
+    configdb.mod_entry("FLEX_COUNTER_TABLE", "QUEUE", queue_info)
+
+# Port counter commands
+@cli.group()
+def port():
+    """ Queue counter commands """
+
+@port.command()
+@click.argument('poll_interval', type=click.IntRange(100, 30000))
+def interval(poll_interval):
+    """ Set queue counter query interval """
+    configdb = swsssdk.ConfigDBConnector()
+    configdb.connect()
+    port_info = {}
+    if poll_interval is not None:
+        port_info['POLL_INTERVAL'] = poll_interval
+    configdb.mod_entry("FLEX_COUNTER_TABLE", "test", port_info)
+
+@cli.command()
+def show():
+    """ Show the counter configuration """
+    configdb = swsssdk.ConfigDBConnector()
+    configdb.connect()
+    queue_info = configdb.get_entry('FLEX_COUNTER_TABLE', 'QUEUE')
+    port_info = configdb.get_entry('FLEX_COUNTER_TABLE', 'PORT')
+
+    header = ("Type", "Interval")
+    data = []
+    if queue_info:
+        data.append(["QUEUE_STAT", queue_info["POLL_INTERVAL"]])
+    if port_info:
+        data.append(["PORT_STAT", port_info["POLL_INTERVAL"]])
+
+    print tabulate(data, headers=header, tablefmt="simple", missingval="")
+

data/etc/bash_completion.d/counterpoll

@@ -0,0 +1,8 @@
+_counterpoll_completion() {
+    COMPREPLY=( $( env COMP_WORDS="${COMP_WORDS[*]}" \
+                   COMP_CWORD=$COMP_CWORD \
+                   _COUNTERPOLL_COMPLETE=complete $1 ) )
+    return 0
+}
+
+complete -F _counterpoll_completion -o default counterpoll;

setup.py

@@ -21,6 +21,7 @@ def get_test_suite():
         'acl_loader',
         'clear',
         'config',
+        'counterpoll',
         'crm',
         'debug',
         'pfcwd',
@@ -63,6 +64,7 @@ def get_test_suite():
         'console_scripts': [
             'acl-loader = acl_loader.main:cli',
             'config = config.main:cli',
+            'counterpoll = counterpoll.main:cli',
             'crm = crm.main:cli',
             'debug = debug.main:cli',
             'pfcwd = pfcwd.main:cli',