diff options
Diffstat (limited to 'Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/ToDictionary.cs')
-rw-r--r-- | Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/ToDictionary.cs | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/ToDictionary.cs b/Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/ToDictionary.cs new file mode 100644 index 0000000..919c1a1 --- /dev/null +++ b/Rx.NET/System.Reactive.Linq/Reactive/Linq/Observαble/ToDictionary.cs @@ -0,0 +1,71 @@ +// 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 ToDictionary<TSource, TKey, TElement> : Producer<IDictionary<TKey, TElement>> + { + private readonly IObservable<TSource> _source; + private readonly Func<TSource, TKey> _keySelector; + private readonly Func<TSource, TElement> _elementSelector; + private readonly IEqualityComparer<TKey> _comparer; + + public ToDictionary(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, IEqualityComparer<TKey> comparer) + { + _source = source; + _keySelector = keySelector; + _elementSelector = elementSelector; + _comparer = comparer; + } + + protected override IDisposable Run(IObserver<IDictionary<TKey, TElement>> observer, IDisposable cancel, Action<IDisposable> setSink) + { + var sink = new _(this, observer, cancel); + setSink(sink); + return _source.SubscribeSafe(sink); + } + + class _ : Sink<IDictionary<TKey, TElement>>, IObserver<TSource> + { + private readonly ToDictionary<TSource, TKey, TElement> _parent; + private Dictionary<TKey, TElement> _dictionary; + + public _(ToDictionary<TSource, TKey, TElement> parent, IObserver<IDictionary<TKey, TElement>> observer, IDisposable cancel) + : base(observer, cancel) + { + _parent = parent; + _dictionary = new Dictionary<TKey, TElement>(_parent._comparer); + } + + public void OnNext(TSource value) + { + try + { + _dictionary.Add(_parent._keySelector(value), _parent._elementSelector(value)); + } + catch (Exception ex) + { + base._observer.OnError(ex); + base.Dispose(); + } + } + + public void OnError(Exception error) + { + base._observer.OnError(error); + base.Dispose(); + } + + public void OnCompleted() + { + base._observer.OnNext(_dictionary); + base._observer.OnCompleted(); + base.Dispose(); + } + } + } +} +#endif
\ No newline at end of file |