util: add util.disposer helper to wrap a dispose function

Add `util.disposer` and `util.asyncDisposer` to conveniently wrap a
function to be a disposable, and allow it to be used with `using`
declaration.

Author: legendecas

doc/api/util.md

@@ -3655,6 +3655,89 @@ Returns `true` if the value is a built-in {WeakSet} instance.
 util.types.isWeakSet(new WeakSet());  // Returns true
 ```
 
+## Disposer APIs
+
+> Stability: 1 - Experimental
+
+### `util.asyncDisposer(disposeFn)`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* `disposeFn` {Function} A dispose function returning a promise
+* Returns: {AsyncDisposer}
+
+Create a convenient wrapper on the given async function that can be used
+with `using` declaration.
+
+If an error is thrown in the function, instead of returning a promise,
+the error will be wrapped in a rejected promise.
+
+```mjs
+{
+  await using _ = util.disposer(async function disposer() {
+    // Performing async disposing actions...
+  });
+
+  // Doing some works...
+
+} // When this scope exits, the disposer function is invoked and awaited.
+```
+
+### `util.disposer(disposeFn)`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+* `disposeFn` {Function} A dispose function
+* Returns: {Disposer}
+
+Create a convenient wrapper on the given function that can be used with
+`using` declaration.
+
+```js
+{
+  using _ = util.disposer(function disposer() {
+    // Performing disposing actions...
+  });
+
+  // Doing some works...
+
+} // When this scope exits, the disposer function is invoked.
+```
+
+### Class: `util.AsyncDisposer`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+A convenience wrapper on an async dispose function.
+
+#### `asyncDisposer[Symbol.asyncDispose]()`
+
+Invokes the function specified in `util.asyncDisposer(disposeFn)`.
+
+Multiple invocations on this method only result in a single
+invocation of the `disposeFn`.
+
+### Class: `util.Disposer`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+A convenience wrapper on a dispose function.
+
+#### `disposer[Symbol.dispose]()`
+
+Invokes the function specified in `util.disposer(disposeFn)`.
+
+Multiple invocations on this method only result in a single
+invocation of the `disposeFn`.
+
 ## Deprecated APIs
 
 The following APIs are deprecated and should no longer be used. Existing

lib/internal/util/disposer.js

@@ -0,0 +1,65 @@
+'use strict';
+
+const {
+  PromiseWithResolvers,
+  SymbolAsyncDispose,
+  SymbolDispose,
+} = primordials;
+const {
+  validateFunction,
+} = require('internal/validators');
+
+class Disposer {
+  #disposed = false;
+  #disposeFn;
+  constructor(disposeFn) {
+    validateFunction(disposeFn, 'disposeFn');
+    this.#disposeFn = disposeFn;
+  }
+
+  [SymbolDispose]() {
+    if (this.#disposed) {
+      return;
+    }
+    this.#disposed = true;
+    this.#disposeFn();
+  }
+}
+
+class AsyncDisposer {
+  /**
+   * @type {PromiseWithResolvers<void>}
+   */
+  #disposeDeferred;
+  #disposeFn;
+  constructor(disposeFn) {
+    validateFunction(disposeFn, 'disposeFn');
+    this.#disposeFn = disposeFn;
+  }
+
+  [SymbolAsyncDispose]() {
+    if (this.#disposeDeferred === undefined) {
+      this.#disposeDeferred = PromiseWithResolvers();
+      try {
+        const ret = this.#disposeFn();
+        this.#disposeDeferred.resolve(ret);
+      } catch (err) {
+        this.#disposeDeferred.reject(err);
+      }
+    }
+    return this.#disposeDeferred.promise;
+  }
+}
+
+function disposer(disposeFn) {
+  return new Disposer(disposeFn);
+}
+
+function asyncDisposer(disposeFn) {
+  return new AsyncDisposer(disposeFn);
+}
+
+module.exports = {
+  disposer,
+  asyncDisposer,
+};

lib/util.js

@@ -491,3 +491,9 @@ defineLazyProperties(
   'internal/util/diff',
   ['diff'],
 );
+
+defineLazyProperties(
+  module.exports,
+  'internal/util/disposer',
+  ['disposer', 'asyncDisposer', 'Disposer', 'AsyncDisposer'],
+);

test/parallel/test-util-asyncdisposer.js

@@ -0,0 +1,78 @@
+'use strict';
+
+// This test checks that the semantics of `util.asyncDisposer` are as described in
+// the API docs
+
+const common = require('../common');
+const assert = require('node:assert');
+const { asyncDisposer } = require('node:util');
+const test = require('node:test');
+
+test('util.asyncDisposer should throw on non-function first parameter', () => {
+  const invalidDisposers = [
+    null,
+    undefined,
+    42,
+    'string',
+    {},
+    [],
+    Symbol('symbol'),
+  ];
+  for (const invalidDisposer of invalidDisposers) {
+    assert.throws(() => {
+      asyncDisposer(invalidDisposer);
+    }, {
+      code: 'ERR_INVALID_ARG_TYPE',
+      name: 'TypeError',
+    });
+  }
+});
+
+test('util.asyncDisposer should create a AsyncDisposable object', async () => {
+  const disposeFn = common.mustCall();
+  const disposable = asyncDisposer(disposeFn);
+  assert.strictEqual(typeof disposable, 'object');
+  assert.strictEqual(disposable[Symbol.dispose], undefined);
+  assert.strictEqual(typeof disposable[Symbol.asyncDispose], 'function');
+
+  // Multiple calls to asyncDispose should not throw and only invoke the function once.
+  const p1 = disposable[Symbol.asyncDispose]();
+  const p2 = disposable[Symbol.asyncDispose]();
+  assert.strictEqual(p1, p2);
+  await p1;
+});
+
+test('AsyncDisposer[Symbol.asyncDispose] must be invoked with an AsyncDisposer', () => {
+  const disposeFn = common.mustNotCall();
+  const disposable = asyncDisposer(disposeFn);
+  const asyncDispose = disposable[Symbol.asyncDispose];
+  assert.throws(() => {
+    asyncDispose.call({}); // Call with a non-AsyncDisposer object
+  }, TypeError);
+});
+
+test('AsyncDisposer[Symbol.asyncDispose] should reject if the disposerFn throws sync', async () => {
+  const disposeFn = common.mustCall(() => {
+    throw new Error('Disposer error');
+  });
+  const disposable = asyncDisposer(disposeFn);
+  const promise = disposable[Symbol.asyncDispose]();
+
+  await assert.rejects(promise, {
+    name: 'Error',
+    message: 'Disposer error',
+  });
+});
+
+test('Disposer[Symbol.asyncDispose] should reject if the disposerFn rejects', async () => {
+  const disposeFn = common.mustCall(() => {
+    return Promise.reject(new Error('Disposer error'));
+  });
+  const disposable = asyncDisposer(disposeFn);
+  const promise = disposable[Symbol.asyncDispose]();
+
+  await assert.rejects(promise, {
+    name: 'Error',
+    message: 'Disposer error',
+  });
+});

test/parallel/test-util-disposer.js

@@ -0,0 +1,65 @@
+'use strict';
+
+// This test checks that the semantics of `util.disposer` are as described in
+// the API docs
+
+const common = require('../common');
+const assert = require('node:assert');
+const { disposer } = require('node:util');
+const test = require('node:test');
+
+test('util.disposer should throw on non-function first parameter', () => {
+  const invalidDisposers = [
+    null,
+    undefined,
+    42,
+    'string',
+    {},
+    [],
+    Symbol('symbol'),
+  ];
+  for (const invalidDisposer of invalidDisposers) {
+    assert.throws(() => {
+      disposer(invalidDisposer);
+    }, {
+      code: 'ERR_INVALID_ARG_TYPE',
+      name: 'TypeError',
+    });
+  }
+});
+
+test('util.disposer should create a Disposable object', () => {
+  const disposeFn = common.mustCall();
+  const disposable = disposer(disposeFn);
+  assert.strictEqual(typeof disposable, 'object');
+  assert.strictEqual(typeof disposable[Symbol.dispose], 'function');
+  assert.strictEqual(disposable[Symbol.asyncDispose], undefined);
+  disposable[Symbol.dispose]();
+  // Multiple calls to dispose should not throw and only invoke the function once.
+  disposable[Symbol.dispose]();
+});
+
+test('Disposer[Symbol.dispose] must be invoked with an Disposer', () => {
+  const disposeFn = common.mustNotCall();
+  const disposable = disposer(disposeFn);
+  const dispose = disposable[Symbol.dispose];
+  assert.throws(() => {
+    dispose.call({}); // Call with a non-Disposer object
+  }, TypeError);
+});
+
+test('Disposer[Symbol.dispose] should throw if the disposerFn throws', () => {
+  const disposeFn = common.mustCall(() => {
+    throw new Error('Disposer error');
+  });
+  const disposable = disposer(disposeFn);
+  assert.throws(() => {
+    disposable[Symbol.dispose]();
+  }, {
+    name: 'Error',
+    message: 'Disposer error',
+  });
+
+  // Multiple calls to dispose should not throw and only invoke the function once.
+  disposable[Symbol.dispose]();
+});