From 1aa3a82039650bfd91ce71c65428d281c0e4d063 Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Sat, 23 May 2020 18:34:21 +0800
Subject: [PATCH] remove docker files

---
 agent.py | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/agent.py b/agent.py
index 5e74621..e6098b2 100644
--- a/agent.py
+++ b/agent.py
@@ -25,6 +25,8 @@ KafkaBrokers = os.getenv('KafkaBrokers', 'localhost:9092').split(',')
 PORT = os.getenv('Port', 8000)
 HeartbeatInterval = os.getenv('HeartbeatInterval', 5)
 
+EnableEventTrigger = os.getenv('EnableEventTrigger', 'true')
+
 lock = Lock()
 # pending_tasks = {}
 id2token = {}
@@ -381,12 +383,19 @@ if __name__ == '__main__':
 	os.environ["TZ"] = 'Asia/Shanghai'
 	if hasattr(time, 'tzset'):
 		time.tzset()
+	threads = []
 	t1 = Thread(target=reporter)
+	threads.append(t1)
 	t2 = Thread(target=listener)
-	t3 = Thread(target=event_trigger)
-	t1.start()
-	t2.start()
-	t3.start()
-	t1.join()
-	t2.join()
-	t3.join()
+	threads.append(t2)
+	if EnableEventTrigger == 'true':
+		t3 = Thread(target=event_trigger)
+		threads.append(t3)
+
+	# Start all threads
+	for t in threads:
+		t.start()
+
+	# Wait for all of them to finish
+	for t in threads:
+		t.join()