From 12fd2b6b6e382e7f0e671896e73e0f14a857bbdf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ebbe=20Ba=C3=9F?= <ebbe.bass>
Date: Mon, 19 Feb 2024 16:51:45 +0100
Subject: [PATCH] added pstats
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Ebbe Baß <ebbe.bass>
---
 server/app.py | 30 ++++++++++++++++++------------
 1 file changed, 18 insertions(+), 12 deletions(-)

diff --git a/server/app.py b/server/app.py
index f8f30da..4f83a6e 100644
--- a/server/app.py
+++ b/server/app.py
@@ -6,10 +6,11 @@ import python_artnet as Artnet
 import os
 from getmac import get_mac_address
 import time
-import sys
 from multiprocessing import Process, Pipe, Queue
 from queue import Empty
 from ast import literal_eval
+import cProfile
+import pstats
 
 app = Flask(__name__)
 
@@ -144,7 +145,6 @@ def mqtt_publisher(ti_receiver):
 
         except KeyboardInterrupt:
             artNet.close()
-            sys.exit()
 
 def tube_index_updater(ti_queue):
     while True:
@@ -172,13 +172,19 @@ def tube_index_collector(ti_queue, ti_sender):
                 ti_sender.send(None)
 
 if __name__ == "__main__":
-    (ti_receiver,ti_sender) = Pipe(True)
-    ti_queue = Queue()
-    collector_thread = Process(target=tube_index_collector, args=(ti_queue, ti_sender, ))
-    collector_thread.start()
-    ti_updater_thread = Process(target=tube_index_updater, args=(ti_queue, ))
-    ti_updater_thread.start()
-    publisher_thread = Process(target=mqtt_publisher, args=(ti_receiver, ))
-    publisher_thread.start()
-    flask_thread = Process(target=flask_api)
-    flask_thread.start()
\ No newline at end of file
+    with cProfile.Profile as profile:
+        try:
+            (ti_receiver,ti_sender) = Pipe(True)
+            ti_queue = Queue()
+            collector_thread = Process(target=tube_index_collector, args=(ti_queue, ti_sender, ))
+            collector_thread.start()
+            ti_updater_thread = Process(target=tube_index_updater, args=(ti_queue, ))
+            ti_updater_thread.start()
+            publisher_thread = Process(target=mqtt_publisher, args=(ti_receiver, ))
+            publisher_thread.start()
+            flask_thread = Process(target=flask_api)
+            flask_thread.start()
+        except KeyboardInterrupt:
+            result = pstats.Stats(profile)
+            result.sort_stats(pstats.SortKey.TIME)
+            result.print_stats()
\ No newline at end of file