From cde9fc6a8fe569203cb991121a35c2a9c7f4c420 Mon Sep 17 00:00:00 2001 From: Atsushi Eno Date: Tue, 22 Jan 2013 17:25:22 +0900 Subject: import 2b5dbddd740b, new directory structure in the original rx. --- .../Reactive/Linq/Observable/Take.cs | 176 +++++++++++++++++++++ 1 file changed, 176 insertions(+) create mode 100644 Rx/NET/Source/System.Reactive.Linq/Reactive/Linq/Observable/Take.cs (limited to 'Rx/NET/Source/System.Reactive.Linq/Reactive/Linq/Observable/Take.cs') diff --git a/Rx/NET/Source/System.Reactive.Linq/Reactive/Linq/Observable/Take.cs b/Rx/NET/Source/System.Reactive.Linq/Reactive/Linq/Observable/Take.cs new file mode 100644 index 0000000..738e7c0 --- /dev/null +++ b/Rx/NET/Source/System.Reactive.Linq/Reactive/Linq/Observable/Take.cs @@ -0,0 +1,176 @@ +// 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.Concurrency; +using System.Reactive.Disposables; + +namespace System.Reactive.Linq.Observαble +{ + class Take : Producer + { + private readonly IObservable _source; + private readonly int _count; + private readonly TimeSpan _duration; + internal readonly IScheduler _scheduler; + + public Take(IObservable source, int count) + { + _source = source; + _count = count; + } + + public Take(IObservable source, TimeSpan duration, IScheduler scheduler) + { + _source = source; + _duration = duration; + _scheduler = scheduler; + } + + public IObservable Ω(int count) + { + // + // Minimum semantics: + // + // xs --o--o--o--o--o--o--| xs --o--o--o--o--o--o--| + // xs.Take(5) --o--o--o--o--o| xs.Take(3) --o--o--o| + // xs.Take(5).Take(3) --o--o--o| xs.Take(3).Take(5) --o--o--o| + // + if (_count <= count) + return this; + else + return new Take(_source, count); + } + + public IObservable Ω(TimeSpan duration) + { + // + // Minimum semantics: + // + // t 0--1--2--3--4--5--6--7-> t 0--1--2--3--4--5--6--7-> + // + // xs --o--o--o--o--o--o--| xs --o--o--o--o--o--o--| + // xs.Take(5s) --o--o--o--o--o| xs.Take(3s) --o--o--o| + // xs.Take(5s).Take(3s) --o--o--o| xs.Take(3s).Take(5s) --o--o--o| + // + if (_duration <= duration) + return this; + else + return new Take(_source, duration, _scheduler); + } + + protected override IDisposable Run(IObserver observer, IDisposable cancel, Action setSink) + { + if (_scheduler == null) + { + var sink = new _(this, observer, cancel); + setSink(sink); + return _source.SubscribeSafe(sink); + } + else + { + var sink = new τ(this, observer, cancel); + setSink(sink); + return sink.Run(); + } + } + + class _ : Sink, IObserver + { + private readonly Take _parent; + private int _remaining; + + public _(Take parent, IObserver observer, IDisposable cancel) + : base(observer, cancel) + { + _parent = parent; + _remaining = _parent._count; + } + + public void OnNext(TSource value) + { + if (_remaining > 0) + { + --_remaining; + base._observer.OnNext(value); + + if (_remaining == 0) + { + base._observer.OnCompleted(); + base.Dispose(); + } + } + } + + public void OnError(Exception error) + { + base._observer.OnError(error); + base.Dispose(); + } + + public void OnCompleted() + { + base._observer.OnCompleted(); + base.Dispose(); + } + } + + class τ : Sink, IObserver + { + private readonly Take _parent; + + public τ(Take parent, IObserver observer, IDisposable cancel) + : base(observer, cancel) + { + _parent = parent; + } + + private object _gate; + + public IDisposable Run() + { + _gate = new object(); + + var t = _parent._scheduler.Schedule(_parent._duration, Tick); + var d = _parent._source.SubscribeSafe(this); + return new CompositeDisposable(t, d); + } + + private void Tick() + { + lock (_gate) + { + base._observer.OnCompleted(); + base.Dispose(); + } + } + + public void OnNext(TSource value) + { + lock (_gate) + { + base._observer.OnNext(value); + } + } + + public void OnError(Exception error) + { + lock (_gate) + { + base._observer.OnError(error); + base.Dispose(); + } + } + + public void OnCompleted() + { + lock (_gate) + { + base._observer.OnCompleted(); + base.Dispose(); + } + } + } + } +} +#endif \ No newline at end of file -- cgit v1.2.3