1 回答
TA贡献1826条经验 获得超6个赞
这很有趣。必须稍微调整一下算法。还可以进一步改进。
假设:
有两个通用类型 的
streamA
流。streamB
T
两个流分别排序,使得
streamA[i] < streamA[i+1]
和streamB[i] < stream[i+1]
。您不能假设
streamA[i]
和之间有任何关系streamB[i]
。流 A 和 B 是谨慎的:相同的元素不会从两者中发出。如果发生这种情况,我会扔掉
NotImplementedException
。这个案子很容易处理,但我想避免歧义。有一个
min
类型的函数T
。没有对两条流的相对速度做出任何假设,但如果其中一条始终比另一条快,则背压将成为问题。
这是我使用的算法:
设两个队列,
qA
并且qB
。当您从 获取一个项目时
streamA
,将其排队到qA
。当您从 获取一个项目时
streamB
,将其排队到qB
。当和
qA
中都有一个项目时qB
,比较两个队列的顶部项目。删除并发出这两者的最小值。如果两个队列仍然非空,则重复。如果 或
streamA
完成streamB
,则转储队列的内容并终止。注意:这无疑是懒惰的,可能应该更改为转储,然后继续返回未完成的 observable。
这是代码:
public static IObservable<T> SortedMerge<T>(this IObservable<T> source, IObservable<T> other)
{
return SortedMerge(source, other, (a, b) => Enumerable.Min(new[] { a, b}));
}
public static IObservable<T> SortedMerge<T>(this IObservable<T> source, IObservable<T> other, Func<T, T, T> min)
{
return source
.Select(i => (key: 1, value: i)).Materialize()
.Merge(other.Select(i => (key: 2, value: i)).Materialize())
.Scan((qA: ImmutableQueue<T>.Empty, qB: ImmutableQueue<T>.Empty, exception: (Exception)null, outputMessages: new List<T>()),
(state, message) =>
{
if (message.Kind == NotificationKind.OnNext)
{
var key = message.Value.key;
var value = message.Value.value;
var qA = state.qA;
var qB = state.qB;
if (key == 1)
qA = qA.Enqueue(value);
else
qB = qB.Enqueue(value);
var output = new List<T>();
while(!qA.IsEmpty && !qB.IsEmpty)
{
var aVal = qA.Peek();
var bVal = qB.Peek();
var minVal = min(aVal, bVal);
if(aVal.Equals(minVal) && bVal.Equals(minVal))
throw new NotImplementedException();
if(aVal.Equals(minVal))
{
output.Add(aVal);
qA = qA.Dequeue();
}
else
{
output.Add(bVal);
qB = qB.Dequeue();
}
}
return (qA, qB, null, output);
}
else if (message.Kind == NotificationKind.OnError)
{
return (state.qA, state.qB, message.Exception, new List<T>());
}
else //message.Kind == NotificationKind.OnCompleted
{
var output = state.qA.Concat(state.qB).ToList();
return (ImmutableQueue<T>.Empty, ImmutableQueue<T>.Empty, null, output);
}
})
.Publish(tuples => Observable.Merge(
tuples
.Where(t => t.outputMessages.Any() && (!t.qA.IsEmpty || !t.qB.IsEmpty))
.SelectMany(t => t.outputMessages
.Select(v => Notification.CreateOnNext<T>(v))
.ToObservable()
),
tuples
.Where(t => t.outputMessages.Any() && t.qA.IsEmpty && t.qB.IsEmpty)
.SelectMany(t => t.outputMessages
.Select(v => Notification.CreateOnNext<T>(v))
.ToObservable()
.Concat(Observable.Return(Notification.CreateOnCompleted<T>()))
),
tuples
.Where(t => t.exception != null)
.Select(t => Notification.CreateOnError<T>(t.exception))
))
.Dematerialize();
ImmutableQueue来自System.Collections.Immutable. Scan需要跟踪状态。由于OnCompleted处理需要具体化。诚然,这是一个复杂的解决方案,但我不确定是否有更干净的以 Rx 为中心的方法。
如果您需要进一步说明,请告诉我。
- 1 回答
- 0 关注
- 107 浏览
添加回答
举报