You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

131 lines
4.6 KiB

'use strict';
// https://github.com/tc39/proposal-async-explicit-resource-management
var $ = require('../internals/export');
var DESCRIPTORS = require('../internals/descriptors');
var getBuiltIn = require('../internals/get-built-in');
var aCallable = require('../internals/a-callable');
var anInstance = require('../internals/an-instance');
var defineBuiltIn = require('../internals/define-built-in');
var defineBuiltIns = require('../internals/define-built-ins');
var defineBuiltInAccessor = require('../internals/define-built-in-accessor');
var wellKnownSymbol = require('../internals/well-known-symbol');
var InternalStateModule = require('../internals/internal-state');
var addDisposableResource = require('../internals/add-disposable-resource');
var Promise = getBuiltIn('Promise');
var SuppressedError = getBuiltIn('SuppressedError');
var $ReferenceError = ReferenceError;
var ASYNC_DISPOSE = wellKnownSymbol('asyncDispose');
var TO_STRING_TAG = wellKnownSymbol('toStringTag');
var ASYNC_DISPOSABLE_STACK = 'AsyncDisposableStack';
var setInternalState = InternalStateModule.set;
var getAsyncDisposableStackInternalState = InternalStateModule.getterFor(ASYNC_DISPOSABLE_STACK);
var HINT = 'async-dispose';
var DISPOSED = 'disposed';
var PENDING = 'pending';
var getPendingAsyncDisposableStackInternalState = function (stack) {
var internalState = getAsyncDisposableStackInternalState(stack);
if (internalState.state === DISPOSED) throw new $ReferenceError(ASYNC_DISPOSABLE_STACK + ' already disposed');
return internalState;
};
var $AsyncDisposableStack = function AsyncDisposableStack() {
setInternalState(anInstance(this, AsyncDisposableStackPrototype), {
type: ASYNC_DISPOSABLE_STACK,
state: PENDING,
stack: []
});
if (!DESCRIPTORS) this.disposed = false;
};
var AsyncDisposableStackPrototype = $AsyncDisposableStack.prototype;
defineBuiltIns(AsyncDisposableStackPrototype, {
disposeAsync: function disposeAsync() {
var asyncDisposableStack = this;
return new Promise(function (resolve, reject) {
var internalState = getAsyncDisposableStackInternalState(asyncDisposableStack);
if (internalState.state === DISPOSED) return resolve(undefined);
internalState.state = DISPOSED;
if (!DESCRIPTORS) asyncDisposableStack.disposed = true;
var stack = internalState.stack;
var i = stack.length;
var thrown = false;
var suppressed;
var handleError = function (result) {
if (thrown) {
suppressed = new SuppressedError(result, suppressed);
} else {
thrown = true;
suppressed = result;
}
loop();
};
var loop = function () {
if (i) {
var disposeMethod = stack[--i];
stack[i] = null;
try {
Promise.resolve(disposeMethod()).then(loop, handleError);
} catch (error) {
handleError(error);
}
} else {
internalState.stack = null;
thrown ? reject(suppressed) : resolve(undefined);
}
};
loop();
});
},
use: function use(value) {
addDisposableResource(getPendingAsyncDisposableStackInternalState(this), value, HINT);
return value;
},
adopt: function adopt(value, onDispose) {
var internalState = getPendingAsyncDisposableStackInternalState(this);
aCallable(onDispose);
addDisposableResource(internalState, undefined, HINT, function () {
return onDispose(value);
});
return value;
},
defer: function defer(onDispose) {
var internalState = getPendingAsyncDisposableStackInternalState(this);
aCallable(onDispose);
addDisposableResource(internalState, undefined, HINT, onDispose);
},
move: function move() {
var internalState = getPendingAsyncDisposableStackInternalState(this);
var newAsyncDisposableStack = new $AsyncDisposableStack();
getAsyncDisposableStackInternalState(newAsyncDisposableStack).stack = internalState.stack;
internalState.stack = [];
internalState.state = DISPOSED;
if (!DESCRIPTORS) this.disposed = true;
return newAsyncDisposableStack;
}
});
if (DESCRIPTORS) defineBuiltInAccessor(AsyncDisposableStackPrototype, 'disposed', {
configurable: true,
get: function disposed() {
return getAsyncDisposableStackInternalState(this).state === DISPOSED;
}
});
defineBuiltIn(AsyncDisposableStackPrototype, ASYNC_DISPOSE, AsyncDisposableStackPrototype.disposeAsync, { name: 'disposeAsync' });
defineBuiltIn(AsyncDisposableStackPrototype, TO_STRING_TAG, ASYNC_DISPOSABLE_STACK, { nonWritable: true });
$({ global: true, constructor: true }, {
AsyncDisposableStack: $AsyncDisposableStack
});