Added reporter

svm_prefilter_service/mule.py

@@ -15,6 +15,8 @@ from pyAudioAnalysis import audioBasicIO
 from pyAudioAnalysis import MidTermFeatures
 import numpy
 
+from apscheduler.schedulers.background import BackgroundScheduler
+
 if Config.SENTRY_DSN:
     sentry_sdk.init(
         dsn=Config.SENTRY_DSN,
@@ -119,12 +121,37 @@ def lapatolas(q: Queue):
         q.put(message)
 
 
+def reporter(q: Queue):
+    report = {
+        "site": Config.REPORT_ALIAS,
+        "measurements": {
+            "queue": q.qsize()
+        }
+    }
+
+    print("Reporting queue length of", report)
+
+    r = requests.post(Config.REPORT_URL, json=report)
+
+    r.raise_for_status()
+    if r.status_code != 201:
+        print(Config.REPORT_URL, "Wrong response:", r.status_code)
+
+
 def main():
     memer = ModelMemer()
     requeue = Queue()
 
     Thread(target=lapatolas, args=(requeue,), daemon=True).start()
 
+    scheduler = None
+    if Config.REPORT_URL:
+        scheduler = BackgroundScheduler()
+        scheduler.add_job(lambda: reporter(requeue), trigger='interval', seconds=Config.REPORT_INTERVAL)
+        scheduler.start()
+
+    Thread(target=reporter, args=(requeue,), daemon=True).start()
+
     while True:
         message = requeue.get(block=True)
         task = pickle.loads(message)
@@ -149,6 +176,9 @@ def main():
         finally:
             os.remove(audio_file_path)
 
+#    if scheduler:
+#        scheduler.stop()
+
 
 if __name__ == '__main__':
     main()