wrapAsyncGenerator.js 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = _wrapAsyncGenerator;
  6. var _OverloadYield = require("./OverloadYield.js");
  7. function _wrapAsyncGenerator(fn) {
  8. return function () {
  9. return new AsyncGenerator(fn.apply(this, arguments));
  10. };
  11. }
  12. function AsyncGenerator(gen) {
  13. var front, back;
  14. function send(key, arg) {
  15. return new Promise(function (resolve, reject) {
  16. var request = {
  17. key: key,
  18. arg: arg,
  19. resolve: resolve,
  20. reject: reject,
  21. next: null
  22. };
  23. if (back) {
  24. back = back.next = request;
  25. } else {
  26. front = back = request;
  27. resume(key, arg);
  28. }
  29. });
  30. }
  31. function resume(key, arg) {
  32. try {
  33. var result = gen[key](arg);
  34. var value = result.value;
  35. var overloaded = value instanceof _OverloadYield.default;
  36. Promise.resolve(overloaded ? value.v : value).then(function (arg) {
  37. if (overloaded) {
  38. var nextKey = key === "return" ? "return" : "next";
  39. if (!value.k || arg.done) {
  40. return resume(nextKey, arg);
  41. } else {
  42. arg = gen[nextKey](arg).value;
  43. }
  44. }
  45. settle(result.done ? "return" : "normal", arg);
  46. }, function (err) {
  47. resume("throw", err);
  48. });
  49. } catch (err) {
  50. settle("throw", err);
  51. }
  52. }
  53. function settle(type, value) {
  54. switch (type) {
  55. case "return":
  56. front.resolve({
  57. value: value,
  58. done: true
  59. });
  60. break;
  61. case "throw":
  62. front.reject(value);
  63. break;
  64. default:
  65. front.resolve({
  66. value: value,
  67. done: false
  68. });
  69. break;
  70. }
  71. front = front.next;
  72. if (front) {
  73. resume(front.key, front.arg);
  74. } else {
  75. back = null;
  76. }
  77. }
  78. this._invoke = send;
  79. if (typeof gen.return !== "function") {
  80. this.return = undefined;
  81. }
  82. }
  83. AsyncGenerator.prototype[typeof Symbol === "function" && Symbol.asyncIterator || "@@asyncIterator"] = function () {
  84. return this;
  85. };
  86. AsyncGenerator.prototype.next = function (arg) {
  87. return this._invoke("next", arg);
  88. };
  89. AsyncGenerator.prototype.throw = function (arg) {
  90. return this._invoke("throw", arg);
  91. };
  92. AsyncGenerator.prototype.return = function (arg) {
  93. return this._invoke("return", arg);
  94. };
  95. //# sourceMappingURL=wrapAsyncGenerator.js.map