bamboo_engine/runtime/execution/
event_forwarder.rs

1//! Event forwarding from MPSC to broadcast channels.
2//!
3//! Creates an MPSC channel for agent loop events and spawns a background task
4//! that relays events to the session's broadcast sender while tracking runner
5//! diagnostic state (budget events, tool execution, round progress).
6
7use std::collections::HashMap;
8use std::sync::Arc;
9
10use chrono::Utc;
11use tokio::sync::{broadcast, mpsc, RwLock};
12
13use bamboo_agent_core::AgentEvent;
14
15use super::runner_state::AgentRunner;
16
17/// Create an MPSC channel for agent events and spawn a forwarding task
18/// that relays events to the broadcast sender while tracking runner
19/// diagnostic fields for live visibility.
20///
21/// Returns `(mpsc_tx, forwarder_handle)`.
22pub fn create_event_forwarder(
23    session_id: String,
24    broadcast_tx: broadcast::Sender<AgentEvent>,
25    runners: Arc<RwLock<HashMap<String, AgentRunner>>>,
26) -> (mpsc::Sender<AgentEvent>, tokio::task::JoinHandle<()>) {
27    let (mpsc_tx, mut mpsc_rx) = mpsc::channel::<AgentEvent>(100);
28
29    let forwarder = tokio::spawn(async move {
30        while let Some(event) = mpsc_rx.recv().await {
31            {
32                let mut runners = runners.write().await;
33                if let Some(runner) = runners.get_mut(&session_id) {
34                    runner.last_event_at = Some(Utc::now());
35
36                    match &event {
37                        AgentEvent::TokenBudgetUpdated { .. } => {
38                            runner.last_budget_event = Some(event.clone());
39                        }
40                        AgentEvent::ToolStart { tool_name, .. } => {
41                            runner.last_tool_name = Some(tool_name.clone());
42                            runner.last_tool_phase = Some("begin".to_string());
43                        }
44                        AgentEvent::ToolLifecycle {
45                            tool_name, phase, ..
46                        } => {
47                            runner.last_tool_name = Some(tool_name.clone());
48                            runner.last_tool_phase = Some(phase.clone());
49                        }
50                        AgentEvent::RunnerProgress { round_count, .. } => {
51                            runner.round_count = *round_count;
52                        }
53                        _ => {}
54                    }
55                }
56            }
57            let _ = broadcast_tx.send(event);
58        }
59    });
60
61    (mpsc_tx, forwarder)
62}
bamboo_engine/runtime/execution/event_forwarder.rs

bamboo_engine/runtime/execution/
event_forwarder.rs
