actr_runtime/
context_factory.rs

1//! Context factory
2//!
3//! 负责创建 RuntimeContext 实例，注入 OutGate 和其他依赖。
4
5use crate::context::RuntimeContext;
6use crate::inbound::{DataStreamRegistry, MediaFrameRegistry};
7use crate::outbound::OutGate;
8use crate::transport::InprocTransportManager;
9use crate::wire::webrtc::SignalingClient;
10use actr_protocol::{AIdCredential, ActrId};
11use std::sync::Arc;
12
13/// Context factory
14///
15/// # 职责
16///
17/// - 创建 RuntimeContext 实例
18/// - 注入 OutGate（enum dispatch，零虚函数）
19/// - 管理默认配置
20#[derive(Clone)]
21pub struct ContextFactory {
22    /// 进程内通信 gate（本地调用）- 立即可用
23    pub(crate) inproc_gate: OutGate,
24
25    /// 跨进程通信 gate（远程调用）- 延迟初始化
26    pub(crate) outproc_gate: Option<OutGate>,
27
28    /// Shell → Workload 方向的传输管理器
29    pub(crate) shell_to_workload: Arc<InprocTransportManager>,
30
31    /// Workload → Shell 方向的传输管理器
32    pub(crate) workload_to_shell: Arc<InprocTransportManager>,
33
34    /// DataStream 回调注册表
35    pub(crate) data_stream_registry: Arc<DataStreamRegistry>,
36
37    /// MediaTrack 回调注册表
38    pub(crate) media_frame_registry: Arc<MediaFrameRegistry>,
39
40    /// Signaling client for discovery
41    pub(crate) signaling_client: Arc<dyn SignalingClient>,
42}
43
44impl ContextFactory {
45    /// 创建新的 ContextFactory
46    ///
47    /// # 参数
48    ///
49    /// - `inproc_gate`: 进程内通信 gate（Dest::Shell/Local）- 立即可用
50    /// - `shell_to_workload`: Shell → Workload 方向的传输管理器
51    /// - `workload_to_shell`: Workload → Shell 方向的传输管理器
52    /// - `data_stream_registry`: DataStream 回调注册表
53    /// - `media_frame_registry`: MediaTrack 回调注册表
54    ///
55    /// # 设计说明
56    ///
57    /// - **inproc_gate**: 在 ActrSystem::new() 时创建，立即可用（Shell/Local 通信不需要 ActorId）
58    /// - **outproc_gate**: 初始为 None，在 ActrNode::start() WebRTC 初始化完成后设置
59    /// - **双向 InprocTransportManager**: 确保 Shell 和 Workload 的 pending_requests 完全分离
60    /// - **data_stream_registry**: 管理 DataStream 回调，支持应用数据流传输
61    /// - **media_frame_registry**: 管理 MediaTrack 回调，支持 WebRTC 原生媒体流
62    pub fn new(
63        inproc_gate: OutGate,
64        shell_to_workload: Arc<InprocTransportManager>,
65        workload_to_shell: Arc<InprocTransportManager>,
66        data_stream_registry: Arc<DataStreamRegistry>,
67        media_frame_registry: Arc<MediaFrameRegistry>,
68        signaling_client: Arc<dyn SignalingClient>,
69    ) -> Self {
70        Self {
71            inproc_gate,
72            outproc_gate: None, // 延迟初始化，等待 WebRTC 就绪
73            shell_to_workload,
74            workload_to_shell,
75            data_stream_registry,
76            media_frame_registry,
77            signaling_client,
78        }
79    }
80
81    /// 设置跨进程通信 gate
82    ///
83    /// # 用途
84    ///
85    /// ActrNode::start() 完成 WebRTC 初始化后调用
86    pub fn set_outproc_gate(&mut self, gate: OutGate) {
87        tracing::debug!("🔄 Setting outproc OutGate in ContextFactory");
88        self.outproc_gate = Some(gate);
89    }
90
91    /// 获取 Shell → Workload 方向的传输管理器
92    pub fn shell_to_workload(&self) -> Arc<InprocTransportManager> {
93        self.shell_to_workload.clone()
94    }
95
96    /// 获取 Workload → Shell 方向的传输管理器
97    pub fn workload_to_shell(&self) -> Arc<InprocTransportManager> {
98        self.workload_to_shell.clone()
99    }
100
101    /// 创建 Context（用于消息处理）
102    ///
103    /// # 参数
104    ///
105    /// - `self_id`: 当前 Actor ID
106    /// - `caller_id`: 调用方 Actor ID（可选）
107    /// - `trace_id`: 分布式追踪 ID
108    /// - `request_id`: 请求唯一 ID
109    ///
110    /// # 返回
111    ///
112    /// 返回 RuntimeContext 实例（实现了 Context trait）
113    pub fn create(
114        &self,
115        self_id: &ActrId,
116        caller_id: Option<&ActrId>,
117        trace_id: &str,
118        request_id: &str,
119        credential: &AIdCredential,
120    ) -> RuntimeContext {
121        RuntimeContext::new(
122            self_id.clone(),
123            caller_id.cloned(),
124            trace_id.to_string(),
125            request_id.to_string(),
126            self.inproc_gate.clone(), // Clone OutGate enum（Arc 内部，开销低）
127            self.outproc_gate.clone(), // Clone Option<OutGate>
128            self.data_stream_registry.clone(), // Clone Arc<DataStreamRegistry>
129            self.media_frame_registry.clone(), // Clone Arc<MediaFrameRegistry>
130            self.signaling_client.clone(),
131            credential.clone(),
132        )
133    }
134
135    /// 创建引导 Context（用于生命周期钩子）
136    ///
137    /// # 用途
138    ///
139    /// 用于 on_start/on_stop 钩子，无 caller_id
140    pub fn create_bootstrap(&self, self_id: &ActrId, credential: &AIdCredential) -> RuntimeContext {
141        RuntimeContext::new(
142            self_id.clone(),
143            None,
144            uuid::Uuid::new_v4().to_string(),
145            uuid::Uuid::new_v4().to_string(),
146            self.inproc_gate.clone(),
147            self.outproc_gate.clone(),
148            self.data_stream_registry.clone(),
149            self.media_frame_registry.clone(),
150            self.signaling_client.clone(),
151            credential.clone(),
152        )
153    }
154}
actr_runtime/context_factory.rs

actr_runtime/
context_factory.rs
