add agent metrics sending

talker_agent/config.ini

@@ -9,4 +9,5 @@ port = 6379
 logpath = /var/log/talker.log
 
 [metrics]
-enabled = False
+enabled = True
+host = graphite

talker_agent/talker.py

@@ -42,7 +42,7 @@
     from configparser import ConfigParser
 except:  # python 2.7
     from ConfigParser import ConfigParser
-
+import graphyte
 
 PY3 = sys.version_info[0] == 3
 
@@ -709,6 +709,9 @@ def finalize_previous_session(self):
     def start_job(self, job_data_raw):
         job_data = json.loads(job_data_raw)
         cmd = job_data['cmd']
+        if config.parser.getboolean('metrics', 'enabled'):
+            logger.debug('sending command metrics')
+            graphyte.send('commands_received', 1)
         if isinstance(cmd, list):
             if SIMULATE_DROPS_RATE and cmd != ['true'] and random.random() > SIMULATE_DROPS_RATE:
                 logger.warning("dropping job: %(id)s", job_data)
@@ -932,6 +935,8 @@ def main(*args):
 
     config = Config()
     set_logging_to_file(config.parser.get('logging', 'logpath'))
+    if config.parser.getboolean('metrics', 'enabled'):
+        graphyte.init(config.parser.get('metrics', 'host'))
 
     # to help with WEKAPP-74054
     os.system("df")

tests/integration/docker-compose.yml

@@ -2,6 +2,15 @@ version: '3'
 services:
   redis:
     image: "redis:4.0.14-alpine"
+  graphite:
+    image: "graphiteapp/graphite-statsd"
+    container_name: graphite
+    ports:
+      - 80:80
+      - 2003-2004:2003-2004
+      - 2023-2024:2023-2024
+      - 8125:8125/udp
+      - 8126:8126
   talker:
     build:
       context: ../../talker_agent/.
@@ -10,3 +19,4 @@ services:
         - TALKER_HOST_ID=$TALKER_HOST_ID
     links:
       - redis
+      - graphite