3diy
/
moonraker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

machine: fix perfomance regression 

The `update_service_status()` method could result in the creation of multiple subprocesses for each update, resulting in a performance hit as subprocesses are inherently expensive.

Fetch the state for all detected services in a single subprocess.

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
Eric Callahan 2021-11-03 09:10:12 -04:00
parent 6a39bbd0c3
commit 105858b0c7
1 changed files with 18 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
moonraker/components/machine.py
View File

@ -386,27 +386,29 @@ class Machine:
}
async def update_service_status(self, notify: bool = True) -> None:
if not self.available_services:
return
svcs = self.system_info['available_services']
shell_cmd: SCMDComp = self.server.lookup_component('shell_command')
for svc, state in list(self.available_services.items()):
scmd = shell_cmd.build_shell_command(
f"systemctl show -p ActiveState,SubState --value {svc}")
try:
resp = await scmd.run_with_response(log_complete=False)
active_state, sub_state = resp.strip().split('\n', 1)
scmd = shell_cmd.build_shell_command(
"systemctl show -p ActiveState,SubState --value "
f"{' '.join(svcs)}")
try:
resp = await scmd.run_with_response(log_complete=False)
for svc, state in zip(svcs, resp.strip().split('\n\n')):
active_state, sub_state = state.split('\n', 1)
new_state: Dict[str, str] = {
'active_state': active_state,
'sub_state': sub_state
}
except Exception:
new_state = {
'active_state': "error",
'sub_state': "error"
}
if state != new_state and notify:
self.server.send_event("machine:service_state_changed",
{svc: new_state})
self.available_services[svc] = new_state
if self.available_services[svc] != new_state:
self.available_services[svc] = new_state
if notify:
self.server.send_event(
"machine:service_state_changed",
{svc: new_state})
except Exception:
logging.exception("Error processing service state update")
def load_component(config: ConfigHelper) -> Machine:
return Machine(config)