replay functionality added as example

Author: au650680

record/replay.py

@@ -15,7 +15,7 @@
 RECORDINGS_DIR = "record/mqtt_recordings"
 FILE_NAME = "recording_beam_reduced.jsonl"
 
-REPLAY_SPEED = 1  # Multiplier for replay speed
+REPLAY_SPEED = 10  # Multiplier for replay speed
 
 BUSY_WAIT_THRESHOLD = 10/1000  # Threshold in seconds for busy waiting (10 ms)
 KEEP_UP_TIME = -1 # If delay time (remaining) is lower than this time, warn the user that the replay speed is two fast.
@@ -75,7 +75,7 @@ def publish_massage(publish_client: MQTTClient, PublishTopics: Dict[str,str], qo
     except KeyboardInterrupt:
         raise RuntimeError("Replay interrupted by user.")
 
-def replay_mqtt_messages(loop: int = 1) -> None:
+def replay_mqtt_messages(config_path: str, loop: int = 1) -> None:
     """
     Replay data using jsonl file
     
@@ -84,7 +84,7 @@ def replay_mqtt_messages(loop: int = 1) -> None:
     Returns:
         None
     """
-    config = load_config(CONFIG_PATH)
+    config = load_config(config_path)
     MQTT_config = config["MQTT"]
     publish_client = setup_publish_client(MQTT_config)
     try:
@@ -173,4 +173,4 @@ def replay_mqtt_messages(loop: int = 1) -> None:
         print("[DONE].")
 
 if __name__ == "__main__":
-    replay_mqtt_messages(loop=LOOPS) # Times to loop
+    replay_mqtt_messages(CONFIG_PATH, loop=LOOPS) # Times to loop

src/examples/example.py

@@ -24,6 +24,7 @@
     run_model_update_remote_sysid,
     run_live_model_update_remote_clustering
 )
+from examples.run_replay import replay_mqtt
 
 @click.group()
 @click.option('--config', default="config/production.json", help="Path to config file")
@@ -42,6 +43,11 @@ def accelerometers(ctx):
 def align_readings(ctx):
     align_acceleration_readings(ctx.obj["CONFIG"])
 
+@cli.command()
+@click.pass_context
+def replay(ctx):
+    replay_mqtt(ctx.obj["CONFIG"])
+
 @cli.command()
 @click.pass_context
 def sysid_and_print(ctx):

src/examples/run_replay.py

@@ -0,0 +1,4 @@
+from record import replay
+
+def replay_mqtt(config_path):
+    replay.replay_mqtt_messages(config_path)