33 lines
841 B
JavaScript
33 lines
841 B
JavaScript
'use strict';
|
|
|
|
const common = require('../common');
|
|
const assert = require('assert');
|
|
const gcTrackerMap = new WeakMap();
|
|
const gcTrackerTag = 'NODE_TEST_COMMON_GC_TRACKER';
|
|
|
|
function onGC(obj, gcListener) {
|
|
const async_hooks = require('async_hooks');
|
|
|
|
const onGcAsyncHook = async_hooks.createHook({
|
|
init: common.mustCallAtLeast(function(id, type) {
|
|
if (this.trackedId === undefined) {
|
|
assert.strictEqual(type, gcTrackerTag);
|
|
this.trackedId = id;
|
|
}
|
|
}),
|
|
destroy(id) {
|
|
assert.notStrictEqual(this.trackedId, -1);
|
|
if (id === this.trackedId) {
|
|
this.gcListener.ongc();
|
|
onGcAsyncHook.disable();
|
|
}
|
|
}
|
|
}).enable();
|
|
onGcAsyncHook.gcListener = gcListener;
|
|
|
|
gcTrackerMap.set(obj, new async_hooks.AsyncResource(gcTrackerTag));
|
|
obj = null;
|
|
}
|
|
|
|
module.exports = onGC;
|