1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 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.ObservableImpl
{
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
|