fixing compiler issues

dyc3 · Feb 9, 2024 · d7f892a · d7f892a
1 parent 8c6d093
commit d7f892a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/crates/ott-balancer/src/config.rs b/crates/ott-balancer/src/config.rs
@@ -4,7 +4,9 @@ use clap::{Parser, ValueEnum};
 use figment::providers::Format;
 use serde::Deserialize;
 
-use crate::discovery::{FlyDiscoveryConfig, HarnessDiscoveryConfig, ManualDiscoveryConfig};
+use crate::discovery::{
+    DnsDiscoveryConfig, FlyDiscoveryConfig, HarnessDiscoveryConfig, ManualDiscoveryConfig,
+};
 
 static mut CONFIG: Option<BalancerConfig> = None;
 

diff --git a/crates/ott-balancer/src/discovery.rs b/crates/ott-balancer/src/discovery.rs
@@ -9,6 +9,7 @@ mod fly;
 mod harness;
 mod manual;
 
+pub use dns::*;
 pub use fly::*;
 pub use harness::*;
 pub use manual::*;

diff --git a/crates/ott-balancer/src/lib.rs b/crates/ott-balancer/src/lib.rs
@@ -65,6 +65,10 @@ pub async fn run() -> anyhow::Result<()> {
 
     info!("Starting monolith discovery");
     let _discovery_handle = match &config.discovery {
+        DiscoveryConfig::Dns(config) => {
+            let discovery = discovery::DnsMonolithDiscoverer::new(config.clone());
+            start_discovery_task(discovery, discovery_tx)
+        }
         DiscoveryConfig::Fly(config) => {
             let discovery = discovery::FlyMonolithDiscoverer::new(config.clone());
             start_discovery_task(discovery, discovery_tx)