From 7344d858c68a03af64f6470407abd5199b39458a Mon Sep 17 00:00:00 2001
From: Reid 'arrdem' McKenzie <me@arrdem.com>
Date: Sun, 12 Dec 2021 21:36:21 -0700
Subject: [PATCH] Fairly viable aloe state

---
 projects/aloe/src/python/aloe/__main__.py | 26 +++++++++++------------
 projects/aloe/src/python/aloe/icmp.py     |  6 +++---
 2 files changed, 16 insertions(+), 16 deletions(-)

diff --git a/projects/aloe/src/python/aloe/__main__.py b/projects/aloe/src/python/aloe/__main__.py
index 1e465b0..2fe07f2 100644
--- a/projects/aloe/src/python/aloe/__main__.py
+++ b/projects/aloe/src/python/aloe/__main__.py
@@ -137,29 +137,29 @@ def retrace(shutdown, q, opts, hl, hosts):
     threads = {}
 
     def create_host(distance, address):
-        if address not in hosts:
-            log.info(f"Monitoring {address}...")
-            with hl:
-                hosts[(distance, address)] = monitor = MonitoredHost(address, timedelta(seconds=8))
-                threads[(distance, address)] = t = Thread(target=monitor, args=(shutdown, q))
-            t.start()
-
-        else:
-            log.debug(f"Already monitoring {address}...")
+        with hl:
+            if address not in hosts:
+                log.info(f"Monitoring {address}...")
+                monitor = MonitoredHost(address, timedelta(seconds=1))
+                hosts[address] = (distance, monitor)
+                threads[address] = t = Thread(target=monitor, args=(shutdown, q))
+                t.start()
 
+            else:
+                log.debug(f"Already monitoring {address}...")
 
     while not shutdown.is_set():
         for h in opts.hosts:
             # FIXME: Use a real topology model
-            for distance, hop in zip(count(1), traceroute(q, h)):
+            for hop in traceroute(q, h):
                 if ping(q, hop.address).is_alive:
-                    create_host(distance, hop.address)
+                    create_host(hop.distance, hop.address)
 
         sleep(shutdown, 60 * 5)
 
 
 def render(shutdown, q, stdscr, hl, hosts):
-    dt = timedelta(seconds=8)
+    dt = timedelta(minutes=30)
 
     with open("incidents.txt", "w") as fp:
         incident = False
@@ -169,7 +169,7 @@ def render(shutdown, q, stdscr, hl, hosts):
             now = datetime.now()
             i = 0
             with hl:
-                for i, ((distance, name), host) in zip(count(1), sorted(hosts.items(), key=lambda x: x[0][0])):
+                for i, (name, (distance, host)) in zip(count(1), sorted(hosts.items(), key=lambda x: x[1][0])):
                     loss = host.state.loss(dt) * 100
                     state = host.state.last()
                     if not state:
diff --git a/projects/aloe/src/python/aloe/icmp.py b/projects/aloe/src/python/aloe/icmp.py
index 8d7e020..3f32a1a 100644
--- a/projects/aloe/src/python/aloe/icmp.py
+++ b/projects/aloe/src/python/aloe/icmp.py
@@ -95,7 +95,7 @@ def icmp_worker(shutdown: Event, q: queue.Queue):
         while not shutdown.is_set():
             # Send one
             try:
-                item = q.get(block=False, timeout=0.1)
+                item = q.get(block=False, timeout=0.001)
                 request = item._request
                 state[(request._id, request._sequence)] = item
                 # log.info(f"Sending request {item._request!r}")
@@ -105,7 +105,7 @@ def icmp_worker(shutdown: Event, q: queue.Queue):
 
             # Recieve one
             try:
-                if response := sock.receive(None, 0.1):
+                if response := sock.receive(None, 0.001):
                     key = (response.id, response.sequence)
                     if key in state:
                         # log.info(f"Got response {response!r}")
@@ -123,7 +123,7 @@ def icmp_worker(shutdown: Event, q: queue.Queue):
                     del state[key]
 
             # Sleep one
-            sleep(shutdown, 0.1)
+            sleep(shutdown, 0.001)
 
 
 def traceroute(q: queue.Queue,