Added influx stuff

birb_latency_collector/config.py

@@ -8,10 +8,8 @@ class Config:
     RELEASE_ID = os.environ.get("RELEASE_ID", "test")
     RELEASEMODE = os.environ.get("RELEASEMODE", "dev")
 
-    INFLUXDB_HOST = os.environ['INFLUXDB_HOST']
-    INFLUXDB_PORT = os.environ['INFLUXDB_PORT']
-    INFLUXDB_USER = os.environ['INFLUXDB_USER']
-    INFLUXDB_PASSWORD = os.environ['INFLUXDB_PASSWORD']
-    INFLUXDB_DATABASE = os.environ['INFLUXDB_DATABASE']
+    INFLUXDB_V2_URL = os.environ['INFLUXDB_URL']
+    INFLUXDB_V2_TOKEN = os.environ['INFLUXDB_TOKEN']
+    INFLUXDB_V2_ORG = os.environ['INFLUXDB_ORG']
 
     INFLUXDB_TIMEOUT = 30

birb_latency_collector/utils/influx.py

@@ -1,4 +1,4 @@
 #!/usr/bin/env python3
-from flask_influxdb import InfluxDB
+from flask_influxdb2 import InfluxDB2
 
-influxdb_instance = InfluxDB()
+influxdb_instance = InfluxDB2()

birb_latency_collector/views/report_view.py

@@ -1,7 +1,26 @@
 #!/usr/bin/env python3
 from flask import jsonify, request, abort, current_app, Response
 from flask_classful import FlaskView
+from utils import json_required, influxdb_instance
+from influxdb_client import Point
 
 
 class ReportView(FlaskView):
-    pass
+
+    @json_required
+    def post(self):
+        data = request.json
+
+        points = []
+
+        for type_ in ['latency', 'queue', 'rate']:
+            if type_ in data['measurements']:
+                points.append(
+                    Point(type_)
+                        .tag("client", data['client'])
+                        .tag("cloud", data['cloud'])
+                        .field("latency", data['measurements'][type_])
+                )
+
+        influxdb_instance.connection.write_api().write("alma", "alma", points)
+        return Response(status=201)