Flutter Rust Bridge 中 StreamSink 的单次调用问题解析

在使用 Flutter Rust Bridge 进行跨平台开发时，开发者可能会遇到 StreamSink 只能调用一次的问题。本文将深入分析这一现象的原因，并提供解决方案。

问题现象

当开发者尝试在 Flutter 应用中多次调用 Rust 端的 StreamSink 方法时，发现只有第一次调用能够正常工作，后续调用虽然 Rust 代码执行成功，但 Dart 端的流监听器却无法收到数据更新。

问题根源

问题的核心在于 Rust 端使用了 OnceCell 来存储 StreamSink 实例。OnceCell 的特性决定了它只能被初始化一次，后续的 get_or_init 调用会直接返回已存储的值，而不会重新初始化。

在示例代码中，words_oncecell 函数使用了静态的 MEM OnceCell 来存储 StreamSink：

static MEM: OnceCell<StreamSink<Progress>> = OnceCell::new();

pub fn words_oncecell(sink: StreamSink<Progress>) {
    MEM.get_or_init(|| sink);
    inner_other(delegate);
}

这种实现方式导致了以下问题：

1. 第一次调用时，StreamSink 被正确存储
2. 后续调用时，get_or_init 直接返回已存储的 StreamSink
3. 由于 Stream 是单订阅的，已经完成的 Stream 无法再次使用

解决方案

方案一：避免使用 OnceCell

最简单的解决方案是避免使用 OnceCell，直接在每次调用时使用传入的 StreamSink：

pub fn words(sink: StreamSink<Progress>) {
    inner(|word, current, max| sink.add(Progress { word, current, max }).unwrap())
}

这种方式确保了每次调用都会使用新的 StreamSink 实例，避免了单次初始化带来的问题。

方案二：使用结构体封装状态

更符合 Rust 惯用法的解决方案是使用结构体来封装状态，而不是使用全局变量：

pub struct ProgressReporter {
    sink: Option<StreamSink<Progress>>,
}

impl ProgressReporter {
    pub fn new() -> Self {
        Self { sink: None }
    }
    
    pub fn set_sink(&mut self, sink: StreamSink<Progress>) {
        self.sink = Some(sink);
    }
    
    pub fn report(&self, progress: Progress) {
        if let Some(sink) = &self.sink {
            sink.add(progress).unwrap();
        }
    }
}

这种方式更加灵活，也更容易管理状态的生命周期。

最佳实践建议

1. 避免全局状态：在 Rust 中，全局状态往往会导致各种难以调试的问题，应该尽量避免。

2. 理解 Stream 特性：Dart 中的 Stream 有单订阅和多订阅之分，需要根据场景选择合适的类型。

3. 生命周期管理：确保 StreamSink 的生命周期与使用场景匹配，避免过早释放或长期持有。

4. 错误处理：正确处理 StreamSink.add 可能返回的错误，避免程序因未处理错误而崩溃。

总结

Flutter Rust Bridge 提供了强大的跨语言通信能力，但在使用时需要注意 Rust 和 Dart 两边的特性差异。特别是对于 Stream 这种异步通信机制，需要特别注意其生命周期和状态管理。通过避免全局状态、合理设计接口，可以构建出更加健壮的跨平台应用。

对于刚接触 Rust 的 Flutter 开发者来说，理解 Rust 的所有权系统和生命周期概念尤为重要。这些概念虽然初期学习曲线较陡，但一旦掌握，就能写出更加安全、高效的代码。