From e2c47d30969558351799b85eac91b4a5db6c756e Mon Sep 17 00:00:00 2001
From: Ferdinand Schober <ferdinand.schober@fau.de>
Date: Wed, 17 Jan 2024 00:22:24 +0100
Subject: [PATCH] fix: initial dns resolve was not working

---
 src/server.rs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/server.rs b/src/server.rs
index 10ba9a5..cc2815f 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -199,6 +199,7 @@ impl Server {
         let producer_notify = producer_notify_tx.clone();
         let consumer_notify = consumer_notify_tx.clone();
         let frontend_ch = frontend_tx.clone();
+        let resolve_ch = resolve_tx.clone();
         let mut frontend_task = tokio::task::spawn_local(async move {
             loop {
                 tokio::select! {
@@ -214,7 +215,7 @@ impl Server {
                     }
                     event = frontend_rx.recv() => {
                         let frontend_event = event.ok_or(anyhow!("frontend channel closed"))?;
-                        if server.handle_frontend_event(&producer_notify, &consumer_notify, &resolve_tx, &mut frontend, &port_tx, frontend_event).await {
+                        if server.handle_frontend_event(&producer_notify, &consumer_notify, &resolve_ch, &mut frontend, &port_tx, frontend_event).await {
                             break;
                         }
                     }
@@ -423,16 +424,19 @@ impl Server {
             .get_client_states()
             .filter_map(|s| {
                 if s.active {
-                    Some(s.client.handle)
+                    Some((s.client.handle, s.client.hostname.clone()))
                 } else {
                     None
                 }
             })
             .collect::<Vec<_>>();
-        for client in active {
+        for (handle, hostname) in active {
             frontend_tx
-                .send(FrontendEvent::ActivateClient(client, true))
+                .send(FrontendEvent::ActivateClient(handle, true))
                 .await?;
+            if let Some(hostname) = hostname {
+                let _ = resolve_tx.send((hostname, handle)).await;
+            }
         }
 
         tokio::select! {