-internal.js 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244
  1. import {
  2. objectOrFunction,
  3. isFunction
  4. } from './utils';
  5. import {
  6. asap
  7. } from './asap';
  8. import originalThen from './then';
  9. import originalResolve from './promise/resolve';
  10. export const PROMISE_ID = Math.random().toString(36).substring(2);
  11. function noop() {}
  12. const PENDING = void 0;
  13. const FULFILLED = 1;
  14. const REJECTED = 2;
  15. function selfFulfillment() {
  16. return new TypeError("You cannot resolve a promise with itself");
  17. }
  18. function cannotReturnOwn() {
  19. return new TypeError('A promises callback cannot return that same promise.');
  20. }
  21. function tryThen(then, value, fulfillmentHandler, rejectionHandler) {
  22. try {
  23. then.call(value, fulfillmentHandler, rejectionHandler);
  24. } catch(e) {
  25. return e;
  26. }
  27. }
  28. function handleForeignThenable(promise, thenable, then) {
  29. asap(promise => {
  30. let sealed = false;
  31. let error = tryThen(then, thenable, value => {
  32. if (sealed) { return; }
  33. sealed = true;
  34. if (thenable !== value) {
  35. resolve(promise, value);
  36. } else {
  37. fulfill(promise, value);
  38. }
  39. }, reason => {
  40. if (sealed) { return; }
  41. sealed = true;
  42. reject(promise, reason);
  43. }, 'Settle: ' + (promise._label || ' unknown promise'));
  44. if (!sealed && error) {
  45. sealed = true;
  46. reject(promise, error);
  47. }
  48. }, promise);
  49. }
  50. function handleOwnThenable(promise, thenable) {
  51. if (thenable._state === FULFILLED) {
  52. fulfill(promise, thenable._result);
  53. } else if (thenable._state === REJECTED) {
  54. reject(promise, thenable._result);
  55. } else {
  56. subscribe(thenable, undefined, value => resolve(promise, value),
  57. reason => reject(promise, reason))
  58. }
  59. }
  60. function handleMaybeThenable(promise, maybeThenable, then) {
  61. if (maybeThenable.constructor === promise.constructor &&
  62. then === originalThen &&
  63. maybeThenable.constructor.resolve === originalResolve) {
  64. handleOwnThenable(promise, maybeThenable);
  65. } else {
  66. if (then === undefined) {
  67. fulfill(promise, maybeThenable);
  68. } else if (isFunction(then)) {
  69. handleForeignThenable(promise, maybeThenable, then);
  70. } else {
  71. fulfill(promise, maybeThenable);
  72. }
  73. }
  74. }
  75. function resolve(promise, value) {
  76. if (promise === value) {
  77. reject(promise, selfFulfillment());
  78. } else if (objectOrFunction(value)) {
  79. let then;
  80. try {
  81. then = value.then;
  82. } catch (error) {
  83. reject(promise, error);
  84. return;
  85. }
  86. handleMaybeThenable(promise, value, then);
  87. } else {
  88. fulfill(promise, value);
  89. }
  90. }
  91. function publishRejection(promise) {
  92. if (promise._onerror) {
  93. promise._onerror(promise._result);
  94. }
  95. publish(promise);
  96. }
  97. function fulfill(promise, value) {
  98. if (promise._state !== PENDING) { return; }
  99. promise._result = value;
  100. promise._state = FULFILLED;
  101. if (promise._subscribers.length !== 0) {
  102. asap(publish, promise);
  103. }
  104. }
  105. function reject(promise, reason) {
  106. if (promise._state !== PENDING) { return; }
  107. promise._state = REJECTED;
  108. promise._result = reason;
  109. asap(publishRejection, promise);
  110. }
  111. function subscribe(parent, child, onFulfillment, onRejection) {
  112. let { _subscribers } = parent;
  113. let { length } = _subscribers;
  114. parent._onerror = null;
  115. _subscribers[length] = child;
  116. _subscribers[length + FULFILLED] = onFulfillment;
  117. _subscribers[length + REJECTED] = onRejection;
  118. if (length === 0 && parent._state) {
  119. asap(publish, parent);
  120. }
  121. }
  122. function publish(promise) {
  123. let subscribers = promise._subscribers;
  124. let settled = promise._state;
  125. if (subscribers.length === 0) { return; }
  126. let child, callback, detail = promise._result;
  127. for (let i = 0; i < subscribers.length; i += 3) {
  128. child = subscribers[i];
  129. callback = subscribers[i + settled];
  130. if (child) {
  131. invokeCallback(settled, child, callback, detail);
  132. } else {
  133. callback(detail);
  134. }
  135. }
  136. promise._subscribers.length = 0;
  137. }
  138. function invokeCallback(settled, promise, callback, detail) {
  139. let hasCallback = isFunction(callback),
  140. value, error, succeeded = true;
  141. if (hasCallback) {
  142. try {
  143. value = callback(detail);
  144. } catch (e) {
  145. succeeded = false;
  146. error = e;
  147. }
  148. if (promise === value) {
  149. reject(promise, cannotReturnOwn());
  150. return;
  151. }
  152. } else {
  153. value = detail;
  154. }
  155. if (promise._state !== PENDING) {
  156. // noop
  157. } else if (hasCallback && succeeded) {
  158. resolve(promise, value);
  159. } else if (succeeded === false) {
  160. reject(promise, error);
  161. } else if (settled === FULFILLED) {
  162. fulfill(promise, value);
  163. } else if (settled === REJECTED) {
  164. reject(promise, value);
  165. }
  166. }
  167. function initializePromise(promise, resolver) {
  168. try {
  169. resolver(function resolvePromise(value){
  170. resolve(promise, value);
  171. }, function rejectPromise(reason) {
  172. reject(promise, reason);
  173. });
  174. } catch(e) {
  175. reject(promise, e);
  176. }
  177. }
  178. let id = 0;
  179. function nextId() {
  180. return id++;
  181. }
  182. function makePromise(promise) {
  183. promise[PROMISE_ID] = id++;
  184. promise._state = undefined;
  185. promise._result = undefined;
  186. promise._subscribers = [];
  187. }
  188. export {
  189. nextId,
  190. makePromise,
  191. noop,
  192. resolve,
  193. reject,
  194. fulfill,
  195. subscribe,
  196. publish,
  197. publishRejection,
  198. initializePromise,
  199. invokeCallback,
  200. FULFILLED,
  201. REJECTED,
  202. PENDING,
  203. handleMaybeThenable
  204. };