esnext.async-disposable-stack.constructor.js 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. 'use strict';
  2. // https://github.com/tc39/proposal-async-explicit-resource-management
  3. var $ = require('../internals/export');
  4. var DESCRIPTORS = require('../internals/descriptors');
  5. var getBuiltIn = require('../internals/get-built-in');
  6. var aCallable = require('../internals/a-callable');
  7. var anInstance = require('../internals/an-instance');
  8. var defineBuiltIn = require('../internals/define-built-in');
  9. var defineBuiltIns = require('../internals/define-built-ins');
  10. var defineBuiltInAccessor = require('../internals/define-built-in-accessor');
  11. var wellKnownSymbol = require('../internals/well-known-symbol');
  12. var InternalStateModule = require('../internals/internal-state');
  13. var addDisposableResource = require('../internals/add-disposable-resource');
  14. var Promise = getBuiltIn('Promise');
  15. var SuppressedError = getBuiltIn('SuppressedError');
  16. var $ReferenceError = ReferenceError;
  17. var ASYNC_DISPOSE = wellKnownSymbol('asyncDispose');
  18. var TO_STRING_TAG = wellKnownSymbol('toStringTag');
  19. var ASYNC_DISPOSABLE_STACK = 'AsyncDisposableStack';
  20. var setInternalState = InternalStateModule.set;
  21. var getAsyncDisposableStackInternalState = InternalStateModule.getterFor(ASYNC_DISPOSABLE_STACK);
  22. var HINT = 'async-dispose';
  23. var DISPOSED = 'disposed';
  24. var PENDING = 'pending';
  25. var ALREADY_DISPOSED = ASYNC_DISPOSABLE_STACK + ' already disposed';
  26. var $AsyncDisposableStack = function AsyncDisposableStack() {
  27. setInternalState(anInstance(this, AsyncDisposableStackPrototype), {
  28. type: ASYNC_DISPOSABLE_STACK,
  29. state: PENDING,
  30. stack: []
  31. });
  32. if (!DESCRIPTORS) this.disposed = false;
  33. };
  34. var AsyncDisposableStackPrototype = $AsyncDisposableStack.prototype;
  35. defineBuiltIns(AsyncDisposableStackPrototype, {
  36. disposeAsync: function disposeAsync() {
  37. var asyncDisposableStack = this;
  38. return new Promise(function (resolve, reject) {
  39. var internalState = getAsyncDisposableStackInternalState(asyncDisposableStack);
  40. if (internalState.state == DISPOSED) return resolve(undefined);
  41. internalState.state = DISPOSED;
  42. if (!DESCRIPTORS) asyncDisposableStack.disposed = true;
  43. var stack = internalState.stack;
  44. var i = stack.length;
  45. var thrown = false;
  46. var suppressed;
  47. var handleError = function (result) {
  48. if (thrown) {
  49. suppressed = new SuppressedError(result, suppressed);
  50. } else {
  51. thrown = true;
  52. suppressed = result;
  53. }
  54. loop();
  55. };
  56. var loop = function () {
  57. if (i) {
  58. var disposeMethod = stack[--i];
  59. stack[i] = null;
  60. try {
  61. Promise.resolve(disposeMethod()).then(loop, handleError);
  62. } catch (error) {
  63. handleError(error);
  64. }
  65. } else {
  66. internalState.stack = null;
  67. thrown ? reject(suppressed) : resolve(undefined);
  68. }
  69. };
  70. loop();
  71. });
  72. },
  73. use: function use(value) {
  74. var internalState = getAsyncDisposableStackInternalState(this);
  75. if (internalState.state == DISPOSED) throw $ReferenceError(ALREADY_DISPOSED);
  76. addDisposableResource(internalState, value, HINT);
  77. return value;
  78. },
  79. adopt: function adopt(value, onDispose) {
  80. var internalState = getAsyncDisposableStackInternalState(this);
  81. if (internalState.state == DISPOSED) throw $ReferenceError(ALREADY_DISPOSED);
  82. aCallable(onDispose);
  83. addDisposableResource(internalState, undefined, HINT, function () {
  84. onDispose(value);
  85. });
  86. return value;
  87. },
  88. defer: function defer(onDispose) {
  89. var internalState = getAsyncDisposableStackInternalState(this);
  90. if (internalState.state == DISPOSED) throw $ReferenceError(ALREADY_DISPOSED);
  91. aCallable(onDispose);
  92. addDisposableResource(internalState, undefined, HINT, onDispose);
  93. },
  94. move: function move() {
  95. var internalState = getAsyncDisposableStackInternalState(this);
  96. if (internalState.state == DISPOSED) throw $ReferenceError(ALREADY_DISPOSED);
  97. var newAsyncDisposableStack = new $AsyncDisposableStack();
  98. getAsyncDisposableStackInternalState(newAsyncDisposableStack).stack = internalState.stack;
  99. internalState.stack = [];
  100. return newAsyncDisposableStack;
  101. }
  102. });
  103. if (DESCRIPTORS) defineBuiltInAccessor(AsyncDisposableStackPrototype, 'disposed', {
  104. configurable: true,
  105. get: function disposed() {
  106. return getAsyncDisposableStackInternalState(this).state == DISPOSED;
  107. }
  108. });
  109. defineBuiltIn(AsyncDisposableStackPrototype, ASYNC_DISPOSE, AsyncDisposableStackPrototype.disposeAsync, { name: 'disposeAsync' });
  110. defineBuiltIn(AsyncDisposableStackPrototype, TO_STRING_TAG, ASYNC_DISPOSABLE_STACK, { nonWritable: true });
  111. $({ global: true, constructor: true, forced: true }, {
  112. AsyncDisposableStack: $AsyncDisposableStack
  113. });