fix(hydro_deploy): only instantiate Localhost once

Fixes #841

Author: shadaj

docs/docs/hydroflow_plus/quickstart/clusters.mdx

@@ -95,23 +95,22 @@ use hydroflow_plus_cli_integration::{DeployProcessSpec, DeployClusterSpec};
 #[tokio::main]
 async fn main() {
     let mut deployment = Deployment::new();
-    let localhost = deployment.Localhost();
     let profile = "dev";
 
     let builder = hydroflow_plus::FlowBuilder::new();
     let (leader, workers) = flow::broadcast::broadcast(&builder);
 
     flow.with_default_optimize()
         .with_process(&leader, DeployProcessSpec::new({
-            HydroflowCrate::new(".", localhost.clone())
+            HydroflowCrate::new(".", deployment.Localhost())
                 .bin("broadcast")
                 .profile(profile)
                 .display_name("leader")
         }))
         .with_cluster(&workers, DeployClusterSpec::new({
             (0..2)
                 .map(|idx| {
-                    HydroflowCrate::new(".", localhost.clone())
+                    HydroflowCrate::new(".", deployment.Localhost())
                         .bin("broadcast")
                         .profile(profile)
                         .display_name(format!("worker/{}", idx))

hydro_deploy/core/src/deployment.rs

@@ -18,19 +18,22 @@ pub struct Deployment {
     pub hosts: Vec<Weak<dyn Host>>,
     pub services: Vec<Weak<RwLock<dyn Service>>>,
     pub resource_pool: ResourcePool,
+    localhost_host: Option<Arc<LocalhostHost>>,
     last_resource_result: Option<Arc<ResourceResult>>,
     next_host_id: usize,
     next_service_id: usize,
 }
 
 impl Deployment {
     pub fn new() -> Self {
-        Self::default()
+        let mut ret = Self::default();
+        ret.localhost_host = Some(ret.add_host(LocalhostHost::new));
+        ret
     }
 
     #[allow(non_snake_case)]
-    pub fn Localhost(&mut self) -> Arc<LocalhostHost> {
-        self.add_host(LocalhostHost::new)
+    pub fn Localhost(&self) -> Arc<LocalhostHost> {
+        self.localhost_host.clone().unwrap()
     }
 
     #[allow(non_snake_case)]

hydro_deploy/core/src/hydroflow_crate/mod.rs

@@ -143,10 +143,8 @@ mod tests {
     async fn test_crate_panic() {
         let mut deployment = deployment::Deployment::new();
 
-        let localhost = deployment.Localhost();
-
         let service = deployment.add_service(
-            HydroflowCrate::new("../hydro_cli_examples", localhost.clone())
+            HydroflowCrate::new("../hydro_cli_examples", deployment.Localhost())
                 .example("panic_program")
                 .profile("dev"),
         );

hydroflow_plus_test/src/cluster/many_to_many.rs

@@ -33,7 +33,6 @@ mod tests {
     #[tokio::test]
     async fn many_to_many() {
         let mut deployment = Deployment::new();
-        let localhost = deployment.Localhost();
 
         let builder = hydroflow_plus::FlowBuilder::new();
         let cluster = super::many_to_many(&builder);
@@ -47,7 +46,7 @@ mod tests {
                 DeployClusterSpec::new({
                     (0..2)
                         .map(|_| {
-                            HydroflowCrate::new(".", localhost.clone())
+                            HydroflowCrate::new(".", deployment.Localhost())
                                 .bin("many_to_many")
                                 .profile("dev")
                         })

hydroflow_plus_test/src/cluster/simple_cluster.rs

@@ -50,7 +50,6 @@ mod tests {
     #[tokio::test]
     async fn simple_cluster() {
         let mut deployment = Deployment::new();
-        let localhost = deployment.Localhost();
 
         let builder = hydroflow_plus::FlowBuilder::new();
         let (node, cluster) = super::simple_cluster(&builder);
@@ -62,7 +61,7 @@ mod tests {
             .with_process(
                 &node,
                 DeployProcessSpec::new({
-                    HydroflowCrate::new(".", localhost.clone())
+                    HydroflowCrate::new(".", deployment.Localhost())
                         .bin("simple_cluster")
                         .profile("dev")
                 }),
@@ -72,7 +71,7 @@ mod tests {
                 DeployClusterSpec::new({
                     (0..2)
                         .map(|_| {
-                            HydroflowCrate::new(".", localhost.clone())
+                            HydroflowCrate::new(".", deployment.Localhost())
                                 .bin("simple_cluster")
                                 .profile("dev")
                         })

hydroflow_plus_test/src/distributed/first_ten.rs

@@ -46,7 +46,6 @@ mod tests {
     #[tokio::test]
     async fn first_ten_distributed() {
         let mut deployment = Deployment::new();
-        let localhost = deployment.Localhost();
 
         let builder = hydroflow_plus::FlowBuilder::new();
         let (first_node, second_node) = super::first_ten_distributed(&builder);
@@ -60,15 +59,15 @@ mod tests {
             .with_process(
                 &first_node,
                 DeployProcessSpec::new({
-                    HydroflowCrate::new(".", localhost.clone())
+                    HydroflowCrate::new(".", deployment.Localhost())
                         .bin("first_ten_distributed")
                         .profile("dev")
                 }),
             )
             .with_process(
                 &second_node,
                 DeployProcessSpec::new({
-                    HydroflowCrate::new(".", localhost.clone())
+                    HydroflowCrate::new(".", deployment.Localhost())
                         .bin("first_ten_distributed")
                         .profile("dev")
                 }),

template/hydroflow_plus/flow/examples/first_ten_distributed.rs

@@ -4,7 +4,6 @@ use hydroflow_plus_cli_integration::DeployProcessSpec;
 #[tokio::main]
 async fn main() {
     let mut deployment = Deployment::new();
-    let localhost = deployment.Localhost();
 
     let flow = hydroflow_plus::FlowBuilder::new();
     let (p1, p2) = flow::first_ten_distributed::first_ten_distributed(&flow);
@@ -14,15 +13,15 @@ async fn main() {
         .with_process(
             &p1,
             DeployProcessSpec::new({
-                HydroflowCrate::new(".", localhost.clone())
+                HydroflowCrate::new(".", deployment.Localhost())
                     .bin("first_ten_distributed")
                     .profile("dev")
             }),
         )
         .with_process(
             &p2,
             DeployProcessSpec::new({
-                HydroflowCrate::new(".", localhost.clone())
+                HydroflowCrate::new(".", deployment.Localhost())
                     .bin("first_ten_distributed")
                     .profile("dev")
             }),

template/hydroflow_plus/flow/src/first_ten_distributed.rs

@@ -41,7 +41,6 @@ mod tests {
     #[tokio::test]
     async fn first_ten_distributed() {
         let mut deployment = Deployment::new();
-        let localhost = deployment.Localhost();
 
         let flow = hydroflow_plus::FlowBuilder::new();
         let (p1, p2) = super::first_ten_distributed(&flow);
@@ -51,15 +50,15 @@ mod tests {
             .with_process(
                 &p1,
                 DeployProcessSpec::new({
-                    HydroflowCrate::new(".", localhost.clone())
+                    HydroflowCrate::new(".", deployment.Localhost())
                         .bin("first_ten_distributed")
                         .profile("dev")
                 }),
             )
             .with_process(
                 &p2,
                 DeployProcessSpec::new({
-                    HydroflowCrate::new(".", localhost.clone())
+                    HydroflowCrate::new(".", deployment.Localhost())
                         .bin("first_ten_distributed")
                         .profile("dev")
                 }),