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 72 73 74 75 76
|
// 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.Reactive.Disposables;
namespace System.Reactive.Linq.ObservableImpl
{
class Using<TSource, TResource> : Producer<TSource>
where TResource : IDisposable
{
private readonly Func<TResource> _resourceFactory;
private readonly Func<TResource, IObservable<TSource>> _observableFactory;
public Using(Func<TResource> resourceFactory, Func<TResource, IObservable<TSource>> observableFactory)
{
_resourceFactory = resourceFactory;
_observableFactory = observableFactory;
}
protected override IDisposable Run(IObserver<TSource> observer, IDisposable cancel, Action<IDisposable> setSink)
{
var sink = new _(this, observer, cancel);
setSink(sink);
return sink.Run();
}
class _ : Sink<TSource>, IObserver<TSource>
{
private readonly Using<TSource, TResource> _parent;
public _(Using<TSource, TResource> parent, IObserver<TSource> observer, IDisposable cancel)
: base(observer, cancel)
{
_parent = parent;
}
public IDisposable Run()
{
var source = default(IObservable<TSource>);
var disposable = Disposable.Empty;
try
{
var resource = _parent._resourceFactory();
if (resource != null)
disposable = resource;
source = _parent._observableFactory(resource);
}
catch (Exception exception)
{
return new CompositeDisposable(Observable.Throw<TSource>(exception).SubscribeSafe(this), disposable);
}
return new CompositeDisposable(source.SubscribeSafe(this), disposable);
}
public void OnNext(TSource value)
{
base._observer.OnNext(value);
}
public void OnError(Exception error)
{
base._observer.OnError(error);
base.Dispose();
}
public void OnCompleted()
{
base._observer.OnCompleted();
base.Dispose();
}
}
}
}
#endif
|