1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
# Support for logging periodic statistics
#
# Copyright (C) 2018-2021 Kevin O'Connor <kevin@koconnor.net>
#
# This file may be distributed under the terms of the GNU GPLv3 license.
import os, time, logging
class PrinterSysStats:
def __init__(self, config):
printer = config.get_printer()
self.last_process_time = self.total_process_time = 0.0
self.last_load_avg = 0.0
self.last_mem_avail = 0
self.mem_file = None
try:
self.mem_file = open("/proc/meminfo", "r")
except:
pass
printer.register_event_handler("klippy:disconnect", self._disconnect)
def _disconnect(self):
if self.mem_file is not None:
self.mem_file.close()
self.mem_file = None
def stats(self, eventtime):
# Get core usage stats
ptime = time.process_time()
pdiff = ptime - self.last_process_time
self.last_process_time = ptime
if pdiff > 0.0:
self.total_process_time += pdiff
self.last_load_avg = os.getloadavg()[0]
msg = "sysload=%.2f cputime=%.3f" % (
self.last_load_avg,
self.total_process_time,
)
# Get available system memory
if self.mem_file is not None:
try:
self.mem_file.seek(0)
data = self.mem_file.read()
for line in data.split("\n"):
if line.startswith("MemAvailable:"):
self.last_mem_avail = int(line.split()[1])
msg = "%s memavail=%d" % (msg, self.last_mem_avail)
break
except:
pass
return (False, msg)
def get_status(self, eventtime):
return {
"sysload": self.last_load_avg,
"cputime": self.total_process_time,
"memavail": self.last_mem_avail,
}
class PrinterStats:
def __init__(self, config):
self.printer = config.get_printer()
reactor = self.printer.get_reactor()
self.stats_timer = reactor.register_timer(self.generate_stats)
self.stats_cb = []
self.printer.register_event_handler("klippy:ready", self.handle_ready)
def handle_ready(self):
self.stats_cb = [
o.stats for n, o in self.printer.lookup_objects() if hasattr(o, "stats")
]
if self.printer.get_start_args().get("debugoutput") is None:
reactor = self.printer.get_reactor()
reactor.update_timer(self.stats_timer, reactor.NOW)
def generate_stats(self, eventtime):
stats = [cb(eventtime) for cb in self.stats_cb]
if max([s[0] for s in stats]):
logging.info("Stats %.1f: %s", eventtime, " ".join([s[1] for s in stats]))
return eventtime + 1.0
def load_config(config):
config.get_printer().add_object("system_stats", PrinterSysStats(config))
return PrinterStats(config)
|
