Mark SchedulerState as pub (apache#688) (#38)

* Mark as pub

* Fmt

---------

Co-authored-by: Daniël Heres <[email protected]>

Author: Dandandan

ballista/scheduler/src/scheduler_server/mod.rs

@@ -64,7 +64,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,

ballista/scheduler/src/state/executor_manager.rs

@@ -89,7 +89,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,

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,