Select.cs 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845
  1. // Licensed to the .NET Foundation under one or more agreements.
  2. // The .NET Foundation licenses this file to you under the Apache 2.0 License.
  3. // See the LICENSE file in the project root for more information.
  4. using System.Collections.Generic;
  5. using System.Diagnostics;
  6. using System.Threading;
  7. using System.Threading.Tasks;
  8. namespace System.Linq
  9. {
  10. public static partial class AsyncEnumerable
  11. {
  12. public static IAsyncEnumerable<TResult> Select<TSource, TResult>(this IAsyncEnumerable<TSource> source, Func<TSource, TResult> selector)
  13. {
  14. if (source == null)
  15. throw Error.ArgumentNull(nameof(source));
  16. if (selector == null)
  17. throw Error.ArgumentNull(nameof(selector));
  18. switch (source)
  19. {
  20. case AsyncIterator<TSource> iterator:
  21. return iterator.Select(selector);
  22. case IList<TSource> list:
  23. return new SelectIListIterator<TSource, TResult>(list, selector);
  24. }
  25. return new SelectEnumerableAsyncIterator<TSource, TResult>(source, selector);
  26. }
  27. public static IAsyncEnumerable<TResult> Select<TSource, TResult>(this IAsyncEnumerable<TSource> source, Func<TSource, int, TResult> selector)
  28. {
  29. if (source == null)
  30. throw Error.ArgumentNull(nameof(source));
  31. if (selector == null)
  32. throw Error.ArgumentNull(nameof(selector));
  33. return new SelectEnumerableWithIndexAsyncIterator<TSource, TResult>(source, selector);
  34. }
  35. public static IAsyncEnumerable<TResult> Select<TSource, TResult>(this IAsyncEnumerable<TSource> source, Func<TSource, ValueTask<TResult>> selector)
  36. {
  37. if (source == null)
  38. throw Error.ArgumentNull(nameof(source));
  39. if (selector == null)
  40. throw Error.ArgumentNull(nameof(selector));
  41. switch (source)
  42. {
  43. case AsyncIterator<TSource> iterator:
  44. return iterator.Select(selector);
  45. case IList<TSource> list:
  46. return new SelectIListIteratorWithTask<TSource, TResult>(list, selector);
  47. }
  48. return new SelectEnumerableAsyncIteratorWithTask<TSource, TResult>(source, selector);
  49. }
  50. #if !NO_DEEP_CANCELLATION
  51. public static IAsyncEnumerable<TResult> Select<TSource, TResult>(this IAsyncEnumerable<TSource> source, Func<TSource, CancellationToken, ValueTask<TResult>> selector)
  52. {
  53. if (source == null)
  54. throw Error.ArgumentNull(nameof(source));
  55. if (selector == null)
  56. throw Error.ArgumentNull(nameof(selector));
  57. switch (source)
  58. {
  59. case AsyncIterator<TSource> iterator:
  60. return iterator.Select(selector);
  61. case IList<TSource> list:
  62. return new SelectIListIteratorWithTaskAndCancellation<TSource, TResult>(list, selector);
  63. }
  64. return new SelectEnumerableAsyncIteratorWithTaskAndCancellation<TSource, TResult>(source, selector);
  65. }
  66. #endif
  67. public static IAsyncEnumerable<TResult> Select<TSource, TResult>(this IAsyncEnumerable<TSource> source, Func<TSource, int, ValueTask<TResult>> selector)
  68. {
  69. if (source == null)
  70. throw Error.ArgumentNull(nameof(source));
  71. if (selector == null)
  72. throw Error.ArgumentNull(nameof(selector));
  73. return new SelectEnumerableWithIndexAsyncIteratorWithTask<TSource, TResult>(source, selector);
  74. }
  75. #if !NO_DEEP_CANCELLATION
  76. public static IAsyncEnumerable<TResult> Select<TSource, TResult>(this IAsyncEnumerable<TSource> source, Func<TSource, int, CancellationToken, ValueTask<TResult>> selector)
  77. {
  78. if (source == null)
  79. throw Error.ArgumentNull(nameof(source));
  80. if (selector == null)
  81. throw Error.ArgumentNull(nameof(selector));
  82. return new SelectEnumerableWithIndexAsyncIteratorWithTaskAndCancellation<TSource, TResult>(source, selector);
  83. }
  84. #endif
  85. private static Func<TSource, TResult> CombineSelectors<TSource, TMiddle, TResult>(Func<TSource, TMiddle> selector1, Func<TMiddle, TResult> selector2)
  86. {
  87. return x => selector2(selector1(x));
  88. }
  89. private static Func<TSource, ValueTask<TResult>> CombineSelectors<TSource, TMiddle, TResult>(Func<TSource, ValueTask<TMiddle>> selector1, Func<TMiddle, ValueTask<TResult>> selector2)
  90. {
  91. return async x => await selector2(await selector1(x).ConfigureAwait(false)).ConfigureAwait(false);
  92. }
  93. #if !NO_DEEP_CANCELLATION
  94. private static Func<TSource, CancellationToken, ValueTask<TResult>> CombineSelectors<TSource, TMiddle, TResult>(Func<TSource, CancellationToken, ValueTask<TMiddle>> selector1, Func<TMiddle, CancellationToken, ValueTask<TResult>> selector2)
  95. {
  96. return async (x, ct) => await selector2(await selector1(x, ct).ConfigureAwait(false), ct).ConfigureAwait(false);
  97. }
  98. #endif
  99. internal sealed class SelectEnumerableAsyncIterator<TSource, TResult> : AsyncIterator<TResult>
  100. {
  101. private readonly Func<TSource, TResult> _selector;
  102. private readonly IAsyncEnumerable<TSource> _source;
  103. private IAsyncEnumerator<TSource> _enumerator;
  104. public SelectEnumerableAsyncIterator(IAsyncEnumerable<TSource> source, Func<TSource, TResult> selector)
  105. {
  106. Debug.Assert(source != null);
  107. Debug.Assert(selector != null);
  108. _source = source;
  109. _selector = selector;
  110. }
  111. public override AsyncIteratorBase<TResult> Clone()
  112. {
  113. return new SelectEnumerableAsyncIterator<TSource, TResult>(_source, _selector);
  114. }
  115. public override async ValueTask DisposeAsync()
  116. {
  117. if (_enumerator != null)
  118. {
  119. await _enumerator.DisposeAsync().ConfigureAwait(false);
  120. _enumerator = null;
  121. }
  122. await base.DisposeAsync().ConfigureAwait(false);
  123. }
  124. public override IAsyncEnumerable<TResult1> Select<TResult1>(Func<TResult, TResult1> selector)
  125. {
  126. return new SelectEnumerableAsyncIterator<TSource, TResult1>(_source, CombineSelectors(_selector, selector));
  127. }
  128. protected override async ValueTask<bool> MoveNextCore()
  129. {
  130. switch (_state)
  131. {
  132. case AsyncIteratorState.Allocated:
  133. _enumerator = _source.GetAsyncEnumerator(_cancellationToken);
  134. _state = AsyncIteratorState.Iterating;
  135. goto case AsyncIteratorState.Iterating;
  136. case AsyncIteratorState.Iterating:
  137. if (await _enumerator.MoveNextAsync().ConfigureAwait(false))
  138. {
  139. _current = _selector(_enumerator.Current);
  140. return true;
  141. }
  142. break;
  143. }
  144. await DisposeAsync().ConfigureAwait(false);
  145. return false;
  146. }
  147. }
  148. internal sealed class SelectEnumerableWithIndexAsyncIterator<TSource, TResult> : AsyncIterator<TResult>
  149. {
  150. private readonly Func<TSource, int, TResult> _selector;
  151. private readonly IAsyncEnumerable<TSource> _source;
  152. private IAsyncEnumerator<TSource> _enumerator;
  153. private int _index;
  154. public SelectEnumerableWithIndexAsyncIterator(IAsyncEnumerable<TSource> source, Func<TSource, int, TResult> selector)
  155. {
  156. Debug.Assert(source != null);
  157. Debug.Assert(selector != null);
  158. _source = source;
  159. _selector = selector;
  160. }
  161. public override AsyncIteratorBase<TResult> Clone()
  162. {
  163. return new SelectEnumerableWithIndexAsyncIterator<TSource, TResult>(_source, _selector);
  164. }
  165. public override async ValueTask DisposeAsync()
  166. {
  167. if (_enumerator != null)
  168. {
  169. await _enumerator.DisposeAsync().ConfigureAwait(false);
  170. _enumerator = null;
  171. }
  172. await base.DisposeAsync().ConfigureAwait(false);
  173. }
  174. protected override async ValueTask<bool> MoveNextCore()
  175. {
  176. switch (_state)
  177. {
  178. case AsyncIteratorState.Allocated:
  179. _enumerator = _source.GetAsyncEnumerator(_cancellationToken);
  180. _index = -1;
  181. _state = AsyncIteratorState.Iterating;
  182. goto case AsyncIteratorState.Iterating;
  183. case AsyncIteratorState.Iterating:
  184. if (await _enumerator.MoveNextAsync().ConfigureAwait(false))
  185. {
  186. var item = _enumerator.Current;
  187. checked
  188. {
  189. _index++;
  190. }
  191. _current = _selector(item, _index);
  192. return true;
  193. }
  194. break;
  195. }
  196. await DisposeAsync().ConfigureAwait(false);
  197. return false;
  198. }
  199. }
  200. internal sealed class SelectIListIterator<TSource, TResult> : AsyncIterator<TResult>, IAsyncIListProvider<TResult>
  201. {
  202. private readonly Func<TSource, TResult> _selector;
  203. private readonly IList<TSource> _source;
  204. private IEnumerator<TSource> _enumerator;
  205. public SelectIListIterator(IList<TSource> source, Func<TSource, TResult> selector)
  206. {
  207. Debug.Assert(source != null);
  208. Debug.Assert(selector != null);
  209. _source = source;
  210. _selector = selector;
  211. }
  212. public override AsyncIteratorBase<TResult> Clone()
  213. {
  214. return new SelectIListIterator<TSource, TResult>(_source, _selector);
  215. }
  216. public override async ValueTask DisposeAsync()
  217. {
  218. if (_enumerator != null)
  219. {
  220. _enumerator.Dispose();
  221. _enumerator = null;
  222. }
  223. await base.DisposeAsync().ConfigureAwait(false);
  224. }
  225. public Task<int> GetCountAsync(bool onlyIfCheap, CancellationToken cancellationToken)
  226. {
  227. if (onlyIfCheap)
  228. {
  229. return TaskExt.MinusOne;
  230. }
  231. var count = 0;
  232. foreach (var item in _source)
  233. {
  234. _selector(item);
  235. checked
  236. {
  237. count++;
  238. }
  239. }
  240. return Task.FromResult(count);
  241. }
  242. public override IAsyncEnumerable<TResult1> Select<TResult1>(Func<TResult, TResult1> selector)
  243. {
  244. return new SelectIListIterator<TSource, TResult1>(_source, CombineSelectors(_selector, selector));
  245. }
  246. public Task<TResult[]> ToArrayAsync(CancellationToken cancellationToken)
  247. {
  248. var n = _source.Count;
  249. var res = new TResult[n];
  250. for (var i = 0; i < n; i++)
  251. {
  252. res[i] = _selector(_source[i]);
  253. }
  254. return Task.FromResult(res);
  255. }
  256. public Task<List<TResult>> ToListAsync(CancellationToken cancellationToken)
  257. {
  258. var n = _source.Count;
  259. var res = new List<TResult>(n);
  260. for (var i = 0; i < n; i++)
  261. {
  262. res.Add(_selector(_source[i]));
  263. }
  264. return Task.FromResult(res);
  265. }
  266. protected override async ValueTask<bool> MoveNextCore()
  267. {
  268. switch (_state)
  269. {
  270. case AsyncIteratorState.Allocated:
  271. _enumerator = _source.GetEnumerator();
  272. _state = AsyncIteratorState.Iterating;
  273. goto case AsyncIteratorState.Iterating;
  274. case AsyncIteratorState.Iterating:
  275. if (_enumerator.MoveNext())
  276. {
  277. _current = _selector(_enumerator.Current);
  278. return true;
  279. }
  280. await DisposeAsync().ConfigureAwait(false);
  281. break;
  282. }
  283. return false;
  284. }
  285. }
  286. internal sealed class SelectEnumerableAsyncIteratorWithTask<TSource, TResult> : AsyncIterator<TResult>
  287. {
  288. private readonly Func<TSource, ValueTask<TResult>> _selector;
  289. private readonly IAsyncEnumerable<TSource> _source;
  290. private IAsyncEnumerator<TSource> _enumerator;
  291. public SelectEnumerableAsyncIteratorWithTask(IAsyncEnumerable<TSource> source, Func<TSource, ValueTask<TResult>> selector)
  292. {
  293. Debug.Assert(source != null);
  294. Debug.Assert(selector != null);
  295. _source = source;
  296. _selector = selector;
  297. }
  298. public override AsyncIteratorBase<TResult> Clone()
  299. {
  300. return new SelectEnumerableAsyncIteratorWithTask<TSource, TResult>(_source, _selector);
  301. }
  302. public override async ValueTask DisposeAsync()
  303. {
  304. if (_enumerator != null)
  305. {
  306. await _enumerator.DisposeAsync().ConfigureAwait(false);
  307. _enumerator = null;
  308. }
  309. await base.DisposeAsync().ConfigureAwait(false);
  310. }
  311. public override IAsyncEnumerable<TResult1> Select<TResult1>(Func<TResult, ValueTask<TResult1>> selector)
  312. {
  313. return new SelectEnumerableAsyncIteratorWithTask<TSource, TResult1>(_source, CombineSelectors(_selector, selector));
  314. }
  315. protected override async ValueTask<bool> MoveNextCore()
  316. {
  317. switch (_state)
  318. {
  319. case AsyncIteratorState.Allocated:
  320. _enumerator = _source.GetAsyncEnumerator(_cancellationToken);
  321. _state = AsyncIteratorState.Iterating;
  322. goto case AsyncIteratorState.Iterating;
  323. case AsyncIteratorState.Iterating:
  324. if (await _enumerator.MoveNextAsync().ConfigureAwait(false))
  325. {
  326. _current = await _selector(_enumerator.Current).ConfigureAwait(false);
  327. return true;
  328. }
  329. break;
  330. }
  331. await DisposeAsync().ConfigureAwait(false);
  332. return false;
  333. }
  334. }
  335. #if !NO_DEEP_CANCELLATION
  336. internal sealed class SelectEnumerableAsyncIteratorWithTaskAndCancellation<TSource, TResult> : AsyncIterator<TResult>
  337. {
  338. private readonly Func<TSource, CancellationToken, ValueTask<TResult>> _selector;
  339. private readonly IAsyncEnumerable<TSource> _source;
  340. private IAsyncEnumerator<TSource> _enumerator;
  341. public SelectEnumerableAsyncIteratorWithTaskAndCancellation(IAsyncEnumerable<TSource> source, Func<TSource, CancellationToken, ValueTask<TResult>> selector)
  342. {
  343. Debug.Assert(source != null);
  344. Debug.Assert(selector != null);
  345. _source = source;
  346. _selector = selector;
  347. }
  348. public override AsyncIteratorBase<TResult> Clone()
  349. {
  350. return new SelectEnumerableAsyncIteratorWithTaskAndCancellation<TSource, TResult>(_source, _selector);
  351. }
  352. public override async ValueTask DisposeAsync()
  353. {
  354. if (_enumerator != null)
  355. {
  356. await _enumerator.DisposeAsync().ConfigureAwait(false);
  357. _enumerator = null;
  358. }
  359. await base.DisposeAsync().ConfigureAwait(false);
  360. }
  361. public override IAsyncEnumerable<TResult1> Select<TResult1>(Func<TResult, CancellationToken, ValueTask<TResult1>> selector)
  362. {
  363. return new SelectEnumerableAsyncIteratorWithTaskAndCancellation<TSource, TResult1>(_source, CombineSelectors(_selector, selector));
  364. }
  365. protected override async ValueTask<bool> MoveNextCore()
  366. {
  367. switch (_state)
  368. {
  369. case AsyncIteratorState.Allocated:
  370. _enumerator = _source.GetAsyncEnumerator(_cancellationToken);
  371. _state = AsyncIteratorState.Iterating;
  372. goto case AsyncIteratorState.Iterating;
  373. case AsyncIteratorState.Iterating:
  374. if (await _enumerator.MoveNextAsync().ConfigureAwait(false))
  375. {
  376. _current = await _selector(_enumerator.Current, _cancellationToken).ConfigureAwait(false);
  377. return true;
  378. }
  379. break;
  380. }
  381. await DisposeAsync().ConfigureAwait(false);
  382. return false;
  383. }
  384. }
  385. #endif
  386. internal sealed class SelectEnumerableWithIndexAsyncIteratorWithTask<TSource, TResult> : AsyncIterator<TResult>
  387. {
  388. private readonly Func<TSource, int, ValueTask<TResult>> _selector;
  389. private readonly IAsyncEnumerable<TSource> _source;
  390. private IAsyncEnumerator<TSource> _enumerator;
  391. private int _index;
  392. public SelectEnumerableWithIndexAsyncIteratorWithTask(IAsyncEnumerable<TSource> source, Func<TSource, int, ValueTask<TResult>> selector)
  393. {
  394. Debug.Assert(source != null);
  395. Debug.Assert(selector != null);
  396. _source = source;
  397. _selector = selector;
  398. }
  399. public override AsyncIteratorBase<TResult> Clone()
  400. {
  401. return new SelectEnumerableWithIndexAsyncIteratorWithTask<TSource, TResult>(_source, _selector);
  402. }
  403. public override async ValueTask DisposeAsync()
  404. {
  405. if (_enumerator != null)
  406. {
  407. await _enumerator.DisposeAsync().ConfigureAwait(false);
  408. _enumerator = null;
  409. }
  410. await base.DisposeAsync().ConfigureAwait(false);
  411. }
  412. protected override async ValueTask<bool> MoveNextCore()
  413. {
  414. switch (_state)
  415. {
  416. case AsyncIteratorState.Allocated:
  417. _enumerator = _source.GetAsyncEnumerator(_cancellationToken);
  418. _index = -1;
  419. _state = AsyncIteratorState.Iterating;
  420. goto case AsyncIteratorState.Iterating;
  421. case AsyncIteratorState.Iterating:
  422. if (await _enumerator.MoveNextAsync().ConfigureAwait(false))
  423. {
  424. var item = _enumerator.Current;
  425. checked
  426. {
  427. _index++;
  428. }
  429. _current = await _selector(item, _index).ConfigureAwait(false);
  430. return true;
  431. }
  432. break;
  433. }
  434. await DisposeAsync().ConfigureAwait(false);
  435. return false;
  436. }
  437. }
  438. #if !NO_DEEP_CANCELLATION
  439. internal sealed class SelectEnumerableWithIndexAsyncIteratorWithTaskAndCancellation<TSource, TResult> : AsyncIterator<TResult>
  440. {
  441. private readonly Func<TSource, int, CancellationToken, ValueTask<TResult>> _selector;
  442. private readonly IAsyncEnumerable<TSource> _source;
  443. private IAsyncEnumerator<TSource> _enumerator;
  444. private int _index;
  445. public SelectEnumerableWithIndexAsyncIteratorWithTaskAndCancellation(IAsyncEnumerable<TSource> source, Func<TSource, int, CancellationToken, ValueTask<TResult>> selector)
  446. {
  447. Debug.Assert(source != null);
  448. Debug.Assert(selector != null);
  449. _source = source;
  450. _selector = selector;
  451. }
  452. public override AsyncIteratorBase<TResult> Clone()
  453. {
  454. return new SelectEnumerableWithIndexAsyncIteratorWithTaskAndCancellation<TSource, TResult>(_source, _selector);
  455. }
  456. public override async ValueTask DisposeAsync()
  457. {
  458. if (_enumerator != null)
  459. {
  460. await _enumerator.DisposeAsync().ConfigureAwait(false);
  461. _enumerator = null;
  462. }
  463. await base.DisposeAsync().ConfigureAwait(false);
  464. }
  465. protected override async ValueTask<bool> MoveNextCore()
  466. {
  467. switch (_state)
  468. {
  469. case AsyncIteratorState.Allocated:
  470. _enumerator = _source.GetAsyncEnumerator(_cancellationToken);
  471. _index = -1;
  472. _state = AsyncIteratorState.Iterating;
  473. goto case AsyncIteratorState.Iterating;
  474. case AsyncIteratorState.Iterating:
  475. if (await _enumerator.MoveNextAsync().ConfigureAwait(false))
  476. {
  477. var item = _enumerator.Current;
  478. checked
  479. {
  480. _index++;
  481. }
  482. _current = await _selector(item, _index, _cancellationToken).ConfigureAwait(false);
  483. return true;
  484. }
  485. break;
  486. }
  487. await DisposeAsync().ConfigureAwait(false);
  488. return false;
  489. }
  490. }
  491. #endif
  492. // NB: LINQ to Objects implements IPartition<TResult> for this. However, it seems incorrect to do so in a trivial
  493. // manner where e.g. TryGetLast simply indexes into the list without running the selector for the first n - 1
  494. // elements in order to ensure side-effects. We should consider whether we want to follow this implementation
  495. // strategy or support IAsyncPartition<TResult> in a less efficient but more correct manner here.
  496. internal sealed class SelectIListIteratorWithTask<TSource, TResult> : AsyncIterator<TResult>, IAsyncIListProvider<TResult>
  497. {
  498. private readonly Func<TSource, ValueTask<TResult>> _selector;
  499. private readonly IList<TSource> _source;
  500. private IEnumerator<TSource> _enumerator;
  501. public SelectIListIteratorWithTask(IList<TSource> source, Func<TSource, ValueTask<TResult>> selector)
  502. {
  503. Debug.Assert(source != null);
  504. Debug.Assert(selector != null);
  505. _source = source;
  506. _selector = selector;
  507. }
  508. public override AsyncIteratorBase<TResult> Clone()
  509. {
  510. return new SelectIListIteratorWithTask<TSource, TResult>(_source, _selector);
  511. }
  512. public override async ValueTask DisposeAsync()
  513. {
  514. if (_enumerator != null)
  515. {
  516. _enumerator.Dispose();
  517. _enumerator = null;
  518. }
  519. await base.DisposeAsync().ConfigureAwait(false);
  520. }
  521. public Task<int> GetCountAsync(bool onlyIfCheap, CancellationToken cancellationToken)
  522. {
  523. if (onlyIfCheap)
  524. {
  525. return TaskExt.MinusOne;
  526. }
  527. return Core();
  528. async Task<int> Core()
  529. {
  530. var count = 0;
  531. foreach (var item in _source)
  532. {
  533. await _selector(item).ConfigureAwait(false);
  534. checked
  535. {
  536. count++;
  537. }
  538. }
  539. return count;
  540. }
  541. }
  542. public override IAsyncEnumerable<TResult1> Select<TResult1>(Func<TResult, ValueTask<TResult1>> selector)
  543. {
  544. return new SelectIListIteratorWithTask<TSource, TResult1>(_source, CombineSelectors(_selector, selector));
  545. }
  546. public async Task<TResult[]> ToArrayAsync(CancellationToken cancellationToken)
  547. {
  548. var n = _source.Count;
  549. var res = new TResult[n];
  550. for (var i = 0; i < n; i++)
  551. {
  552. res[i] = await _selector(_source[i]).ConfigureAwait(false);
  553. }
  554. return res;
  555. }
  556. public async Task<List<TResult>> ToListAsync(CancellationToken cancellationToken)
  557. {
  558. var n = _source.Count;
  559. var res = new List<TResult>(n);
  560. for (var i = 0; i < n; i++)
  561. {
  562. res.Add(await _selector(_source[i]).ConfigureAwait(false));
  563. }
  564. return res;
  565. }
  566. protected override async ValueTask<bool> MoveNextCore()
  567. {
  568. switch (_state)
  569. {
  570. case AsyncIteratorState.Allocated:
  571. _enumerator = _source.GetEnumerator();
  572. _state = AsyncIteratorState.Iterating;
  573. goto case AsyncIteratorState.Iterating;
  574. case AsyncIteratorState.Iterating:
  575. if (_enumerator.MoveNext())
  576. {
  577. _current = await _selector(_enumerator.Current).ConfigureAwait(false);
  578. return true;
  579. }
  580. break;
  581. }
  582. await DisposeAsync().ConfigureAwait(false);
  583. return false;
  584. }
  585. }
  586. #if !NO_DEEP_CANCELLATION
  587. internal sealed class SelectIListIteratorWithTaskAndCancellation<TSource, TResult> : AsyncIterator<TResult>, IAsyncIListProvider<TResult>
  588. {
  589. private readonly Func<TSource, CancellationToken, ValueTask<TResult>> _selector;
  590. private readonly IList<TSource> _source;
  591. private IEnumerator<TSource> _enumerator;
  592. public SelectIListIteratorWithTaskAndCancellation(IList<TSource> source, Func<TSource, CancellationToken, ValueTask<TResult>> selector)
  593. {
  594. Debug.Assert(source != null);
  595. Debug.Assert(selector != null);
  596. _source = source;
  597. _selector = selector;
  598. }
  599. public override AsyncIteratorBase<TResult> Clone()
  600. {
  601. return new SelectIListIteratorWithTaskAndCancellation<TSource, TResult>(_source, _selector);
  602. }
  603. public override async ValueTask DisposeAsync()
  604. {
  605. if (_enumerator != null)
  606. {
  607. _enumerator.Dispose();
  608. _enumerator = null;
  609. }
  610. await base.DisposeAsync().ConfigureAwait(false);
  611. }
  612. public Task<int> GetCountAsync(bool onlyIfCheap, CancellationToken cancellationToken)
  613. {
  614. if (onlyIfCheap)
  615. {
  616. return TaskExt.MinusOne;
  617. }
  618. return Core();
  619. async Task<int> Core()
  620. {
  621. var count = 0;
  622. foreach (var item in _source)
  623. {
  624. await _selector(item, cancellationToken).ConfigureAwait(false);
  625. checked
  626. {
  627. count++;
  628. }
  629. }
  630. return count;
  631. }
  632. }
  633. public override IAsyncEnumerable<TResult1> Select<TResult1>(Func<TResult, CancellationToken, ValueTask<TResult1>> selector)
  634. {
  635. return new SelectIListIteratorWithTaskAndCancellation<TSource, TResult1>(_source, CombineSelectors(_selector, selector));
  636. }
  637. public async Task<TResult[]> ToArrayAsync(CancellationToken cancellationToken)
  638. {
  639. var n = _source.Count;
  640. var res = new TResult[n];
  641. for (var i = 0; i < n; i++)
  642. {
  643. res[i] = await _selector(_source[i], cancellationToken).ConfigureAwait(false);
  644. }
  645. return res;
  646. }
  647. public async Task<List<TResult>> ToListAsync(CancellationToken cancellationToken)
  648. {
  649. var n = _source.Count;
  650. var res = new List<TResult>(n);
  651. for (var i = 0; i < n; i++)
  652. {
  653. res.Add(await _selector(_source[i], cancellationToken).ConfigureAwait(false));
  654. }
  655. return res;
  656. }
  657. protected override async ValueTask<bool> MoveNextCore()
  658. {
  659. switch (_state)
  660. {
  661. case AsyncIteratorState.Allocated:
  662. _enumerator = _source.GetEnumerator();
  663. _state = AsyncIteratorState.Iterating;
  664. goto case AsyncIteratorState.Iterating;
  665. case AsyncIteratorState.Iterating:
  666. if (_enumerator.MoveNext())
  667. {
  668. _current = await _selector(_enumerator.Current, _cancellationToken).ConfigureAwait(false);
  669. return true;
  670. }
  671. break;
  672. }
  673. await DisposeAsync().ConfigureAwait(false);
  674. return false;
  675. }
  676. }
  677. #endif
  678. }
  679. }