diff options
Diffstat (limited to 'Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/Distinct.cs')
-rw-r--r-- | Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/Distinct.cs | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/Distinct.cs b/Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/Distinct.cs new file mode 100644 index 0000000..af20277 --- /dev/null +++ b/Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/Distinct.cs @@ -0,0 +1,75 @@ +// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information. + +#if !NO_PERF +using System; +using System.Collections.Generic; + +namespace System.Reactive.Linq.Observαble +{ + class Distinct<TSource, TKey> : Producer<TSource> + { + private readonly IObservable<TSource> _source; + private readonly Func<TSource, TKey> _keySelector; + private readonly IEqualityComparer<TKey> _comparer; + + public Distinct(IObservable<TSource> source, Func<TSource, TKey> keySelector, IEqualityComparer<TKey> comparer) + { + _source = source; + _keySelector = keySelector; + _comparer = comparer; + } + + protected override IDisposable Run(IObserver<TSource> observer, IDisposable cancel, Action<IDisposable> setSink) + { + var sink = new _(this, observer, cancel); + setSink(sink); + return _source.SubscribeSafe(sink); + } + + class _ : Sink<TSource>, IObserver<TSource> + { + private readonly Distinct<TSource, TKey> _parent; + private HashSet<TKey> _hashSet; + + public _(Distinct<TSource, TKey> parent, IObserver<TSource> observer, IDisposable cancel) + : base(observer, cancel) + { + _parent = parent; + _hashSet = new HashSet<TKey>(_parent._comparer); + } + + public void OnNext(TSource value) + { + var key = default(TKey); + var hasAdded = false; + try + { + key = _parent._keySelector(value); + hasAdded = _hashSet.Add(key); + } + catch (Exception exception) + { + base._observer.OnError(exception); + base.Dispose(); + return; + } + + if (hasAdded) + base._observer.OnNext(value); + } + + public void OnError(Exception error) + { + base._observer.OnError(error); + base.Dispose(); + } + + public void OnCompleted() + { + base._observer.OnCompleted(); + base.Dispose(); + } + } + } +} +#endif
\ No newline at end of file |