esnext.async-iterator.take.js 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. 'use strict';
  2. // https://github.com/tc39/proposal-iterator-helpers
  3. var $ = require('../internals/export');
  4. var call = require('../internals/function-call');
  5. var anObject = require('../internals/an-object');
  6. var getIteratorDirect = require('../internals/get-iterator-direct');
  7. var notANaN = require('../internals/not-a-nan');
  8. var toPositiveInteger = require('../internals/to-positive-integer');
  9. var createAsyncIteratorProxy = require('../internals/async-iterator-create-proxy');
  10. var AsyncIteratorProxy = createAsyncIteratorProxy(function (Promise) {
  11. var state = this;
  12. var iterator = state.iterator;
  13. var returnMethod;
  14. if (!state.remaining--) {
  15. var resultDone = { value: undefined, done: true };
  16. state.done = true;
  17. returnMethod = iterator['return'];
  18. if (returnMethod !== undefined) {
  19. return Promise.resolve(call(returnMethod, iterator, undefined)).then(function () {
  20. return resultDone;
  21. });
  22. }
  23. return resultDone;
  24. } return Promise.resolve(call(state.next, iterator)).then(function (step) {
  25. if (anObject(step).done) {
  26. state.done = true;
  27. return { value: undefined, done: true };
  28. } return { value: step.value, done: false };
  29. }).then(null, function (error) {
  30. state.done = true;
  31. throw error;
  32. });
  33. });
  34. $({ target: 'AsyncIterator', proto: true, real: true, forced: true }, {
  35. take: function take(limit) {
  36. return new AsyncIteratorProxy(getIteratorDirect(this), {
  37. remaining: toPositiveInteger(notANaN(+limit))
  38. });
  39. }
  40. });