Configurable scheduler external host

systemxlabs · Apr 1, 2024 · 96c5320 · 96c5320
1 parent 30885ef
commit 96c5320
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/ballista/core/src/serde/generated/ballista.rs b/ballista/core/src/serde/generated/ballista.rs
@@ -980,8 +980,6 @@ pub mod scheduler_grpc_client {
                 );
             self.inner.unary(req, path, codec).await
         }
-        /// Push-based task scheduler will only leverage this interface
-        /// rather than the PollWork interface to report executor states
         pub async fn heart_beat_from_executor(
             &mut self,
             request: impl tonic::IntoRequest<super::HeartBeatParams>,
@@ -1253,8 +1251,6 @@ pub mod scheduler_grpc_server {
             tonic::Response<super::RegisterExecutorResult>,
             tonic::Status,
         >;
-        /// Push-based task scheduler will only leverage this interface
-        /// rather than the PollWork interface to report executor states
         async fn heart_beat_from_executor(
             &self,
             request: tonic::Request<super::HeartBeatParams>,

diff --git a/ballista/scheduler/src/bin/main.rs b/ballista/scheduler/src/bin/main.rs
@@ -64,7 +64,7 @@ async fn main() -> Result<()> {
 
     let config = SchedulerConfig {
         namespace: "ballista".to_string(),
-        external_host: "localhost".to_string(),
+        external_host: env::var("EXTERNAL_HOST").unwrap_or("localhost".to_string()),
         bind_port,
         event_loop_buffer_size: 10000,
         task_distribution: TaskDistributionPolicy::Bias,