mf_core/helpers/
runtime_common.rs

1//! 运行时公共辅助函数
2//!
3//! 此模块提供了多个运行时实现之间共享的公共功能，包括：
4//! - 扩展管理器的创建和合并逻辑
5//! - 中间件执行辅助函数
6//!
7//! 这些函数被以下运行时使用：
8//! - ForgeRuntime (runtime.rs)
9//! - ForgeActorRuntime (actor_runtime.rs)
10//! - ForgeAsyncRuntime (async_runtime.rs)
11
12use crate::{
13    config::ForgeConfig, debug::debug, extension_manager::ExtensionManager,
14    types::RuntimeOptions, ForgeResult,
15};
16
17/// 扩展管理器辅助函数
18///
19/// 提供扩展管理器的创建和合并功能，确保所有运行时使用相同的逻辑
20pub struct ExtensionManagerHelper;
21
22impl ExtensionManagerHelper {
23    /// 创建扩展管理器 - 自动处理XML schema配置并合并代码扩展
24    ///
25    /// 优先级顺序：
26    /// 1. 使用 `config.extension.xml_schema_paths` 中配置的路径
27    /// 2. 如果没有配置，尝试加载默认的 `schema/main.xml`
28    /// 3. 如果都没有，使用代码定义的扩展
29    ///
30    /// # 参数
31    /// * `runtime_options` - 运行时选项（包含代码定义的扩展）
32    /// * `forge_config` - Forge配置（包含XML路径配置）
33    ///
34    /// # 返回值
35    /// * `ForgeResult<ExtensionManager>` - 扩展管理器实例或错误
36    pub fn create_extension_manager(
37        runtime_options: &RuntimeOptions,
38        forge_config: &ForgeConfig,
39    ) -> ForgeResult<ExtensionManager> {
40        // 检查是否有配置的XML schema路径
41        if !forge_config.extension.xml_schema_paths.is_empty() {
42            debug!(
43                "使用配置的XML schema路径: {:?}",
44                forge_config.extension.xml_schema_paths
45            );
46
47            // 转换为字符串引用
48            let paths: Vec<&str> = forge_config
49                .extension
50                .xml_schema_paths
51                .iter()
52                .map(|s| s.as_str())
53                .collect();
54            let extension_manager = ExtensionManager::from_xml_files(&paths)?;
55
56            // 合并现有的扩展
57            let merged_extensions = Self::merge_extensions_with_xml(
58                runtime_options,
59                extension_manager,
60            )?;
61            return Ok(merged_extensions);
62        }
63
64        // 检查默认的 schema/main.xml 文件
65        let default_schema_path = "schema/main.xml";
66        if std::path::Path::new(default_schema_path).exists() {
67            debug!("使用默认的 schema 文件: {}", default_schema_path);
68            let extension_manager =
69                ExtensionManager::from_xml_file(default_schema_path)?;
70            let merged_extensions = Self::merge_extensions_with_xml(
71                runtime_options,
72                extension_manager,
73            )?;
74            return Ok(merged_extensions);
75        }
76
77        // 没有找到任何XML schema，使用默认配置
78        debug!("未找到XML schema配置，使用默认扩展");
79        ExtensionManager::new(&runtime_options.get_extensions())
80    }
81
82    /// 合并XML扩展和代码扩展
83    ///
84    /// 合并策略：
85    /// 1. XML扩展优先（节点和标记定义）
86    /// 2. 代码扩展补充（Extension类型，包含插件）
87    /// 3. 避免重复添加相同名称的节点/标记
88    ///
89    /// # 参数
90    /// * `runtime_options` - 运行时选项（包含代码定义的扩展）
91    /// * `xml_extension_manager` - 从XML加载的扩展管理器
92    ///
93    /// # 返回值
94    /// * `ForgeResult<ExtensionManager>` - 合并后的扩展管理器
95    pub fn merge_extensions_with_xml(
96        runtime_options: &RuntimeOptions,
97        xml_extension_manager: ExtensionManager,
98    ) -> ForgeResult<ExtensionManager> {
99        let schema = xml_extension_manager.get_schema();
100        let mut all_extensions = Vec::new();
101        let factory = schema.factory();
102        let (nodes, marks) = factory.definitions();
103        // 先添加XML扩展（优先级更高）
104        for (name, node_type) in nodes {
105            let node = crate::node::Node::create(name, node_type.spec.clone());
106            all_extensions.push(crate::types::Extensions::N(node));
107        }
108
109        for (name, mark_type) in marks {
110            let mark = crate::mark::Mark::new(name, mark_type.spec.clone());
111            all_extensions.push(crate::types::Extensions::M(mark));
112        }
113
114        // 再添加代码扩展（避免重复）
115        for ext in runtime_options.get_extensions() {
116            let name = match &ext {
117                crate::types::Extensions::N(node) => &node.name,
118                crate::types::Extensions::M(mark) => &mark.name,
119                crate::types::Extensions::E(_) => {
120                    // 直接添加Extension扩展（包含插件），不需要检查重复
121                    all_extensions.push(ext);
122                    continue;
123                },
124            };
125
126            // 检查是否已经存在
127            let exists = match &ext {
128                crate::types::Extensions::N(_) => nodes.contains_key(name),
129                crate::types::Extensions::M(_) => marks.contains_key(name),
130                crate::types::Extensions::E(_) => false, // Extension扩展总是添加
131            };
132
133            if !exists {
134                all_extensions.push(ext);
135            }
136        }
137
138        ExtensionManager::new(&all_extensions)
139    }
140}
141
142#[cfg(test)]
143mod tests {
144    use super::*;
145
146    #[test]
147    fn test_extension_manager_helper_creation() {
148        // 基本创建测试
149        let options = RuntimeOptions::default();
150        let config = ForgeConfig::default();
151
152        // 没有XML文件时应该使用代码扩展
153        let result =
154            ExtensionManagerHelper::create_extension_manager(&options, &config);
155        assert!(result.is_ok());
156    }
157}
mf_core/helpers/runtime_common.rs

mf_core/helpers/
runtime_common.rs
