rick/LEDMatrix
1
0
Fork 0
mirror of https://github.com/ChuckBuilds/LEDMatrix.git synced 2026-04-10 21:03:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

added threaded worker to separate state updates

Browse Source
This commit is contained in:
ChuckBuilds
2025-05-25 20:39:23 -05:00
parent f053963f43
commit f2b632400a
2 changed files with 96 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/ytm_client.py
View File

@@ -4,6 +4,7 @@ import json
import os
import time
import threading
from concurrent.futures import ThreadPoolExecutor
# Ensure application-level logging is configured (as it is)
# logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -35,6 +36,8 @@ class YTMClient:
self._data_lock = threading.Lock()
self._connection_event = threading.Event()
self.external_update_callback = update_callback
# For offloading external_update_callback to prevent blocking socketio thread
self._callback_executor = ThreadPoolExecutor(max_workers=1, thread_name_prefix='ytm_callback_worker')
@self.sio.event(namespace='/api/v1/realtime')
def connect():
@@ -66,12 +69,12 @@ class YTMClient:
logging.debug(f"YTM state update received. Title: {title}. Callback Exists: {self.external_update_callback is not None}")
if self.external_update_callback:
logging.debug(f"--> Attempting to call YTM external_update_callback for title: {title}")
logging.debug(f"--> Submitting YTM external_update_callback for title: {title} to executor")
try:
# Pass the full 'data' object to the callback
self.external_update_callback(data)
# Offload the callback to the executor
self._callback_executor.submit(self.external_update_callback, data)
except Exception as cb_ex:
logging.error(f"Error executing YTMClient external_update_callback: {cb_ex}")
logging.error(f"Error submitting YTMClient external_update_callback to executor: {cb_ex}")
def load_config(self):
default_url = "http://localhost:9863"
@@ -180,6 +183,16 @@ class YTMClient:
else:
logging.debug("YTM Socket.IO client already disconnected or not connected.")
def shutdown(self):
"""Shuts down the callback executor."""
logging.info("YTMClient: Shutting down callback executor...")
if self._callback_executor:
self._callback_executor.shutdown(wait=True) # Wait for pending tasks to complete
self._callback_executor = None # Clear reference
logging.info("YTMClient: Callback executor shut down.")
else:
logging.debug("YTMClient: Callback executor already None or not initialized.")
# Example Usage (for testing - needs to be adapted for Socket.IO async nature)
# if __name__ == '__main__':
# client = YTMClient()