RxSwift 中的滑动窗口

debugcn 发表于 Dev

但

来自 RxJava 背景，我无法提出在 RxSwift 中实现滑动窗口的标准方法。例如，我有以下事件序列：

1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, ...

让我们想象一下事件发射在一秒钟内发生两次。我想要做的是将这个序列转换成一个缓冲区序列，每个缓冲区包含最后三秒的数据。另外，每个缓冲区将每秒发出一次。所以结果看起来像这样：

[1,2,3,4,5,6], [3,4,5,6,7,8], [5,6,7,8,9,10], ...

我将在 RxJava 中做的是我将使用该buffer方法的重载之一，如下所示：

stream.buffer(3000, 1000, TimeUnit.MILLISECONDS)

这正是我需要完成的结果：缓冲区序列，每个缓冲区每秒发出一次并包含最后三秒的数据。

我广泛地检查了 RxSwift 文档，但没有发现任何buffer允许我这样做的运算符重载。我是否错过了一些不明显的（对于 RxJava 用户，ofc）操作符？

丹尼尔·T。

我最初使用自定义运算符编写了解决方案。从那以后，我想出了如何使用标准运算符来完成它。

extension ObservableType {

    func buffer(timeSpan: RxTimeInterval, timeShift: RxTimeInterval, scheduler: SchedulerType) -> Observable<[E]> {
        let trigger = Observable<Int>.timer(timeSpan, period: timeShift, scheduler: scheduler)
            .takeUntil(self.takeLast(1))

        let buffer = self
            .scan([Date: E]()) { previous, current in
                var next = previous
                let now = scheduler.now
                next[now] = current
                return next.filter { $0.key > now.addingTimeInterval(-timeSpan) }
        }

        return trigger.withLatestFrom(buffer)
            .map { $0.sorted(by: { $0.key <= $1.key }).map { $0.value } }
    }
}

我将在下面留下我的原始解决方案以供后代使用：

编写您自己的运算符是这里的解决方案。

extension ObservableType {

    func buffer(timeSpan: RxTimeInterval, timeShift: RxTimeInterval, scheduler: SchedulerType) -> Observable<[E]> {
        return Observable.create { observer in
            var buf: [Date: E] = [:]
            let lock = NSRecursiveLock()
            let elementDispoable = self.subscribe { event in
                lock.lock(); defer { lock.unlock() }
                switch event {
                case let .next(element):
                    buf[Date()] = element
                case .completed:
                    observer.onCompleted()
                case let .error(error):
                    observer.onError(error)
                }
            }
            let spanDisposable = scheduler.schedulePeriodic((), startAfter: timeSpan, period: timeShift, action: { state in
                lock.lock(); defer { lock.unlock() }
                let now = Date()
                buf = buf.filter { $0.key > now.addingTimeInterval(-timeSpan) }
                observer.onNext(buf.sorted(by: { $0.key <= $1.key }).map { $0.value })
            })
            return Disposables.create([spanDisposable, elementDispoable])
        }
    }
}

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。