@@ -45,7 +45,7 @@ const {
|
45 | 45 | } = require('internal/streams/utils'); |
46 | 46 | |
47 | 47 | // Lazy load |
48 | | -let AsyncLocalStorage; |
| 48 | +let AsyncResource; |
49 | 49 | let addAbortListener; |
50 | 50 | |
51 | 51 | function isRequest(stream) { |
@@ -54,6 +54,14 @@ function isRequest(stream) {
|
54 | 54 | |
55 | 55 | const nop = () => {}; |
56 | 56 | |
| 57 | +function bindAsyncResource(fn, type) { |
| 58 | +AsyncResource ??= require('async_hooks').AsyncResource; |
| 59 | +const resource = new AsyncResource(type); |
| 60 | +return function(...args) { |
| 61 | +return resource.runInAsyncScope(fn, this, ...args); |
| 62 | +}; |
| 63 | +} |
| 64 | + |
57 | 65 | function eos(stream, options, callback) { |
58 | 66 | if (arguments.length === 2) { |
59 | 67 | callback = options; |
@@ -66,8 +74,9 @@ function eos(stream, options, callback) {
|
66 | 74 | validateFunction(callback, 'callback'); |
67 | 75 | validateAbortSignal(options.signal, 'options.signal'); |
68 | 76 | |
69 | | -AsyncLocalStorage ??= require('async_hooks').AsyncLocalStorage; |
70 | | -callback = once(AsyncLocalStorage.bind(callback)); |
| 77 | +// Avoid AsyncResource.bind() because it calls ObjectDefineProperties which |
| 78 | +// is a bottleneck here. |
| 79 | +callback = once(bindAsyncResource(callback, 'STREAM_END_OF_STREAM')); |
71 | 80 | |
72 | 81 | if (isReadableStream(stream) || isWritableStream(stream)) { |
73 | 82 | return eosWeb(stream, options, callback); |
|