[arrow-ballista] branch main updated: Mark `SchedulerState` as pub (#688)

[dheres]

This is an automated email from the ASF dual-hosted git repository.

dheres pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/arrow-ballista.git


The following commit(s) were added to refs/heads/main by this push:
     new 35019e64 Mark `SchedulerState` as pub (#688)
35019e64 is described below

commit 35019e6416ee256e7966bec4c414424dc158a139
Author: Daniël Heres <danielhe...@gmail.com>
AuthorDate: Mon Feb 27 13:28:51 2023 +0100

    Mark `SchedulerState` as pub (#688)
    
    * Mark as pub
    
    * Fmt
    
    ---------
    
    Co-authored-by: Daniël Heres <daniel.he...@coralogix.com>
---
 ballista/scheduler/src/scheduler_server/mod.rs   | 2 +-
 ballista/scheduler/src/state/executor_manager.rs | 2 +-
 ballista/scheduler/src/state/mod.rs              | 3 +--
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/ballista/scheduler/src/scheduler_server/mod.rs 
b/ballista/scheduler/src/scheduler_server/mod.rs
index fc40b017..d31d5031 100644
--- a/ballista/scheduler/src/scheduler_server/mod.rs
+++ b/ballista/scheduler/src/scheduler_server/mod.rs
@@ -63,7 +63,7 @@ pub(crate) type SessionBuilder = fn(SessionConfig) -> 
SessionState;
 pub struct SchedulerServer<T: 'static + AsLogicalPlan, U: 'static + 
AsExecutionPlan> {
     pub scheduler_name: String,
     pub start_time: u128,
-    pub(crate) state: Arc<SchedulerState<T, U>>,
+    pub state: Arc<SchedulerState<T, U>>,
     pub(crate) query_stage_event_loop: EventLoop<QueryStageSchedulerEvent>,
     query_stage_scheduler: Arc<QueryStageScheduler<T, U>>,
     executor_termination_grace_period: u64,
diff --git a/ballista/scheduler/src/state/executor_manager.rs 
b/ballista/scheduler/src/state/executor_manager.rs
index bec97c3b..d3a9468e 100644
--- a/ballista/scheduler/src/state/executor_manager.rs
+++ b/ballista/scheduler/src/state/executor_manager.rs
@@ -90,7 +90,7 @@ pub const DEFAULT_EXECUTOR_TIMEOUT_SECONDS: u64 = 180;
 pub const EXPIRE_DEAD_EXECUTOR_INTERVAL_SECS: u64 = 15;
 
 #[derive(Clone)]
-pub(crate) struct ExecutorManager {
+pub struct ExecutorManager {
     // executor slot policy
     slots_policy: SlotsPolicy,
     task_distribution: TaskDistribution,
diff --git a/ballista/scheduler/src/state/mod.rs 
b/ballista/scheduler/src/state/mod.rs
index 17b1a4fb..0ecba0d7 100644
--- a/ballista/scheduler/src/state/mod.rs
+++ b/ballista/scheduler/src/state/mod.rs
@@ -85,8 +85,7 @@ pub fn encode_protobuf<T: Message + Default>(msg: &T) -> 
Result<Vec<u8>> {
 }
 
 #[derive(Clone)]
-pub(super) struct SchedulerState<T: 'static + AsLogicalPlan, U: 'static + 
AsExecutionPlan>
-{
+pub struct SchedulerState<T: 'static + AsLogicalPlan, U: 'static + 
AsExecutionPlan> {
     pub executor_manager: ExecutorManager,
     pub task_manager: TaskManager<T, U>,
     pub session_manager: SessionManager,