def update(self):
with log_errors():
outgoing = self.worker.outgoing_transfer_log
n = self.worker.outgoing_count - self.last_outgoing
n = min(n, 1000)
outgoing = [outgoing[-i].copy() for i in range(1, n)]
self.last_outgoing = self.worker.outgoing_count
incoming = self.worker.incoming_transfer_log
n = self.worker.incoming_count - self.last_incoming
n = min(n, 1000)
incoming = [incoming[-i].copy() for i in range(1, n)]
self.last_incoming = self.worker.incoming_count
out = []
for msg in incoming:
if msg["keys"]:
d = self.process_msg(msg)
d["inout-color"] = "red"
out.append(d)
for msg in outgoing:
if msg["keys"]:
d = self.process_msg(msg)
d["inout-color"] = "blue"
out.append(d)
if out:
out = transpose(out)
if (len(self.source.data["stop"]) and
min(out["start"]) > self.source.data["stop"][-1] + 10):
update(self.source, out)
else:
self.source.stream(out, rollover=1000)
def update(self):
with log_errors():
outgoing = self.worker.outgoing_transfer_log
n = self.worker.outgoing_count - self.last_outgoing
outgoing = [outgoing[-i].copy() for i in range(1, n + 1)]
self.last_outgoing = self.worker.outgoing_count
incoming = self.worker.incoming_transfer_log
n = self.worker.incoming_count - self.last_incoming
incoming = [incoming[-i].copy() for i in range(1, n + 1)]
self.last_incoming = self.worker.incoming_count
for [msgs, source] in [
[incoming, self.incoming],
[outgoing, self.outgoing],
]:
for msg in msgs:
if "compressed" in msg:
del msg["compressed"]
del msg["keys"]
bandwidth = msg["total"] / (msg["duration"] or 0.5)
bw = max(min(bandwidth / 500e6, 1), 0.3)
msg["alpha"] = bw
try:
msg["y"] = self.who[msg["who"]]
except KeyError:
self.who[msg["who"]] = len(self.who)
msg["y"] = self.who[msg["who"]]
msg["hover"] = "%s / %s = %s/s" % (
format_bytes(msg["total"]),
format_time(msg["duration"]),
format_bytes(msg["total"] / msg["duration"]),
)
for k in ["middle", "duration", "start", "stop"]:
msg[k] = msg[k] * 1000
if msgs:
msgs = transpose(msgs)
if (len(source.data["stop"]) and min(msgs["start"]) >
source.data["stop"][-1] + 10000):
source.data.update(msgs)
else:
source.stream(msgs, rollover=10000)
