1
0

QueryLanguage.StandardSequenceOperators.cs 62 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558
  1. // Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.
  2. using System.Collections.Generic;
  3. using System.Reactive.Concurrency;
  4. #if !NO_TPL
  5. using System.Threading;
  6. using System.Threading.Tasks;
  7. #endif
  8. namespace System.Reactive.Linq
  9. {
  10. #if !NO_PERF
  11. using ObservableImpl;
  12. #endif
  13. internal partial class QueryLanguage
  14. {
  15. #region + Cast +
  16. public virtual IObservable<TResult> Cast<TResult>(IObservable<object> source)
  17. {
  18. #if !NO_PERF
  19. return new Cast<object, TResult>(source);
  20. #else
  21. return source.Select(x => (TResult)x);
  22. #endif
  23. }
  24. #endregion
  25. #region + DefaultIfEmpty +
  26. public virtual IObservable<TSource> DefaultIfEmpty<TSource>(IObservable<TSource> source)
  27. {
  28. #if !NO_PERF
  29. return new DefaultIfEmpty<TSource>(source, default(TSource));
  30. #else
  31. return DefaultIfEmpty_(source, default(TSource));
  32. #endif
  33. }
  34. public virtual IObservable<TSource> DefaultIfEmpty<TSource>(IObservable<TSource> source, TSource defaultValue)
  35. {
  36. #if !NO_PERF
  37. return new DefaultIfEmpty<TSource>(source, defaultValue);
  38. #else
  39. return DefaultIfEmpty_(source, defaultValue);
  40. #endif
  41. }
  42. #if NO_PERF
  43. private static IObservable<TSource> DefaultIfEmpty_<TSource>(IObservable<TSource> source, TSource defaultValue)
  44. {
  45. return new AnonymousObservable<TSource>(observer =>
  46. {
  47. var found = false;
  48. return source.Subscribe(
  49. x =>
  50. {
  51. found = true;
  52. observer.OnNext(x);
  53. },
  54. observer.OnError,
  55. () =>
  56. {
  57. if (!found)
  58. observer.OnNext(defaultValue);
  59. observer.OnCompleted();
  60. }
  61. );
  62. });
  63. }
  64. #endif
  65. #endregion
  66. #region + Distinct +
  67. public virtual IObservable<TSource> Distinct<TSource>(IObservable<TSource> source)
  68. {
  69. #if !NO_PERF
  70. return new Distinct<TSource, TSource>(source, x => x, EqualityComparer<TSource>.Default);
  71. #else
  72. return Distinct_(source, x => x, EqualityComparer<TSource>.Default);
  73. #endif
  74. }
  75. public virtual IObservable<TSource> Distinct<TSource>(IObservable<TSource> source, IEqualityComparer<TSource> comparer)
  76. {
  77. #if !NO_PERF
  78. return new Distinct<TSource, TSource>(source, x => x, comparer);
  79. #else
  80. return Distinct_(source, x => x, comparer);
  81. #endif
  82. }
  83. public virtual IObservable<TSource> Distinct<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector)
  84. {
  85. #if !NO_PERF
  86. return new Distinct<TSource, TKey>(source, keySelector, EqualityComparer<TKey>.Default);
  87. #else
  88. return Distinct_(source, keySelector, EqualityComparer<TKey>.Default);
  89. #endif
  90. }
  91. public virtual IObservable<TSource> Distinct<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector, IEqualityComparer<TKey> comparer)
  92. {
  93. #if !NO_PERF
  94. return new Distinct<TSource, TKey>(source, keySelector, comparer);
  95. #else
  96. return Distinct_(source, keySelector, comparer);
  97. #endif
  98. }
  99. #if NO_PERF
  100. private static IObservable<TSource> Distinct_<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector, IEqualityComparer<TKey> comparer)
  101. {
  102. return new AnonymousObservable<TSource>(observer =>
  103. {
  104. var hashSet = new HashSet<TKey>(comparer);
  105. return source.Subscribe(
  106. x =>
  107. {
  108. var key = default(TKey);
  109. var hasAdded = false;
  110. try
  111. {
  112. key = keySelector(x);
  113. hasAdded = hashSet.Add(key);
  114. }
  115. catch (Exception exception)
  116. {
  117. observer.OnError(exception);
  118. return;
  119. }
  120. if (hasAdded)
  121. observer.OnNext(x);
  122. },
  123. observer.OnError,
  124. observer.OnCompleted
  125. );
  126. });
  127. }
  128. #endif
  129. #endregion
  130. #region + GroupBy +
  131. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupBy<TSource, TKey, TElement>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector)
  132. {
  133. return GroupBy_<TSource, TKey, TElement>(source, keySelector, elementSelector, null, EqualityComparer<TKey>.Default);
  134. }
  135. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupBy<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector, IEqualityComparer<TKey> comparer)
  136. {
  137. return GroupBy_<TSource, TKey, TSource>(source, keySelector, x => x, null, comparer);
  138. }
  139. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupBy<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector)
  140. {
  141. return GroupBy_<TSource, TKey, TSource>(source, keySelector, x => x, null, EqualityComparer<TKey>.Default);
  142. }
  143. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupBy<TSource, TKey, TElement>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, IEqualityComparer<TKey> comparer)
  144. {
  145. return GroupBy_<TSource, TKey, TElement>(source, keySelector, elementSelector, null, comparer);
  146. }
  147. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupBy<TSource, TKey, TElement>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, int capacity)
  148. {
  149. return GroupBy_<TSource, TKey, TElement>(source, keySelector, elementSelector, capacity, EqualityComparer<TKey>.Default);
  150. }
  151. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupBy<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector, int capacity, IEqualityComparer<TKey> comparer)
  152. {
  153. return GroupBy_<TSource, TKey, TSource>(source, keySelector, x => x, capacity, comparer);
  154. }
  155. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupBy<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector, int capacity)
  156. {
  157. return GroupBy_<TSource, TKey, TSource>(source, keySelector, x => x, capacity, EqualityComparer<TKey>.Default);
  158. }
  159. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupBy<TSource, TKey, TElement>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, int capacity, IEqualityComparer<TKey> comparer)
  160. {
  161. return GroupBy_<TSource, TKey, TElement>(source, keySelector, elementSelector, capacity, comparer);
  162. }
  163. private static IObservable<IGroupedObservable<TKey, TElement>> GroupBy_<TSource, TKey, TElement>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, int? capacity, IEqualityComparer<TKey> comparer)
  164. {
  165. #if !NO_PERF
  166. return new GroupBy<TSource, TKey, TElement>(source, keySelector, elementSelector, capacity, comparer);
  167. #else
  168. return GroupByUntil_<TSource, TKey, TElement, Unit>(source, keySelector, elementSelector, _ => Observable.Never<Unit>(), capacity, comparer);
  169. #endif
  170. }
  171. #endregion
  172. #region + GroupByUntil +
  173. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupByUntil<TSource, TKey, TElement, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, Func<IGroupedObservable<TKey, TElement>, IObservable<TDuration>> durationSelector, IEqualityComparer<TKey> comparer)
  174. {
  175. return GroupByUntil_<TSource, TKey, TElement, TDuration>(source, keySelector, elementSelector, durationSelector, null, comparer);
  176. }
  177. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupByUntil<TSource, TKey, TElement, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, Func<IGroupedObservable<TKey, TElement>, IObservable<TDuration>> durationSelector)
  178. {
  179. return GroupByUntil_<TSource, TKey, TElement, TDuration>(source, keySelector, elementSelector, durationSelector, null, EqualityComparer<TKey>.Default);
  180. }
  181. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupByUntil<TSource, TKey, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<IGroupedObservable<TKey, TSource>, IObservable<TDuration>> durationSelector, IEqualityComparer<TKey> comparer)
  182. {
  183. return GroupByUntil_<TSource, TKey, TSource, TDuration>(source, keySelector, x => x, durationSelector, null, comparer);
  184. }
  185. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupByUntil<TSource, TKey, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<IGroupedObservable<TKey, TSource>, IObservable<TDuration>> durationSelector)
  186. {
  187. return GroupByUntil_<TSource, TKey, TSource, TDuration>(source, keySelector, x => x, durationSelector, null, EqualityComparer<TKey>.Default);
  188. }
  189. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupByUntil<TSource, TKey, TElement, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, Func<IGroupedObservable<TKey, TElement>, IObservable<TDuration>> durationSelector, int capacity, IEqualityComparer<TKey> comparer)
  190. {
  191. return GroupByUntil_<TSource, TKey, TElement, TDuration>(source, keySelector, elementSelector, durationSelector, capacity, comparer);
  192. }
  193. public virtual IObservable<IGroupedObservable<TKey, TElement>> GroupByUntil<TSource, TKey, TElement, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, Func<IGroupedObservable<TKey, TElement>, IObservable<TDuration>> durationSelector, int capacity)
  194. {
  195. return GroupByUntil_<TSource, TKey, TElement, TDuration>(source, keySelector, elementSelector, durationSelector, capacity, EqualityComparer<TKey>.Default);
  196. }
  197. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupByUntil<TSource, TKey, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<IGroupedObservable<TKey, TSource>, IObservable<TDuration>> durationSelector, int capacity, IEqualityComparer<TKey> comparer)
  198. {
  199. return GroupByUntil_<TSource, TKey, TSource, TDuration>(source, keySelector, x => x, durationSelector, capacity, comparer);
  200. }
  201. public virtual IObservable<IGroupedObservable<TKey, TSource>> GroupByUntil<TSource, TKey, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<IGroupedObservable<TKey, TSource>, IObservable<TDuration>> durationSelector, int capacity)
  202. {
  203. return GroupByUntil_<TSource, TKey, TSource, TDuration>(source, keySelector, x => x, durationSelector, capacity, EqualityComparer<TKey>.Default);
  204. }
  205. private static IObservable<IGroupedObservable<TKey, TElement>> GroupByUntil_<TSource, TKey, TElement, TDuration>(IObservable<TSource> source, Func<TSource, TKey> keySelector, Func<TSource, TElement> elementSelector, Func<IGroupedObservable<TKey, TElement>, IObservable<TDuration>> durationSelector, int? capacity, IEqualityComparer<TKey> comparer)
  206. {
  207. #if !NO_PERF
  208. return new GroupByUntil<TSource, TKey, TElement, TDuration>(source, keySelector, elementSelector, durationSelector, capacity, comparer);
  209. #else
  210. return new AnonymousObservable<IGroupedObservable<TKey, TElement>>(observer =>
  211. {
  212. var map = capacity.HasValue
  213. ? new Dictionary<TKey, ISubject<TElement>>(capacity.Value, comparer)
  214. : new Dictionary<TKey, ISubject<TElement>>(comparer);
  215. var groupDisposable = new CompositeDisposable();
  216. var refCountDisposable = new RefCountDisposable(groupDisposable);
  217. groupDisposable.Add(source.Subscribe(x =>
  218. {
  219. var key = default(TKey);
  220. try
  221. {
  222. key = keySelector(x);
  223. }
  224. catch (Exception exception)
  225. {
  226. lock (map)
  227. foreach (var w in map.Values.ToArray())
  228. w.OnError(exception);
  229. observer.OnError(exception);
  230. return;
  231. }
  232. var fireNewMapEntry = false;
  233. var writer = default(ISubject<TElement>);
  234. try
  235. {
  236. lock (map)
  237. {
  238. if (!map.TryGetValue(key, out writer))
  239. {
  240. writer = new Subject<TElement>();
  241. map.Add(key, writer);
  242. fireNewMapEntry = true;
  243. }
  244. }
  245. }
  246. catch (Exception exception)
  247. {
  248. lock (map)
  249. {
  250. foreach (var w in map.Values.ToArray())
  251. w.OnError(exception);
  252. }
  253. observer.OnError(exception);
  254. return;
  255. }
  256. if (fireNewMapEntry)
  257. {
  258. var group = new GroupedObservable<TKey, TElement>(key, writer, refCountDisposable);
  259. var durationGroup = new GroupedObservable<TKey, TElement>(key, writer);
  260. var duration = default(IObservable<TDuration>);
  261. try
  262. {
  263. duration = durationSelector(durationGroup);
  264. }
  265. catch (Exception exception)
  266. {
  267. foreach (var w in map.Values.ToArray())
  268. w.OnError(exception);
  269. observer.OnError(exception);
  270. return;
  271. }
  272. observer.OnNext(group);
  273. var md = new SingleAssignmentDisposable();
  274. groupDisposable.Add(md);
  275. Action expire = () =>
  276. {
  277. lock (map)
  278. {
  279. if (map.Remove(key))
  280. writer.OnCompleted();
  281. }
  282. groupDisposable.Remove(md);
  283. };
  284. md.Disposable = duration.Take(1).Subscribe(
  285. _ => { },
  286. exception =>
  287. {
  288. lock (map)
  289. foreach (var o in map.Values.ToArray())
  290. o.OnError(exception);
  291. observer.OnError(exception);
  292. },
  293. expire);
  294. }
  295. var element = default(TElement);
  296. try
  297. {
  298. element = elementSelector(x);
  299. }
  300. catch (Exception exception)
  301. {
  302. lock (map)
  303. foreach (var w in map.Values.ToArray())
  304. w.OnError(exception);
  305. observer.OnError(exception);
  306. return;
  307. }
  308. writer.OnNext(element);
  309. },
  310. e =>
  311. {
  312. lock (map)
  313. foreach (var w in map.Values.ToArray())
  314. w.OnError(e);
  315. observer.OnError(e);
  316. },
  317. () =>
  318. {
  319. lock (map)
  320. foreach (var w in map.Values.ToArray())
  321. w.OnCompleted();
  322. observer.OnCompleted();
  323. }));
  324. return refCountDisposable;
  325. });
  326. #endif
  327. }
  328. #endregion
  329. #region + GroupJoin +
  330. public virtual IObservable<TResult> GroupJoin<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(IObservable<TLeft> left, IObservable<TRight> right, Func<TLeft, IObservable<TLeftDuration>> leftDurationSelector, Func<TRight, IObservable<TRightDuration>> rightDurationSelector, Func<TLeft, IObservable<TRight>, TResult> resultSelector)
  331. {
  332. return GroupJoin_<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(left, right, leftDurationSelector, rightDurationSelector, resultSelector);
  333. }
  334. private static IObservable<TResult> GroupJoin_<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(IObservable<TLeft> left, IObservable<TRight> right, Func<TLeft, IObservable<TLeftDuration>> leftDurationSelector, Func<TRight, IObservable<TRightDuration>> rightDurationSelector, Func<TLeft, IObservable<TRight>, TResult> resultSelector)
  335. {
  336. #if !NO_PERF
  337. return new GroupJoin<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(left, right, leftDurationSelector, rightDurationSelector, resultSelector);
  338. #else
  339. return new AnonymousObservable<TResult>(observer =>
  340. {
  341. var gate = new object();
  342. var group = new CompositeDisposable();
  343. var r = new RefCountDisposable(group);
  344. var leftMap = new Dictionary<int, IObserver<TRight>>();
  345. var rightMap = new Dictionary<int, TRight>();
  346. var leftID = 0;
  347. var rightID = 0;
  348. group.Add(left.Subscribe(
  349. value =>
  350. {
  351. var s = new Subject<TRight>();
  352. var id = 0;
  353. lock (gate)
  354. {
  355. id = leftID++;
  356. leftMap.Add(id, s);
  357. }
  358. lock (gate)
  359. {
  360. var result = default(TResult);
  361. try
  362. {
  363. result = resultSelector(value, s.AddRef(r));
  364. }
  365. catch (Exception exception)
  366. {
  367. foreach (var o in leftMap.Values.ToArray())
  368. o.OnError(exception);
  369. observer.OnError(exception);
  370. return;
  371. }
  372. observer.OnNext(result);
  373. foreach (var rightValue in rightMap.Values.ToArray())
  374. {
  375. s.OnNext(rightValue);
  376. }
  377. }
  378. var md = new SingleAssignmentDisposable();
  379. group.Add(md);
  380. Action expire = () =>
  381. {
  382. lock (gate)
  383. if (leftMap.Remove(id))
  384. s.OnCompleted();
  385. group.Remove(md);
  386. };
  387. var duration = default(IObservable<TLeftDuration>);
  388. try
  389. {
  390. duration = leftDurationSelector(value);
  391. }
  392. catch (Exception exception)
  393. {
  394. lock (gate)
  395. {
  396. foreach (var o in leftMap.Values.ToArray())
  397. o.OnError(exception);
  398. observer.OnError(exception);
  399. }
  400. return;
  401. }
  402. md.Disposable = duration.Take(1).Subscribe(
  403. _ => { },
  404. exception =>
  405. {
  406. lock (gate)
  407. {
  408. foreach (var o in leftMap.Values.ToArray())
  409. o.OnError(exception);
  410. observer.OnError(exception);
  411. }
  412. },
  413. expire);
  414. },
  415. exception =>
  416. {
  417. lock (gate)
  418. {
  419. foreach (var o in leftMap.Values.ToArray())
  420. o.OnError(exception);
  421. observer.OnError(exception);
  422. }
  423. },
  424. () =>
  425. {
  426. lock (gate)
  427. observer.OnCompleted();
  428. }));
  429. group.Add(right.Subscribe(
  430. value =>
  431. {
  432. var id = 0;
  433. lock (gate)
  434. {
  435. id = rightID++;
  436. rightMap.Add(id, value);
  437. }
  438. var md = new SingleAssignmentDisposable();
  439. group.Add(md);
  440. Action expire = () =>
  441. {
  442. lock (gate)
  443. rightMap.Remove(id);
  444. group.Remove(md);
  445. };
  446. var duration = default(IObservable<TRightDuration>);
  447. try
  448. {
  449. duration = rightDurationSelector(value);
  450. }
  451. catch (Exception exception)
  452. {
  453. lock (gate)
  454. {
  455. foreach (var o in leftMap.Values.ToArray())
  456. o.OnError(exception);
  457. observer.OnError(exception);
  458. }
  459. return;
  460. }
  461. md.Disposable = duration.Take(1).Subscribe(
  462. _ => { },
  463. exception =>
  464. {
  465. lock (gate)
  466. {
  467. foreach (var o in leftMap.Values.ToArray())
  468. o.OnError(exception);
  469. observer.OnError(exception);
  470. }
  471. },
  472. expire);
  473. lock (gate)
  474. {
  475. foreach (var o in leftMap.Values.ToArray())
  476. o.OnNext(value);
  477. }
  478. },
  479. exception =>
  480. {
  481. lock (gate)
  482. {
  483. foreach (var o in leftMap.Values.ToArray())
  484. o.OnError(exception);
  485. observer.OnError(exception);
  486. }
  487. }));
  488. return r;
  489. });
  490. #endif
  491. }
  492. #endregion
  493. #region + Join +
  494. public virtual IObservable<TResult> Join<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(IObservable<TLeft> left, IObservable<TRight> right, Func<TLeft, IObservable<TLeftDuration>> leftDurationSelector, Func<TRight, IObservable<TRightDuration>> rightDurationSelector, Func<TLeft, TRight, TResult> resultSelector)
  495. {
  496. return Join_<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(left, right, leftDurationSelector, rightDurationSelector, resultSelector);
  497. }
  498. private static IObservable<TResult> Join_<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(IObservable<TLeft> left, IObservable<TRight> right, Func<TLeft, IObservable<TLeftDuration>> leftDurationSelector, Func<TRight, IObservable<TRightDuration>> rightDurationSelector, Func<TLeft, TRight, TResult> resultSelector)
  499. {
  500. #if !NO_PERF
  501. return new Join<TLeft, TRight, TLeftDuration, TRightDuration, TResult>(left, right, leftDurationSelector, rightDurationSelector, resultSelector);
  502. #else
  503. return new AnonymousObservable<TResult>(observer =>
  504. {
  505. var gate = new object();
  506. var leftDone = false;
  507. var rightDone = false;
  508. var group = new CompositeDisposable();
  509. var leftMap = new Dictionary<int, TLeft>();
  510. var rightMap = new Dictionary<int, TRight>();
  511. var leftID = 0;
  512. var rightID = 0;
  513. group.Add(left.Subscribe(
  514. value =>
  515. {
  516. var id = 0;
  517. lock (gate)
  518. {
  519. id = leftID++;
  520. leftMap.Add(id, value);
  521. }
  522. var md = new SingleAssignmentDisposable();
  523. group.Add(md);
  524. Action expire = () =>
  525. {
  526. lock (gate)
  527. {
  528. if (leftMap.Remove(id) && leftMap.Count == 0 && leftDone)
  529. observer.OnCompleted();
  530. }
  531. group.Remove(md);
  532. };
  533. var duration = default(IObservable<TLeftDuration>);
  534. try
  535. {
  536. duration = leftDurationSelector(value);
  537. }
  538. catch (Exception exception)
  539. {
  540. observer.OnError(exception);
  541. return;
  542. }
  543. md.Disposable = duration.Take(1).Subscribe(
  544. _ => { },
  545. error =>
  546. {
  547. lock (gate)
  548. observer.OnError(error);
  549. },
  550. expire);
  551. lock (gate)
  552. {
  553. foreach (var rightValue in rightMap.Values.ToArray())
  554. {
  555. var result = default(TResult);
  556. try
  557. {
  558. result = resultSelector(value, rightValue);
  559. }
  560. catch (Exception exception)
  561. {
  562. observer.OnError(exception);
  563. return;
  564. }
  565. observer.OnNext(result);
  566. }
  567. }
  568. },
  569. error =>
  570. {
  571. lock (gate)
  572. observer.OnError(error);
  573. },
  574. () =>
  575. {
  576. lock (gate)
  577. {
  578. leftDone = true;
  579. if (rightDone || leftMap.Count == 0)
  580. observer.OnCompleted();
  581. }
  582. }));
  583. group.Add(right.Subscribe(
  584. value =>
  585. {
  586. var id = 0;
  587. lock (gate)
  588. {
  589. id = rightID++;
  590. rightMap.Add(id, value);
  591. }
  592. var md = new SingleAssignmentDisposable();
  593. group.Add(md);
  594. Action expire = () =>
  595. {
  596. lock (gate)
  597. {
  598. if (rightMap.Remove(id) && rightMap.Count == 0 && rightDone)
  599. observer.OnCompleted();
  600. }
  601. group.Remove(md);
  602. };
  603. var duration = default(IObservable<TRightDuration>);
  604. try
  605. {
  606. duration = rightDurationSelector(value);
  607. }
  608. catch (Exception exception)
  609. {
  610. observer.OnError(exception);
  611. return;
  612. }
  613. md.Disposable = duration.Take(1).Subscribe(
  614. _ => { },
  615. error =>
  616. {
  617. lock (gate)
  618. observer.OnError(error);
  619. },
  620. expire);
  621. lock (gate)
  622. {
  623. foreach (var leftValue in leftMap.Values.ToArray())
  624. {
  625. var result = default(TResult);
  626. try
  627. {
  628. result = resultSelector(leftValue, value);
  629. }
  630. catch (Exception exception)
  631. {
  632. observer.OnError(exception);
  633. return;
  634. }
  635. observer.OnNext(result);
  636. }
  637. }
  638. },
  639. error =>
  640. {
  641. lock (gate)
  642. observer.OnError(error);
  643. },
  644. () =>
  645. {
  646. lock (gate)
  647. {
  648. rightDone = true;
  649. if (leftDone || rightMap.Count == 0)
  650. observer.OnCompleted();
  651. }
  652. }));
  653. return group;
  654. });
  655. #endif
  656. }
  657. #endregion
  658. #region + OfType +
  659. public virtual IObservable<TResult> OfType<TResult>(IObservable<object> source)
  660. {
  661. #if !NO_PERF
  662. return new OfType<object, TResult>(source);
  663. #else
  664. return source.Where(x => x is TResult).Cast<TResult>();
  665. #endif
  666. }
  667. #endregion
  668. #region + OrderBy +
  669. public virtual IOrderedObservable<TSource> OrderBy<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector)
  670. {
  671. return new OrderBy<TSource, TKey>(source, keySelector, comparer: null, descending: false);
  672. }
  673. public virtual IOrderedObservable<TSource> OrderBy<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector, IComparer<TKey> comparer)
  674. {
  675. return new OrderBy<TSource, TKey>(source, keySelector, comparer, descending: false);
  676. }
  677. public virtual IOrderedObservable<TSource> OrderBy<TSource, TOther>(IObservable<TSource> source, Func<TSource, IObservable<TOther>> timeSelector)
  678. {
  679. return new OrderBy<TSource, TOther>(source, timeSelector, descending: false);
  680. }
  681. #endregion
  682. #region + OrderByDescending +
  683. public virtual IOrderedObservable<TSource> OrderByDescending<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector)
  684. {
  685. return new OrderBy<TSource, TKey>(source, keySelector, comparer: null, descending: true);
  686. }
  687. public virtual IOrderedObservable<TSource> OrderByDescending<TSource, TKey>(IObservable<TSource> source, Func<TSource, TKey> keySelector, IComparer<TKey> comparer)
  688. {
  689. return new OrderBy<TSource, TKey>(source, keySelector, comparer, descending: true);
  690. }
  691. public virtual IOrderedObservable<TSource> OrderByDescending<TSource, TOther>(IObservable<TSource> source, Func<TSource, IObservable<TOther>> timeSelector)
  692. {
  693. return new OrderBy<TSource, TOther>(source, timeSelector, descending: true);
  694. }
  695. #endregion
  696. #region + Select +
  697. public virtual IObservable<TResult> Select<TSource, TResult>(IObservable<TSource> source, Func<TSource, TResult> selector)
  698. {
  699. #if !NO_PERF
  700. var select = source as Select<TSource>;
  701. if (select != null)
  702. return select.Omega(selector);
  703. return new Select<TSource, TResult>(source, selector);
  704. #else
  705. var s = source as SelectObservable<TSource>;
  706. if (s != null)
  707. return s.Select(selector);
  708. return new SelectObservable<TSource, TResult>(source, selector);
  709. #endif
  710. }
  711. #if NO_PERF
  712. abstract class SelectObservable<TResult> : ObservableBase<TResult>
  713. {
  714. public abstract IObservable<TResult2> Select<TResult2>(Func<TResult, TResult2> selector);
  715. }
  716. class SelectObservable<TSource, TResult> : SelectObservable<TResult>
  717. {
  718. private readonly IObservable<TSource> _source;
  719. private readonly Func<TSource, TResult> _selector;
  720. public SelectObservable(IObservable<TSource> source, Func<TSource, TResult> selector)
  721. {
  722. _source = source;
  723. _selector = selector;
  724. }
  725. protected override IDisposable SubscribeCore(IObserver<TResult> observer)
  726. {
  727. return _source.Subscribe(new Observer(observer, _selector));
  728. }
  729. public override IObservable<TResult2> Select<TResult2>(Func<TResult, TResult2> selector)
  730. {
  731. return new SelectObservable<TSource, TResult2>(_source, x => selector(_selector(x)));
  732. }
  733. class Observer : ObserverBase<TSource>
  734. {
  735. private readonly IObserver<TResult> _observer;
  736. private readonly Func<TSource, TResult> _selector;
  737. public Observer(IObserver<TResult> observer, Func<TSource, TResult> selector)
  738. {
  739. _observer = observer;
  740. _selector = selector;
  741. }
  742. protected override void OnNextCore(TSource value)
  743. {
  744. TResult result;
  745. try
  746. {
  747. result = _selector(value);
  748. }
  749. catch (Exception exception)
  750. {
  751. _observer.OnError(exception);
  752. return;
  753. }
  754. _observer.OnNext(result);
  755. }
  756. protected override void OnErrorCore(Exception error)
  757. {
  758. _observer.OnError(error);
  759. }
  760. protected override void OnCompletedCore()
  761. {
  762. _observer.OnCompleted();
  763. }
  764. }
  765. }
  766. #endif
  767. public virtual IObservable<TResult> Select<TSource, TResult>(IObservable<TSource> source, Func<TSource, int, TResult> selector)
  768. {
  769. #if !NO_PERF
  770. return new Select<TSource, TResult>(source, selector);
  771. #else
  772. return Defer(() =>
  773. {
  774. var index = 0;
  775. return source.Select(x => selector(x, checked(index++)));
  776. });
  777. #endif
  778. }
  779. #endregion
  780. #region + SelectMany +
  781. public virtual IObservable<TOther> SelectMany<TSource, TOther>(IObservable<TSource> source, IObservable<TOther> other)
  782. {
  783. return SelectMany_<TSource, TOther>(source, _ => other);
  784. }
  785. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, IObservable<TResult>> selector)
  786. {
  787. return SelectMany_<TSource, TResult>(source, selector);
  788. }
  789. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, int, IObservable<TResult>> selector)
  790. {
  791. return SelectMany_<TSource, TResult>(source, selector);
  792. }
  793. #if !NO_TPL
  794. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, Task<TResult>> selector)
  795. {
  796. #if !NO_PERF
  797. return new SelectMany<TSource, TResult>(source, (x, token) => selector(x));
  798. #else
  799. return SelectMany_<TSource, TResult>(source, x => selector(x).ToObservable());
  800. #endif
  801. }
  802. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, int, Task<TResult>> selector)
  803. {
  804. #if !NO_PERF
  805. return new SelectMany<TSource, TResult>(source, (x, i, token) => selector(x, i));
  806. #else
  807. return SelectMany_<TSource, TResult>(source, (x, i) => selector(x, i).ToObservable());
  808. #endif
  809. }
  810. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, CancellationToken, Task<TResult>> selector)
  811. {
  812. #if !NO_PERF
  813. return new SelectMany<TSource, TResult>(source, selector);
  814. #else
  815. return SelectMany_<TSource, TResult>(source, x => FromAsync(ct => selector(x, ct)));
  816. #endif
  817. }
  818. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, int, CancellationToken, Task<TResult>> selector)
  819. {
  820. #if !NO_PERF
  821. return new SelectMany<TSource, TResult>(source, selector);
  822. #else
  823. return SelectMany_<TSource, TResult>(source, (x, i) => FromAsync(ct => selector(x, i, ct)));
  824. #endif
  825. }
  826. #endif
  827. public virtual IObservable<TResult> SelectMany<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, IObservable<TCollection>> collectionSelector, Func<TSource, TCollection, TResult> resultSelector)
  828. {
  829. return SelectMany_<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  830. }
  831. public virtual IObservable<TResult> SelectMany<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, int, IObservable<TCollection>> collectionSelector, Func<TSource, int, TCollection, int, TResult> resultSelector)
  832. {
  833. return SelectMany_<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  834. }
  835. #if !NO_TPL
  836. public virtual IObservable<TResult> SelectMany<TSource, TTaskResult, TResult>(IObservable<TSource> source, Func<TSource, Task<TTaskResult>> taskSelector, Func<TSource, TTaskResult, TResult> resultSelector)
  837. {
  838. #if !NO_PERF
  839. return new SelectMany<TSource, TTaskResult, TResult>(source, (x, token) => taskSelector(x), resultSelector);
  840. #else
  841. return SelectMany_<TSource, TTaskResult, TResult>(source, x => taskSelector(x).ToObservable(), resultSelector);
  842. #endif
  843. }
  844. public virtual IObservable<TResult> SelectMany<TSource, TTaskResult, TResult>(IObservable<TSource> source, Func<TSource, int, Task<TTaskResult>> taskSelector, Func<TSource, int, TTaskResult, TResult> resultSelector)
  845. {
  846. #if !NO_PERF
  847. return new SelectMany<TSource, TTaskResult, TResult>(source, (x, i, token) => taskSelector(x, i), resultSelector);
  848. #else
  849. return SelectMany_<TSource, TTaskResult, TResult>(source, (x, i) => taskSelector(x, i).ToObservable(), (x, i, t, _) => resultSelector(x, i, t));
  850. #endif
  851. }
  852. public virtual IObservable<TResult> SelectMany<TSource, TTaskResult, TResult>(IObservable<TSource> source, Func<TSource, CancellationToken, Task<TTaskResult>> taskSelector, Func<TSource, TTaskResult, TResult> resultSelector)
  853. {
  854. #if !NO_PERF
  855. return new SelectMany<TSource, TTaskResult, TResult>(source, taskSelector, resultSelector);
  856. #else
  857. return SelectMany_<TSource, TTaskResult, TResult>(source, x => FromAsync(ct => taskSelector(x, ct)), resultSelector);
  858. #endif
  859. }
  860. public virtual IObservable<TResult> SelectMany<TSource, TTaskResult, TResult>(IObservable<TSource> source, Func<TSource, int, CancellationToken, Task<TTaskResult>> taskSelector, Func<TSource, int, TTaskResult, TResult> resultSelector)
  861. {
  862. #if !NO_PERF
  863. return new SelectMany<TSource, TTaskResult, TResult>(source, taskSelector, resultSelector);
  864. #else
  865. return SelectMany_<TSource, TTaskResult, TResult>(source, (x, i) => FromAsync(ct => taskSelector(x, i, ct)), (x, i, t, _) => resultSelector(x, i, t));
  866. #endif
  867. }
  868. #endif
  869. private static IObservable<TResult> SelectMany_<TSource, TResult>(IObservable<TSource> source, Func<TSource, IObservable<TResult>> selector)
  870. {
  871. #if !NO_PERF
  872. return new SelectMany<TSource, TResult>(source, selector);
  873. #else
  874. return source.Select(selector).Merge();
  875. #endif
  876. }
  877. private static IObservable<TResult> SelectMany_<TSource, TResult>(IObservable<TSource> source, Func<TSource, int, IObservable<TResult>> selector)
  878. {
  879. #if !NO_PERF
  880. return new SelectMany<TSource, TResult>(source, selector);
  881. #else
  882. return source.Select(selector).Merge();
  883. #endif
  884. }
  885. private static IObservable<TResult> SelectMany_<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, IObservable<TCollection>> collectionSelector, Func<TSource, TCollection, TResult> resultSelector)
  886. {
  887. #if !NO_PERF
  888. return new SelectMany<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  889. #else
  890. return SelectMany_<TSource, TResult>(source, x => collectionSelector(x).Select(y => resultSelector(x, y)));
  891. #endif
  892. }
  893. private static IObservable<TResult> SelectMany_<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, int, IObservable<TCollection>> collectionSelector, Func<TSource, int, TCollection, int, TResult> resultSelector)
  894. {
  895. #if !NO_PERF
  896. return new SelectMany<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  897. #else
  898. return SelectMany_<TSource, TResult>(source, (x, i) => collectionSelector(x, i).Select((y, i2) => resultSelector(x, i, y, i2)));
  899. #endif
  900. }
  901. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, IObservable<TResult>> onNext, Func<Exception, IObservable<TResult>> onError, Func<IObservable<TResult>> onCompleted)
  902. {
  903. #if !NO_PERF
  904. return new SelectMany<TSource, TResult>(source, onNext, onError, onCompleted);
  905. #else
  906. return source.Materialize().SelectMany(notification =>
  907. {
  908. if (notification.Kind == NotificationKind.OnNext)
  909. return onNext(notification.Value);
  910. else if (notification.Kind == NotificationKind.OnError)
  911. return onError(notification.Exception);
  912. else
  913. return onCompleted();
  914. });
  915. #endif
  916. }
  917. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, int, IObservable<TResult>> onNext, Func<Exception, IObservable<TResult>> onError, Func<IObservable<TResult>> onCompleted)
  918. {
  919. #if !NO_PERF
  920. return new SelectMany<TSource, TResult>(source, onNext, onError, onCompleted);
  921. #else
  922. return Defer(() =>
  923. {
  924. var index = 0;
  925. return source.Materialize().SelectMany(notification =>
  926. {
  927. if (notification.Kind == NotificationKind.OnNext)
  928. return onNext(notification.Value, checked(index++));
  929. else if (notification.Kind == NotificationKind.OnError)
  930. return onError(notification.Exception);
  931. else
  932. return onCompleted();
  933. });
  934. });
  935. #endif
  936. }
  937. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, IEnumerable<TResult>> selector)
  938. {
  939. #if !NO_PERF
  940. return new SelectMany<TSource, TResult>(source, selector);
  941. #else
  942. return SelectMany_<TSource, TResult, TResult>(source, selector, (_, x) => x);
  943. #endif
  944. }
  945. public virtual IObservable<TResult> SelectMany<TSource, TResult>(IObservable<TSource> source, Func<TSource, int, IEnumerable<TResult>> selector)
  946. {
  947. #if !NO_PERF
  948. return new SelectMany<TSource, TResult>(source, selector);
  949. #else
  950. return SelectMany_<TSource, TResult, TResult>(source, selector, (_, __, x, ___) => x);
  951. #endif
  952. }
  953. public virtual IObservable<TResult> SelectMany<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, IEnumerable<TCollection>> collectionSelector, Func<TSource, TCollection, TResult> resultSelector)
  954. {
  955. return SelectMany_<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  956. }
  957. public virtual IObservable<TResult> SelectMany<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, int, IEnumerable<TCollection>> collectionSelector, Func<TSource, int, TCollection, int, TResult> resultSelector)
  958. {
  959. return SelectMany_<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  960. }
  961. private static IObservable<TResult> SelectMany_<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, IEnumerable<TCollection>> collectionSelector, Func<TSource, TCollection, TResult> resultSelector)
  962. {
  963. #if !NO_PERF
  964. return new SelectMany<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  965. #else
  966. return new AnonymousObservable<TResult>(observer =>
  967. source.Subscribe(
  968. x =>
  969. {
  970. var xs = default(IEnumerable<TCollection>);
  971. try
  972. {
  973. xs = collectionSelector(x);
  974. }
  975. catch (Exception exception)
  976. {
  977. observer.OnError(exception);
  978. return;
  979. }
  980. var e = xs.GetEnumerator();
  981. try
  982. {
  983. var hasNext = true;
  984. while (hasNext)
  985. {
  986. hasNext = false;
  987. var current = default(TResult);
  988. try
  989. {
  990. hasNext = e.MoveNext();
  991. if (hasNext)
  992. current = resultSelector(x, e.Current);
  993. }
  994. catch (Exception exception)
  995. {
  996. observer.OnError(exception);
  997. return;
  998. }
  999. if (hasNext)
  1000. observer.OnNext(current);
  1001. }
  1002. }
  1003. finally
  1004. {
  1005. if (e != null)
  1006. e.Dispose();
  1007. }
  1008. },
  1009. observer.OnError,
  1010. observer.OnCompleted
  1011. )
  1012. );
  1013. #endif
  1014. }
  1015. private static IObservable<TResult> SelectMany_<TSource, TCollection, TResult>(IObservable<TSource> source, Func<TSource, int, IEnumerable<TCollection>> collectionSelector, Func<TSource, int, TCollection, int, TResult> resultSelector)
  1016. {
  1017. #if !NO_PERF
  1018. return new SelectMany<TSource, TCollection, TResult>(source, collectionSelector, resultSelector);
  1019. #else
  1020. return new AnonymousObservable<TResult>(observer =>
  1021. {
  1022. var index = 0;
  1023. return source.Subscribe(
  1024. x =>
  1025. {
  1026. var xs = default(IEnumerable<TCollection>);
  1027. try
  1028. {
  1029. xs = collectionSelector(x, checked(index++));
  1030. }
  1031. catch (Exception exception)
  1032. {
  1033. observer.OnError(exception);
  1034. return;
  1035. }
  1036. var e = xs.GetEnumerator();
  1037. try
  1038. {
  1039. var eIndex = 0;
  1040. var hasNext = true;
  1041. while (hasNext)
  1042. {
  1043. hasNext = false;
  1044. var current = default(TResult);
  1045. try
  1046. {
  1047. hasNext = e.MoveNext();
  1048. if (hasNext)
  1049. current = resultSelector(x, index, e.Current, checked(eIndex++));
  1050. }
  1051. catch (Exception exception)
  1052. {
  1053. observer.OnError(exception);
  1054. return;
  1055. }
  1056. if (hasNext)
  1057. observer.OnNext(current);
  1058. }
  1059. }
  1060. finally
  1061. {
  1062. if (e != null)
  1063. e.Dispose();
  1064. }
  1065. },
  1066. observer.OnError,
  1067. observer.OnCompleted
  1068. )
  1069. });
  1070. #endif
  1071. }
  1072. #endregion
  1073. #region + Skip +
  1074. public virtual IObservable<TSource> Skip<TSource>(IObservable<TSource> source, int count)
  1075. {
  1076. #if !NO_PERF
  1077. var skip = source as Skip<TSource>;
  1078. if (skip != null && skip._scheduler == null)
  1079. return skip.Omega(count);
  1080. return new Skip<TSource>(source, count);
  1081. #else
  1082. return new AnonymousObservable<TSource>(observer =>
  1083. {
  1084. var remaining = count;
  1085. return source.Subscribe(
  1086. x =>
  1087. {
  1088. if (remaining <= 0)
  1089. observer.OnNext(x);
  1090. else
  1091. remaining--;
  1092. },
  1093. observer.OnError,
  1094. observer.OnCompleted);
  1095. });
  1096. #endif
  1097. }
  1098. #endregion
  1099. #region + SkipWhile +
  1100. public virtual IObservable<TSource> SkipWhile<TSource>(IObservable<TSource> source, Func<TSource, bool> predicate)
  1101. {
  1102. #if !NO_PERF
  1103. return new SkipWhile<TSource>(source, predicate);
  1104. #else
  1105. return SkipWhile_(source, (x, i) => predicate(x));
  1106. #endif
  1107. }
  1108. public virtual IObservable<TSource> SkipWhile<TSource>(IObservable<TSource> source, Func<TSource, int, bool> predicate)
  1109. {
  1110. #if !NO_PERF
  1111. return new SkipWhile<TSource>(source, predicate);
  1112. #else
  1113. return SkipWhile_(source, predicate);
  1114. #endif
  1115. }
  1116. #if NO_PERF
  1117. private static IObservable<TSource> SkipWhile_<TSource>(IObservable<TSource> source, Func<TSource, int, bool> predicate)
  1118. {
  1119. return new AnonymousObservable<TSource>(observer =>
  1120. {
  1121. var running = false;
  1122. var i = 0;
  1123. return source.Subscribe(
  1124. x =>
  1125. {
  1126. if (!running)
  1127. try
  1128. {
  1129. running = !predicate(x, checked(i++));
  1130. }
  1131. catch (Exception exception)
  1132. {
  1133. observer.OnError(exception);
  1134. return;
  1135. }
  1136. if (running)
  1137. observer.OnNext(x);
  1138. },
  1139. observer.OnError,
  1140. observer.OnCompleted);
  1141. });
  1142. }
  1143. #endif
  1144. #endregion
  1145. #region + Take +
  1146. public virtual IObservable<TSource> Take<TSource>(IObservable<TSource> source, int count)
  1147. {
  1148. if (count == 0)
  1149. return Empty<TSource>();
  1150. return Take_(source, count);
  1151. }
  1152. public virtual IObservable<TSource> Take<TSource>(IObservable<TSource> source, int count, IScheduler scheduler)
  1153. {
  1154. if (count == 0)
  1155. return Empty<TSource>(scheduler);
  1156. return Take_(source, count);
  1157. }
  1158. #if !NO_PERF
  1159. private static IObservable<TSource> Take_<TSource>(IObservable<TSource> source, int count)
  1160. {
  1161. var take = source as Take<TSource>;
  1162. if (take != null && take._scheduler == null)
  1163. return take.Omega(count);
  1164. return new Take<TSource>(source, count);
  1165. }
  1166. #else
  1167. private static IObservable<TSource> Take_<TSource>(IObservable<TSource> source, int count)
  1168. {
  1169. return new AnonymousObservable<TSource>(observer =>
  1170. {
  1171. var remaining = count;
  1172. return source.Subscribe(
  1173. x =>
  1174. {
  1175. if (remaining > 0)
  1176. {
  1177. --remaining;
  1178. observer.OnNext(x);
  1179. if (remaining == 0)
  1180. observer.OnCompleted();
  1181. }
  1182. },
  1183. observer.OnError,
  1184. observer.OnCompleted);
  1185. });
  1186. }
  1187. #endif
  1188. #endregion
  1189. #region + TakeWhile +
  1190. public virtual IObservable<TSource> TakeWhile<TSource>(IObservable<TSource> source, Func<TSource, bool> predicate)
  1191. {
  1192. #if !NO_PERF
  1193. return new TakeWhile<TSource>(source, predicate);
  1194. #else
  1195. return TakeWhile_(source, (x, i) => predicate(x));
  1196. #endif
  1197. }
  1198. public virtual IObservable<TSource> TakeWhile<TSource>(IObservable<TSource> source, Func<TSource, int, bool> predicate)
  1199. {
  1200. #if !NO_PERF
  1201. return new TakeWhile<TSource>(source, predicate);
  1202. #else
  1203. return TakeWhile_(source, predicate);
  1204. #endif
  1205. }
  1206. #if NO_PERF
  1207. private static IObservable<TSource> TakeWhile_<TSource>(IObservable<TSource> source, Func<TSource, int, bool> predicate)
  1208. {
  1209. return new AnonymousObservable<TSource>(observer =>
  1210. {
  1211. var running = true;
  1212. var i = 0;
  1213. return source.Subscribe(
  1214. x =>
  1215. {
  1216. if (running)
  1217. {
  1218. try
  1219. {
  1220. running = predicate(x, checked(i++));
  1221. }
  1222. catch (Exception exception)
  1223. {
  1224. observer.OnError(exception);
  1225. return;
  1226. }
  1227. if (running)
  1228. observer.OnNext(x);
  1229. else
  1230. observer.OnCompleted();
  1231. }
  1232. },
  1233. observer.OnError,
  1234. observer.OnCompleted);
  1235. });
  1236. }
  1237. #endif
  1238. #endregion
  1239. #region + ThenBy +
  1240. public virtual IOrderedObservable<TSource> ThenBy<TSource, TKey>(IOrderedObservable<TSource> source, Func<TSource, TKey> keySelector)
  1241. {
  1242. return source.CreateOrderedObservable(keySelector, comparer: null, descending: false);
  1243. }
  1244. public virtual IOrderedObservable<TSource> ThenBy<TSource, TKey>(IOrderedObservable<TSource> source, Func<TSource, TKey> keySelector, IComparer<TKey> comparer)
  1245. {
  1246. return source.CreateOrderedObservable(keySelector, comparer, descending: false);
  1247. }
  1248. public virtual IOrderedObservable<TSource> ThenBy<TSource, TOther>(IOrderedObservable<TSource> source, Func<TSource, IObservable<TOther>> timeSelector)
  1249. {
  1250. return source.CreateOrderedObservable(timeSelector, descending: false);
  1251. }
  1252. #endregion
  1253. #region + ThenByDescending +
  1254. public virtual IOrderedObservable<TSource> ThenByDescending<TSource, TKey>(IOrderedObservable<TSource> source, Func<TSource, TKey> keySelector)
  1255. {
  1256. return source.CreateOrderedObservable(keySelector, comparer: null, descending: true);
  1257. }
  1258. public virtual IOrderedObservable<TSource> ThenByDescending<TSource, TKey>(IOrderedObservable<TSource> source, Func<TSource, TKey> keySelector, IComparer<TKey> comparer)
  1259. {
  1260. return source.CreateOrderedObservable(keySelector, comparer, descending: true);
  1261. }
  1262. public virtual IOrderedObservable<TSource> ThenByDescending<TSource, TOther>(IOrderedObservable<TSource> source, Func<TSource, IObservable<TOther>> timeSelector)
  1263. {
  1264. return source.CreateOrderedObservable(timeSelector, descending: true);
  1265. }
  1266. #endregion
  1267. #region + Where +
  1268. public virtual IObservable<TSource> Where<TSource>(IObservable<TSource> source, Func<TSource, bool> predicate)
  1269. {
  1270. #if !NO_PERF
  1271. var where = source as Where<TSource>;
  1272. if (where != null)
  1273. return where.Omega(predicate);
  1274. return new Where<TSource>(source, predicate);
  1275. #else
  1276. var w = source as WhereObservable<TSource>;
  1277. if (w != null)
  1278. return w.Where(predicate);
  1279. return new WhereObservable<TSource>(source, predicate);
  1280. #endif
  1281. }
  1282. #if NO_PERF
  1283. class WhereObservable<TSource> : ObservableBase<TSource>
  1284. {
  1285. private readonly IObservable<TSource> _source;
  1286. private readonly Func<TSource, bool> _predicate;
  1287. public WhereObservable(IObservable<TSource> source, Func<TSource, bool> predicate)
  1288. {
  1289. _source = source;
  1290. _predicate = predicate;
  1291. }
  1292. protected override IDisposable SubscribeCore(IObserver<TSource> observer)
  1293. {
  1294. return _source.Subscribe(new Observer(observer, _predicate));
  1295. }
  1296. public IObservable<TSource> Where(Func<TSource, bool> predicate)
  1297. {
  1298. return new WhereObservable<TSource>(_source, x => _predicate(x) && predicate(x));
  1299. }
  1300. class Observer : ObserverBase<TSource>
  1301. {
  1302. private readonly IObserver<TSource> _observer;
  1303. private readonly Func<TSource, bool> _predicate;
  1304. public Observer(IObserver<TSource> observer, Func<TSource, bool> predicate)
  1305. {
  1306. _observer = observer;
  1307. _predicate = predicate;
  1308. }
  1309. protected override void OnNextCore(TSource value)
  1310. {
  1311. bool shouldRun;
  1312. try
  1313. {
  1314. shouldRun = _predicate(value);
  1315. }
  1316. catch (Exception exception)
  1317. {
  1318. _observer.OnError(exception);
  1319. return;
  1320. }
  1321. if (shouldRun)
  1322. _observer.OnNext(value);
  1323. }
  1324. protected override void OnErrorCore(Exception error)
  1325. {
  1326. _observer.OnError(error);
  1327. }
  1328. protected override void OnCompletedCore()
  1329. {
  1330. _observer.OnCompleted();
  1331. }
  1332. }
  1333. }
  1334. #endif
  1335. public virtual IObservable<TSource> Where<TSource>(IObservable<TSource> source, Func<TSource, int, bool> predicate)
  1336. {
  1337. #if !NO_PERF
  1338. return new Where<TSource>(source, predicate);
  1339. #else
  1340. return Defer(() =>
  1341. {
  1342. var index = 0;
  1343. return source.Where(x => predicate(x, checked(index++)));
  1344. });
  1345. #endif
  1346. }
  1347. #endregion
  1348. }
  1349. }