diff --git a/src/runtime2/native.rs b/src/runtime2/native.rs index cf25163fb61249b33639cc6d0d24bb2acba8e214..f91ec36c5879a3eab121b373a0bfa3acd8a0dd7e 100644 --- a/src/runtime2/native.rs +++ b/src/runtime2/native.rs @@ -5,6 +5,7 @@ use std::sync::atomic::Ordering; use crate::protocol::ComponentCreationError; use crate::protocol::eval::ValueGroup; use crate::ProtocolDescription; +use crate::runtime2::scheduler::ComponentCtxFancy; use super::{ConnectorKey, ConnectorId, RuntimeInner, ConnectorCtx}; use super::scheduler::SchedulerCtx; @@ -21,7 +22,7 @@ pub(crate) trait Connector { fn handle_message(&mut self, message: Message, ctx: &ConnectorCtx, delta_state: &mut RunDeltaState); /// Should run the connector's behaviour up until the next blocking point. - fn run(&mut self, sched_ctx: SchedulerCtx, conn_ctx: &ConnectorCtx, delta_state: &mut RunDeltaState) -> ConnectorScheduling; + fn run(&mut self, sched_ctx: SchedulerCtx, comp_ctx: &mut ComponentCtxFancy, conn_ctx: &ConnectorCtx, delta_state: &mut RunDeltaState) -> ConnectorScheduling; } type SyncDone = Arc<(Mutex, Condvar)>; @@ -66,7 +67,7 @@ impl Connector for ConnectorApplication { } } - fn run(&mut self, _sched_ctx: SchedulerCtx, _conn_ctx: &ConnectorCtx, delta_state: &mut RunDeltaState) -> ConnectorScheduling { + fn run(&mut self, _sched_ctx: SchedulerCtx, comp_ctx: &mut ComponentCtxFancy, _conn_ctx: &ConnectorCtx, delta_state: &mut RunDeltaState) -> ConnectorScheduling { let mut queue = self.job_queue.lock().unwrap(); while let Some(job) = queue.pop_front() { match job {