diff --git a/components/event_store/service/event_logger.py b/components/event_store/service/event_logger.py
index d886372..8dfca4f 100644
--- a/components/event_store/service/event_logger.py
+++ b/components/event_store/service/event_logger.py
@@ -1,10 +1,11 @@
+import json
import pickle
-from redis import StrictRedis
from dse_shared_libs.daf import DAF
+from dse_shared_libs.message_broker_wrapper import MBWrapper
from dse_shared_libs.target_velocity import TargetVelocity
from dse_shared_libs.traffic_light_state import TrafficLightState
-from dse_shared_libs.message_broker_wrapper import MBWrapper
+from redis import StrictRedis
class EventLogger:
@@ -13,6 +14,17 @@ class EventLogger:
self.log_to_redis = log_to_redis
self.verbose = verbose
+ def setup_msq_queue(self):
+ mb = MBWrapper(exchange_name='logger', callback=self.log)
+ mb.setup_receiver()
+
+ def get_keys(self):
+ return json.dumps([key.decode() for key in self.redis.keys()])
+
+ def get_value_of(self, key):
+ key = self.redis.get(key)
+ return key.decode() if key else json.dumps("")
+
def log(self, msg):
try:
msg = pickle.loads(msg)
@@ -39,7 +51,3 @@ class EventLogger:
self.redis.append(key, "{}
".format(to_log))
if self.verbose:
print(to_log)
-
- def setup_msq_queue(self):
- mb = MBWrapper(exchange_name='logger', callback=self.log)
- mb.setup_receiver()
diff --git a/components/event_store/service/event_store_server.py b/components/event_store/service/event_store_server.py
index 0ec461f..837e96a 100644
--- a/components/event_store/service/event_store_server.py
+++ b/components/event_store/service/event_store_server.py
@@ -1,4 +1,3 @@
-import json
import threading
from flask import Flask
@@ -25,13 +24,12 @@ redis: StrictRedis = Redis(app)
@app.route("/api/keys")
def api_get_keys():
- return json.dumps([key.decode() for key in redis.keys()])
+ return el.get_keys()
@app.route("/api/keys//")
def api_get_of_key(key):
- key = redis.get(key)
- return key.decode() if key else json.dumps("")
+ return el.get_value_of(key)
@app.route("/")