This is an automated email from the ASF dual-hosted git repository. dheres pushed a commit to branch expose_state in repository https://gitbox.apache.org/repos/asf/arrow-ballista.git
commit a95a0eef2ba27f27f3a5232d957a4a2d37a1e7cb Author: Daniël Heres <[email protected]> AuthorDate: Mon Feb 27 10:59:36 2023 +0100 Mark as pub --- ballista/scheduler/src/scheduler_server/mod.rs | 2 +- ballista/scheduler/src/state/executor_manager.rs | 2 +- ballista/scheduler/src/state/mod.rs | 2 +- 3 files changed, 3 insertions(+), 3 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..4105877d 100644 --- a/ballista/scheduler/src/state/mod.rs +++ b/ballista/scheduler/src/state/mod.rs @@ -85,7 +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>,
